i6ddlZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z eegefZGddZeZGdd eZGd d ej&Zy) N)wraps)AnyCallableDict FrozenSetOptionalSet) warn_or_error)AdapterDeprecationWarningcjeZdZdedefdZdedefdZ d dedeedefdZ dedefd Z dedefd Z y) _Availablefuncreturncd|_|S)NT)_is_available_)selfrs W/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt/adapters/base/meta.py__call__z_Available.__call__s" parse_replacementcfd}|S)aA decorator factory to indicate that a method on the adapter will be exposed to the database wrapper, and will be stubbed out at parse time with the given function. @available.parse() def my_method(self, a, b): if something: return None return big_expensive_db_query() @available.parse(lambda *args, **args: {}) def my_other_method(self, a, b): x = {} x.update(big_expensive_db_query()) return x c"|_|SN)_parse_replacement_)rrrs rinnerz_Available.parse..inner$s'8D $: r)rrrs`` rparsez_Available.parses$  rNsupported_namecfd}|S)a A decorator that marks a function as available, but also prints a deprecation warning. Use like @available.deprecated('my_new_method') def my_old_method(self, arg): args = compatability_shim(arg) return self.my_new_method(*args) @available.deprecated('my_new_slow_method', lambda *a, **k: (0, '')) def my_old_slow_method(self, arg): args = compatibility_shim(arg) return self.my_new_slow_method(*args) To make `adapter.my_old_method` available but also print out a warning on use directing users to `my_new_method`. The optional parse_replacement, if provided, will provide a parse-time replacement for the actual method (see `available.parse`). cjtfd}rj}n}||S)Nc@tt|i|S)N)old_namenew_name)r r )argskwargsr func_namers rrz5_Available.deprecated..wrapper..innerDs'-y>ZT,V,,r)__name__rr)rravailable_functionr&rrrs` @rwrapperz&_Available.deprecated..wrapperAsH I 4[ - - !%)ZZ0A%B"%)"%e, ,rr)rrrr)s``` r deprecatedz_Available.deprecated*s. - rc6|jd}||S)Ncyrraks rz'_Available.parse_none..Tsrrrrr)s r parse_nonez_Available.parse_noneSs**12t}rc6|jd}||S)NcgSrrr-s rr0z'_Available.parse_list..XsRrr1r2s r parse_listz_Available.parse_listWs**/0t}rr) r' __module__ __qualname__rr Decoratorrstrrr*r3r6rrrr r sxX(xI2LP'!'6>x6H' 'RxHxHrr ceZdZdZdZy)available_propertya This supports making dynamic properties (`@property`) available in the jinja context. We use `@available` to make methods available in the jinja context, but this mechanism relies on the method being callable. Intuitively, we should be able to use both `@available` and `@property` to create a dynamic property that's available in the jinja context. Using the `@property` decorator as the inner decorator supplies `@available` with something that is not callable. Instead of returning the method, `@property` returns the value itself, not the method that is called to create the value. Using the `@available` decorator as the inner decorator adds `_is_available_ = True` to the function. However, when the `@property` decorator executes, it returns a `property` object which does not have the `_is_available_` attribute. This decorator solves this problem by simply adding `_is_available_ = True` as an attribute on the `property` built-in. TN)r'r7r8__doc__rrrrr<r<_s Nrr<c:eZdZUeeed<eeefed<ddZy) AdapterMeta _available__parse_replacements_c tjj||||fi|}t}i}|D]J}|j t |dt|j t |dtL|j D]8\}} t | ddr|j|t | dd} | 4| ||<:t||_ ||_ |S)Nr@rArFr) abcABCMeta__new__setupdategetattritemsadd frozensetr@rA) mclsnamebases namespacer%cls available replacementsbasevaluers rrEzAdapterMeta.__new__vs kk!!$eYI&I"e ')  ND   WT=#%@ A   .Dce L M N %??, 7KD%u.6 d# '/Dd K  ,%6 T"  7$I.#/  rN)rr?) r'r7r8rr:__annotations__rrrErrrr?r?rs 3sH}-- rr?)rC functoolsrtypingrrrrrr dbt_common.events.functionsr dbt.adapters.events.typesr r9r rQpropertyr<rDr?rrrr[s[ @@5? cUH_ % LL^ L &$#++$r