i# vUddlZddlmZmZmZmZmZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZddlmZmZej*rdd lmZiaeeefed <d ed efd Z dd edeeefdedd eefdZdeeefdedd eefdZded eeeeffdZded eeeeffdZ d efdZ!y)N)AnyDictListOptionalUnion)RefArgs)MacroNamespaceNotStringError ParsingError)get_environment)MacroNameNotStringError)test_caching_enabled)ExtractionErrorpy_extract_from_source)ParseDatabaseWrapper_TESTING_MACRO_CACHEsourcereturnctdd}|j|}t|jtj j }|D] }t|ds|jdk(s yy)z9Checks whether the raw jinja has any references to `this`NTcapture_macrosnamethisF) r parsetuplefind_alljinja2nodesNamehasattrr)renvparsednamesrs Y/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt/clients/jinja_static.py statically_extract_has_name_thisr$se $t 4C YYv F &//&,,"3"34 5E 4 TYY&%8 ctx db_wrapperrctdd}tr+|tvr#tj|d}t |d}nc|j |}t |jtjj}tr|t|<t|d|gd}g}|D]A}d} t|dr-t|jdr|jj} nt|drt|jdrt|jjj dk(rt|jdri|jjj} |jj"} | d k(r&| d k(rt%|||} |j'| n | d | } n| s| |vr|j| r+| |vs1|j)| D|S) NTr_dbt_cached_calls)rrefconfignoderrattradapterdispatch.)r r rgetgetattrrrrrrCallsetattrrr,rtype__name__r-!statically_parse_adapter_dispatchextendappend) rr&r'r r! func_callsstandard_callspossible_macro_calls func_call func_name package_name macro_namead_macro_callss r#statically_extract_macro_callsrB s $t 4C&,@"@%))&$7V%89 6"6??6<<+<+<=>  !+1  ( F/ <0N"7  9f %')..&*I!++I 6*INNF3,,-66&@INNF3(~~2277 &^^00 9,!Z/)J%sJ*-33NC!#/.* >I   &  WWY   44$++I6E"7H r%c|g}d}t|jdkDr|jdj}|r|j|d}d}d}t|jdkDr1|jd}t |jdj }|j r|j D]}|jdk(rjt |jj dk(r(|jj}|j|\t|jj|jdk(st |jj } | dk(r|jj}t| |rE|dk(r/g} |jD]} | j| jn|dk(r |j}|rK|j||j} | jd | j}|j||S|r|g} ng} | D]} |j| d ||S) Nrr@Const) kwarg_valuemacro_namespacer)rGr0)lenargsvaluer9r5r6kwargskeyr r itemsr/macror?r)r=r&r'r<r>rG packages_argpackages_arg_typekwarg kwarg_typepackagesitemrNr?s r#r7r7[s)I 9>>QNN1%++ ##I.OL 9>>Q ~~a(  !23<<%% CEyyL( $--8 % 1 1I(// :1ekk>O>OPP//!%++.77 (&+kk&7&7O6zBB C"  &H$** , + , ' )*00O##I#OUU))*!EJJ<8 ##I.  '(HH$ GL ' '<.)(E F G r% expressionc td|d}|jdrNt |dd}t |jd|jd|jd }|S|jd rt |d d\}}||g}|Std |#t$rtd|wxYw) a Returns a RefArgs or List[str] object, corresponding to ref or source respectively, given an input jinja expression. input: str representing how input node is referenced in tested model sql * examples: - "ref('my_model_a')" - "ref('my_model_a', version=3)" - "ref('package', 'my_model_a', version=3)" - "source('my_source_schema', 'my_source_name')" If input is not a well-formed jinja ref or source expression, a ParsingError is raised. z{{ z }}zInvalid jinja expression: refsrpackagerversion)rXrrYsourcesz"Invalid ref or source expression: )rrr r1listr)rUstatically_parsedraw_ref ref_or_source source_namesource_table_names r#statically_parse_ref_or_sourcerasF2U:,e3LMV$(01!4KK *V$KK *     y ))-.? .J)KA)N& &$&78  ? |LMM F7 |DEEFs B))Cstringcd|vrytdd}tr1tr+|tvr#tj|d}t |d}nC|j |}t |jtjj}ttd|}|r|dnd}|syi}|jD]}t|||j<|S)a Given a string with jinja, extract an unrendered config call. If no config call is present, returns None. For example, given: "{{ config(materialized=env_var('DBT_TEST_STATE_MODIFIED')) }} select 1 as id" returns: {'materialized': "Keyword(key='materialized', value=Call(node=Name(name='env_var', ctx='load'), args=[Const(value='DBT_TEST_STATE_MODIFIED')], kwargs=[], dyn_args=None, dyn_kwargs=None))"} No config call: "select 1 as id" returns: None zconfig(NTrr)ct|dxr1t|jdxr|jjdk(S)Nr,rr+)rr,r)fs r#z4statically_parse_unrendered_config..s0ga(`WQVVV-D`X`I`r%r)r r rr1r2rrrrrr3r[filterrKconstruct_static_kwarg_valuerL)rbr r!r:config_func_callsconfig_func_callunrendered_configrQs r#"statically_parse_unrendered_configrls $t 4C"66EY;Y%))&$7V%89 6"6??6<<+<+<=>  `  0A(+d !((K'CE'J%))$K r%ct|SN)str)rQs r#rhrhs u:r%rn)"typingrrrrrrdbt.artifacts.resourcesrdbt.exceptionsr r dbt_common.clients.jinjar dbt_common.exceptions.macrosr dbt_common.testsr dbt_extractorrr TYPE_CHECKINGdbt.context.providersrrro__annotations__boolr$rBr7rarlrhr%r#r|s 33 +E4@1A :(*d38n) S T VZ8 8 38n8 2:;Q2R8  #Y8 v@ cN@ 089O0P@  #Y@ F!s!uWd3i=O7P!H,s,xS#X7O,^3r%