i?>ddlZddlZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z mZmZmZddlmZddlmZddlmZddlmZmZddlmZdd lmZdd lmZm Z m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.ddl/m0Z0m1Z1m2Z2ddl3m4Z4ddl5m6Z6m7Z7m8Z8ddl9m:Z:ddl;mZ>ddl?m@Z@mAZAddlBmCZCmDZDddlEmFZFddlGmHZHmIZIddlJmKZKmLZLddlMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYddlZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlddlmmnZnddlompZpddlqmrZrmsZsddltmuZumvZvddlwmxZxdd lymzZzdd!l{m|Z|dd"l}m~Z~dd#lmZmZmZmZdd$lmZmZmZerddlZeZGd%d&ZGd'd(ZGd)d*ej+ZGd,d-eZGd.d/eZGd0d1eZGd2d3eZGd4d5eZGd6d7eZGd8d9eZGd:d;eZGd<d=eZGd>d?eZeeHeRfZGd@dAeZGdBdCeZGdDdEeZGdFdGeZGdHdIeZGdJdKeZGdLdMeZGdNdOeZGdPdQe6ZGdRdSeZGdTdUeZGdVdWeZGdXdYeZGdZd[eZGd\d]eZGd^d_eZGd`daeZGdbdceZGdddeeZGdfdgeZedhZGdidjeFZGdkdleZGdmdneZGdodpeZGdqdreZGdsdteZdueRdve2dweIdxedg}|dg}|St|tri|jjj |}|r|}|S|s:||jjj vr|jj |g}|Stdt|d|d)NzIn adapter.dispatch, got a z macro_namespace argument ("z3"), but macro_namespace should be None or a string.) isinstancestrrrkget_macro_search_order dependencies project_namerCr)rprsearch_packagesmacro_search_orders rr_get_search_packagesz(BaseDatabaseWrapper._get_search_packagess04v  #fO 3 '!%!5!5!L!LY!W !"4(I9M9M9Z9Z,Z#';;#;#;Y"G #-d9o->?KRT rt macro_namemacro_namespacepackagescd|vr,|jdd\}}d|d|d|d}t|| t||j|}g}|D]m} |j D]X} | d|} |j j | | } | |j| n|j| d| | T| ccSodjd |D} d |d |d | }t|#t$rd} YqwxYw) N.z*In adapter.dispatch, got a macro name of "zP", but "." is not a valid macro name component. Did you mean `adapter.dispatch("z", macro_namespace="z")`?__z, c32K|]}t|ywrj)repr).0as rr z/BaseDatabaseWrapper.dispatch..s7T!W7szIn dispatch: No macro named 'z' found within namespace: 'z' Searched for: ) splitrCrJrrrget_from_packageappendjoin)rprrrsuggest_macro_namespacesuggest_macro_namemsgrattempts package_nameprefix search_namemacrosearcheds rrdispatchzBaseDatabaseWrapper.dispatchsn * :D:J:J3PQ:R 7 #%7C?rjNN)rrr__doc__r0rsrxpropertyrkrrr rrr rrrrrtrrrrs $>$ G$$$8T#YhsmtHUXMGZ,*.(, .$.$"#.$49% .$  .$rtrceZdZdZedZedZedeefdZ de de fdZ de dee fdZej d e dee eeffd Zy ) BaseResolverc<||_||_||_||_yrj) db_wrappermodelrkmanifest)rprrrkrs rrrszBaseResolver.__init__s$    rtc.|jjSrj)rkrrs rrcurrent_projectzBaseResolver.current_projects{{'''rtc.|jjSrj)rrnrs rrrnzBaseResolver.Relations'''rtrcLt|jjddrdSdS)NEMPTYFr)rvrkr~rs rr resolve_limitzBaseResolver.resolve_limits"DKK,,gu=qG4Grttargetcd}d}d}t|jdr|jjyt|dr|jrt |jt r|jj D]\}}|j|jjk(s*d}tj|j|jr |jnd}t|dr|j |j}nNt|drBt|jd r,|jj|jj}n|sxt|drBt|jd r,|jj|jj}tj|jjd}|r| |jS|jjS) zIGet the event time field name with proper quoting based on configuration.FN event_timecolumnsTquoterlcolumn)hasattrrkrrrdictitemsrrr| data_typerrlrquoted)rpr should_quote column_foundr_ column_infos rr_resolve_event_time_field_namez+BaseResolver._resolve_event_time_field_names  v}}l3v}}7O7O7W 69 %&..ZX\=]"(.."6"6"8 ;##v}}'?'??#'L#]]#((;CXCX+*?*?^`F{G49J9J9V'2'8'8   2#FNNH="NN11='-~~'<'< # ( *FNNH5NN))5%~~44 ]]6==#;#;R@F F.== ==++ +rtcVd}t|jjdddu}|j|}t |jt t tfr|jjrt |jttfrt |jtr|jjjdk(r|jjjdk(r|jj|jj r|jj"|r:|jjj$j&|jj"j(kDr*|jjj$j&n|jj"j(}|jjj$j*|jj"j,kr*|jjj$j*n|jj"j,}t/|||}|St/||jj"j(|jj"j,}|S|r^t/||jjj$j&|jjj$j*}|S)Nsample incremental microbatch)r) field_namestartend)rvrkr~rrrrrrrr<r@ materializedincremental_strategyruse_microbatch_batchesrbatchrrevent_time_startrevent_time_endr)rprevent_time_filter sample_moderrrs rrresolve_event_time_filterz&BaseResolver.resolve_event_time_filter'sO dkk..$?tK 88@  v}}z:|&L M ((4:: <'@A 4::y1JJ%%22mCJJ%%::lJMM88dkkF^F^8_JJ$$0 ;;++22884::;K;K;\;\\ ((//55!ZZ-->> ;;++22669I9I9X9XX ((//33!ZZ--<< )8#-#)%,! )8#-"jj..?? JJ,,;;)%! $3)++**1177 ((//33%! ! rtr~cyrjrrpr~s rr__call__zBaseResolver.__call__b rtN)rrrrsrrrnr intrr;rrrrabcabstractmethodrrhr5rrrtrrrrs! ((((Hx}HH/,\/,c/,b9! 9!/AZ9!v  c eC,O&P  rtr) metaclassc eZdZej d dedeedeedefdZ dedeedeede fdZ dedeedeefdZ d edefd Z y) BaseRefResolverNrpackageversionrcyrjrrprrrs rrresolvezBaseRefResolver.resolvehsrtct|||S)Nrrr)rrs rr _repack_argszBaseRefResolver._repack_argsmswT7CCrtct|tstdt||'t|tstdt||3t|ttt fstdt|yy)Nz1The name argument to ref() must be a string, got z7D Y!^ MGTDJJT: :**Y':6::c? 4'2||D'733rtr)rrrrrrr rrhrrrrrrrtrrrrgsY]"*3-AI+AV DD"*3-D:B;:OD D # Q\H] 4c4 4rtrcZeZdZejdedefdZdedefdZdedefdZ y) BaseSourceResolver source_name table_namecyrjrrpr r s rrrzBaseSourceResolver.resolverrtct|tstdt|t|tstdt|y)NzCThe source name (first) argument to source() must be a string, got zCThe table name (second) argument to source() must be a string, got rrrCrrs rrrz BaseSourceResolver.validate_argssa+s+"#K013 *c*"#J/02 +rtr~rct|dk7rt||j|j|d|d|j |d|dS)Nrrrr)rrMrrrrs rrrzBaseSourceResolver.__call__sO t9>'4::> > 47DG,||DGT!W--rtN) rrrrrrrrrhrrrtrrr r sN 3 C    # .c.m.rtr ceZdZejd dedeedefdZdedeede efdZ dedeefdZ dedefd Z y) BaseMetricResolverNrrrcyrjrrprrs rrrzBaseMetricResolver.resolvesTWrtc||gS||gSrjrrs rrrzBaseMetricResolver._repack_args ?6MT? "rtct|tstdt||(t|tstdt|yy)Nz4The name argument to metric() must be a string, got z?The package argument to metric() must be a string or None, got rrs rrrz BaseMetricResolver.validate_argss`$$"FtDzlS   z'3'?"QRVW^R_Q`a (@ rtr~cd}t|dk(r|d}n+t|dk(r|\}}nt|j||j|||j ||SNrrrr)rrLrrrrpr~rrs rrrzBaseMetricResolver.__call__sa!% t9>7D Y!^ MGT!tzz= = 4)||D'**rtrj) rrrrrrr r5rr rrrrrtrrrrsuWCW(3-W?WW##x}#c# #   +c +o +rtrceZdZejd dedeefdZdedeedeefdZ dedeefdZ defd Z y) BaseFunctionResolverNrrcyrjrrs rrrzBaseFunctionResolver.resolvesADrtrc||gS||gSrjrrs rrrz!BaseFunctionResolver._repack_argsrrtct|tstdt||(t|tstdt|yy)Nz6The name argument to function() must be a string, got zAThe package argument to function() must be a string or None, got rrs rrrz"BaseFunctionResolver.validate_argss`$$"Hd U   z'3'?"STXY`TaSbc (@ rtr~cd}t|dk(r|d}n+t|dk(r|\}}nt|j||j|||j ||Sr)rrQrrrrs rrrzBaseFunctionResolver.__call__sa!% t9>7D Y!^ MGTDJJT: : 4)||D'**rtrj) rrrrrrr rr rrrrrtrrrrsiDCD(3-DD##x}#c# #   +c +rtrceZdZdeefdZy)Configcontext_configcyrjrrprr%s rrrszConfig.__init__srtN)rrrr r,rsrrtrrr$r$sKh}.EKrtr$cheZdZdeefdZdZdZdZddZ ddZ dd Z dd Z d e fd Zd e fd Zy)ParseConfigObjectr%c ||_||_yrj)rr%r's rrrszParseConfigObject.__init__s ,rtcdD]I}||vs|jdd}||vrt|||j|j|||<K|S)N)pre_hook post_hookr-r)replacerDrr{)rprkoldkeynewkeys rr_transform_configz#ParseConfigObject._transform_configs]/ 4FS1V#4VV$**UU!'F!3v  4  rtct|dk(rt|dk(r|d}n5t|dk(rt|dkDr|}nt|j|j|}|j t dt jrc&kQ.7D Y!^F aD(djj9 9%%d+    &!"ST T ; D D B4::CVCV W  ##>>?PQ ++D1rtc(|j||iSrjrrprvalues rrsetzParseConfigObject.set}}dE]++rtNcyNrrrpr validators rrrequirezParseConfigObject.require rtcyrArrBs rr meta_requirezParseConfigObject.meta_require#rErtcyrArrprrrCs rrrzParseConfigObject.get&rErtcyrArrIs rrmeta_getzParseConfigObject.meta_get)rErtrcyNFrrs rrpersist_relation_docsz'ParseConfigObject.persist_relation_docs,rtcyrMrrs rrpersist_column_docsz%ParseConfigObject.persist_column_docs/rOrtrjr)rrrr r,rsr2rr>rDrGrrKboolrNrQrrtrrr)r)sN-h}.E-2,tTrtr)c~eZdZddeefdZdZdZdZe fdZ e fdZ dd Z dd Z dd Zdd Zd efdZd efdZy)RuntimeConfigObjectNr%c||_yrjrr's rrrszRuntimeConfigObject.__init__4s  rtcyrArr}s rrrzRuntimeConfigObject.__call__8rErtc(|j||iSrjr;r<s rrr>zRuntimeConfigObject.set;r?rtc||yrjr)rprCr=s rr _validatezRuntimeConfigObject._validate>s %rtct|jds|}n&|jjj||}|tur!t |jj ||SNrk) unique_idr)rrrkr_MISSINGrr]rprrresults rr_lookupzRuntimeConfigObject._lookupAsUtzz8,FZZ&&**49F X $tzz/C/C$O O rtct|jds|}n&|jjj||}|tur!t |jj ||Sr\)rrrkrKr^rr]r_s rr _lookup_metaz RuntimeConfigObject._lookup_metaKsUtzz8,FZZ&&//g>F X $tzz/C/C$O O rtcP|j|}||j|||SrjrarZrprrC to_returns rrrDzRuntimeConfigObject.requireUs*LL&  NN9i 0rtcP|j|}||j|||SrjrcrZrfs rrrGz RuntimeConfigObject.meta_require]s,%%d+  NN9i 0rtcV|j||}|||j|||SrjrerprrrCrgs rrrzRuntimeConfigObject.getes1LLw/  W%8 NN9i 0rtcV|j||}|||j|||Srjrirks rrrKzRuntimeConfigObject.meta_getms3%%dG4  W%8 NN9i 0rtrc|jdi}t|ts t||jddS)N persist_docsrrelationFrrrrPrprns rrrNz)RuntimeConfigObject.persist_relation_docsus<xxx; ,-+L9 9 E22rtc|jdi}t|ts t||jddS)NrnrorFrqrrs rrrQz'RuntimeConfigObject.persist_column_docs|s<xxx; ,-+L9 9 511rtrjr)rrrr r,rsrr>rZr^rarcrDrGrrKrRrNrQrrtrrrTrT3sbh}.E,%-*23t32T2rtrTceZdZdZdZy)ParseDatabaseWrapperz_The parser subclass of the database wrapper applies any explicit parse-time overrides. cV||jjvxr||jjv}|r|jj|S||jjvrt|j|St dj |j j|Nz!'{}' object has no attribute '{}')r _available__parse_replacements_rvAttributeErrorformat __class__r)rproverrides rrrxz ParseDatabaseWrapper.__getattr__s4==444cAcAc9c ==55d; ; T]].. .4==$/ / 3::4>>;R;RTXY rtNrrrrrxrrtrrrurus  rtruceZdZdZdZy)RuntimeDatabaseWrapperzUThe runtime database wrapper exposes everything the adapter marks available. c||jjvrt|j|Stdj |j j |rw)rrxrvrzr{r|rrs rrrxz"RuntimeDatabaseWrapper.__getattr__sM 4==,, ,4==$/ / 3::4>>;R;RTXY rtNr~rrtrrrrs rtrc 4eZdZ ddedeedeedefdZy)ParseRefResolverNrrrrc|jjj|j||||jj |j |jSrj)rrefsrrrn create_fromrkrs rrrzParseRefResolver.resolvesJ t00wHI}}((djjAArtr)rrrrr rrhrrrtrrrrs8Y]BB"*3-BAI+AVB Brtrc neZdZ d dedeedeedefdZdedefdZ d ededeedeeddf d Z y) RuntimeRefResolverN target_nametarget_packagetarget_versionrc d|jj|j||||j|jj}|t |t r)t|j|d||t |t |jj|j||jjrNt|jj|jtjt|j |jj#|j||jjrEt|jj|jtj$|j|j'|||||j)|S)Nr)rr target_kindrrdisabled)r] ref_unique_idaccessscope)r resolve_refrrrrr3rTis_invalid_private_refrkrrEr]r7Privateregroupis_invalid_protected_ref Protectedvalidatecreate_relation)rprrr target_models rrrzRuntimeRefResolver.resolvesc }}00 JJ     JJ # #    :lH#E%ZZ'"--#L(;  ]] 1 1 JJ dkk&>&> $**..*44!))!,"4"45  ]] 3 3 JJ dkk&>&> $**..*44!++"//   lKP##L11rtrc`|jr]|jj|jd|jj ||j |j|St|dr|jr|jjjr|jjjr|jtjvsDt!|jj#|j$|j&|j(sL|jj+|j|j|j |j|S|jj+|j||j |j|S)Nlimitrdefer_relation)is_ephemeral_modelrset_cter]rncreate_ephemeral_fromrrrrrkr~defer favor_staterSELECTED_RESOURCESr get_relationdatabaseschema identifierr)rprs rrrz"RuntimeRefResolver.create_relationsd  * * JJ  |55t <==66(("&"@"@"N7  L"2 3++   &&KK$$00$..6H6[6[[#4;;/<< ))<+>+> @W@W ==,, ++(("&"@"@"N - ==,, (("&"@"@"N - rtresolvedc|j|jjjvr*|j |||}t |j|y)Nr)r]r depends_onnodesrrR)rprrrrr~s rrrzRuntimeRefResolver.validate sM   TZZ%:%:%@%@ @$$[..QD$$**4@ @ Artr) rrrrr rrhrr;rrrrtrrrrs)-04 /2/2! /2!- /2  /2b$L$]$L A A A! A !- A  Artrc PeZdZdededeedeeddf dZdedeffd Z xZ S) OperationRefResolverrrrrrNcyrjr)rprrrrs rrrzOperationRefResolver.validates rtrc||jr!t|j|jt||SNr)rrNrrsuperr)rprr|s rrrz$OperationRefResolver.create_relation!s8  * *99J9JQUQ[Q[\ \7*<8 8rt) rrrr;rr rrrhr __classcell__r|s@rrrrsY   !  !-    9L9]99rtrc ^eZdZedeefdZ ddedeedeede ffd Z xZ S)RuntimeUnitTestRefResolverrcyrjrrs rrrz(RuntimeUnitTestRefResolver.resolve_limit+rtrrrc&t||||Srj)rr)rprrrr|s rrrz"RuntimeUnitTestRefResolver.resolve0s w{NNKKrtr) rrrrr rrrrrhrrrs@rrrr*sg x})-04 LL! L!- L  LLrtrceZdZdedefdZy)ParseSourceResolverr r c|jjj||g|jj |j |jSrj)rsourcesrrnrrkrs rrrzParseSourceResolver.resolve;s> !!; ";<}}((djjAArtNrrrrrrrtrrrr:sB3BCBrtrceZdZdedefdZy)RuntimeSourceResolverr r c |jj|||j|jj}|t |t r,t|j|d|dt |t Gdd}|jj|||j|j|S)Nrsourcerrrrc&eZdZUiZeeefed<y)>RuntimeSourceResolver.resolve..SourceQuotingBaseConfigrlNrrrrlrrr__annotations__rrtrrSourceQuotingBaseConfigrU&(GT#s(^ (rtrr) rresolve_sourcerrrrr3rTrnrrr)rpr r  target_sourcers rrrzRuntimeSourceResolver.resolveBs 44    JJ # #    J}h$G%ZZ*m1ZL9$$]H=   ) )}}(( # % $$"<<]K )  rtNrrrtrrrrAs 3 C rtrc8eZdZedeefdZdedefdZy)RuntimeUnitTestSourceResolverrcyrjrrs rrrz+RuntimeUnitTestSourceResolver.resolve_limitarrtr r c |jj|||j|jj}|t |t r,t|j|d|dt |t |jj|jd|jj|j}|jj|jj|jdS)Nrrr)rrF)r)rrrrrrr3rTrr]rnadd_ephemeral_prefixcte_namer|CTEr)rpr r rrs rrrz%RuntimeUnitTestSourceResolver.resolvefs 44    JJ # #   J}h$G%ZZ*m1ZL9$$]H=   =22D9]]77 8N8NO }}##""!$  %5% ! "rtN) rrrrr rrrrrrtrrrr`s2 x}"3"C"rtrc(eZdZddedeedefdZy)ParseMetricResolverNrrrc|jjj|j||t ||Srj)rmetricsrrr5rs rrrzParseMetricResolver.resolves4 !!$"3"3D'"BCtW--rtrjrrrrr r5rrrtrrrrs .C.(3-.?.rtrc(eZdZddedeedefdZy)RuntimeMetricResolverNrrrc|jj|||j|jj}|t |t rt|j|d|t||jS)Nmetric)rrrr) rresolve_metricrrrrr3rTr6)rprr target_metrics rrrzRuntimeMetricResolver.resolvess 44    JJ # #    J}h$G%ZZ'$-  '}dmmDDrtrjrrrtrrrrs%E3E EQ`ErtrcbeZdZdeeefdededdffd Zde e fdZ de eeffdZ xZS) ModelConfiguredVarcontextrkrrNcP|||_t| ||j|yr)_configrrscli_vars)rprrkrr|s rrrszModelConfiguredVar.__init__s' &,  &//=rtc#K|jj}|jj}||jjk7r ||vr|||jywrj)rload_dependencies_noderr)rprrs rrpackages_for_nodez$ModelConfiguredVar.packages_for_nodesV||557 zz.. 4<<44 4|+"<00llsA$A&ct|jtr |j}nt|jj}|j j j}t}|jD]-}|j|jj||/|j|j|Srj)rrr$r+rr credentialsrrYraddvarsvars_for _cli_vars)rp search_node adapter_typemergedprojects rr_generate_mergedz#ModelConfiguredVar._generate_mergeds djj- 0**K#DJJ$;$;>" rt)rrrrrrr&r=rsrr%rr rrrs@rrrrs\>c3h>> >  >8G#4 '#s("3 rtrceZdZdZy)ParseVarcyrjr)rpvar_names rrget_missing_varzParseVar.get_missing_varrtN)rrrrrrtrrrrsrtrc eZdZy) RuntimeVarN)rrrrrtrrrrsrtrc:eZdZdeeefdededdffd ZxZ S) UnitTestVarrrkrrNcd}t|tsJ|jrP|jjr:t |}|j j |jjt|!||xs||yr) rrB overridesrrrupdaterrs)rprrkr config_copyr|s rrrszUnitTestVar.__init__sn  $ --- >>dnn11"6*K  ' '(;(; < +"7dCrt) rrrrrrr&r=rsrrs@rrrrsA Dc3h D D D  D Drtrc$eZdZddedeefdZy)ParseFunctionResolverNrrc|jjj|j|||jj |j |jtjS)N)r) r functionsrrrnrrkrFunctionrs rrrzParseFunctionResolver.resolvesQ ##D$5$5dG$DE}}((djj|G\G\(]]rtrjrrrrr rrrtrrrrs^C^(3-^rtrc$eZdZddedeefdZy)RuntimeFunctionResolverNrrc |jj|||j|jj}|t |t r't|j|dt |t Gdd}|jj|||j|j|tjS)Nfunctionrc&eZdZUiZeeefed<y)@RuntimeFunctionResolver.resolve..SourceQuotingBaseConfigrlNrrrtrrrr rrtr)rrr)rresolve_functionrrrrr3rTrnrrrrr)rprrtarget_functionrs rrrzRuntimeFunctionResolver.resolves--88    JJ # #    "j(&K%ZZ &$_h?   ) )}}(( # % $$"<<_M&& )  rtrjrrrtrrrrs C (3- rtrceZdZUeed<eeed<eeed<eeed<ee ed<ee ed<ee ed<ee ed<y ) Providerexecuter$DatabaseWrapperr(refrrr N) rrrrRrr r$rrrr rrrrtrrrr sX M L-..  !! o  # $$ # $$'((rtrc,eZdZdZeZeZeZ e Z e Z eZeZy) ParseProviderFN)rrrrr)r$rurrr(rrrrrrrr rrtrrrrs*G F*O C C F F$Hrtrc,eZdZdZeZeZeZ e Z e Z eZeZy)GenerateNameProviderFN)rrrrrTr$rurrr(rrrrrrrr rrtrrrr s*G F*O C C F F$Hrtrc,eZdZdZeZeZeZ e Z e Z eZeZy)RuntimeProviderTN)rrrrrTr$rrrr(rrrrrrrr rrtrrrr+s*G F,O C C "F "F&Hrtrc,eZdZdZeZeZeZ e Z e Z eZeZy)RuntimeUnitTestProviderTN)rrrrrTr$rrrr(rrrrrrrr rrtrrrr6s*G F,O C $C *F "F&HrtrceZdZeZy)OperationProviderN)rrrrrrrtrrrrAs CrtrTc eZdZdedededeeddf fd ZdZ e de e e ffd Z e d Ze de e eeffd Zed e deefd Ze d:d e dededde fdZeee ee ee dfd e dedde fdZe dZede de fdZede de fdZede dedefdZed;dZe defdZe defdZe defdZe defdZe dde fd Z!e de"fd!Z#e de e effd"Z$e de fd#Z%e de fd$Z&e de'fd%Z(e d&de)fd'Z*e de e effd(Z+e de,e-fd)Z.e de e effd*Z/e de e effd+Z0e d,de e effd-Z1e dee2e e effd.Z3e dee2e e effd/Z4e dee fd0Z5e de fd1Z6ed e fd2Z7ed:d3e d4ee de fd5Z8e de2e fd6Z9ed7e d8e de:fd9Z;xZ:<7E"* "4;;/ --77 dnnUrtct|jjj}t |jj |j |j||jSrj) rrkrrr1rsearch_package macro_stackr)rpinternal_packagess rr_get_namespace_builderz&ProviderContext._get_namespace_builderbsS5dkk6M6M6R6RS$ KK $ $        JJ   rtctSrjr^rs rrdbt_metadata_envsz!ProviderContext.dbt_metadata_envsls  ""rtc@t|jjSrj)rZrkr~rs rrinvocation_args_dictz$ProviderContext.invocation_args_dictpsDKK,,--rtc|jSrj)r#rs rr _sql_resultszProviderContext._sql_resultstsrtrc||jvrN|dk(r|jdS|j| t||j|}d|j|<|Sy)Nmain)r#rK)rprret_vals rr load_resultzProviderContext.load_resultxst 4## #v~''//!!$'/3D99**40)-  &rtresponse agate_table agate.Tablecddlm}||j}t||j ||d|j |<y)Nr agate_helper)r3datatabler)dbt_common.clientsr8 empty_tablerd as_matrixr#)rprr3r4r8s rr store_resultzProviderContext.store_resultsN 4  &224K!)$$..{;$ " rtcDt|||}|j|||S)N)_messagecode rows_affected)rr>)rprmessagerArBr4r3s rrstore_raw_resultz ProviderContext.store_raw_results'#G$m\  x==rtcDdttgdffd}td|iS)Nrc$dtddffd }|S)Nr=rc D]'}t|trt||ry||k(s'ytdj|dj t t )Nz#Expected value "{}" to be one of {},)rrrbr{rmapr)r=argr~s rrinnerz?ProviderContext.validation..validate_any..innersdC!#t,E31G#  )9@@QTUXZ^Q_H`art)r)r~rKs` rr validate_anyz0ProviderContext.validation..validate_anys Q 4 Lrtany)rrrd)rprLs rr validationzProviderContext.validations2 8QCI#6 |   rtpayloadct|jttfrt |jd}t|jt r|jj jddk(rh|jjRtj|jjj|jj j}|jj|j jd||j_|jj!|j j"|jj|y)Nrrrrun) split_suffixr)rrr:rArcr<rkrrrVformat_batch_startr batch_sizeget_target_write_path target_path build_path write_node project_root)rprOrRs rrwritezProviderContext.writes djj5*:"; </TZZ@ @ tzz9 - !!%%&<=M   ,,??   11 !!,,L !% @ @ KK # #U!A!   dkk66 8M8MwWrtstringcDt||j|jSrj)r"_ctxr)rpr[s rrrenderzProviderContext.rendersFDIItzz::rtmessage_if_exceptionfunccZ ||i|S#t$rt||jwxYwrj) ExceptionrCr)rpr_r`r~rs rrtry_or_compiler_errorz%ProviderContext.try_or_compiler_errors; E(( ( E"#7D D Es *c\ddlm}t|jts+t |jj |j|jj|jjk7rHtjj|jj|jjnd}tjj|jj||jj}tjj!|s`|jj"sJtjj|jj"|jj}|jjj$}|jjj&} |j)|||}tjj/||_|S#t*$r}t-||jd}~wwxYw)Nrr7rr) text_columns delimiter)r;r8rrr>rH resource_typerrkrospathrpackages_install_pathrYoriginal_file_pathexists root_path column_typesrffrom_csv ValueErrorrIabspathoriginal_abspath)rpr8 package_pathrirnrfr:es rrload_agate_tablez ProviderContext.load_agate_tablesy3$**h/,TZZ-E-EDJJW W zz&&$++*B*BB GGLL::DJJ > ?s H H+H&&H+c|jj|j|j|j|j S)aKThe most important function in dbt is `ref()`; it's impossible to build even moderately complex models without it. `ref()` is how you reference one model within another. This is a very common behavior, as typically models are built to be "stacked" on top of one another. Here is how this looks in practice: > model_a.sql: select * from public.raw_data > model_b.sql: select * from {{ref('model_a')}} `ref()` is, under the hood, actually doing two important things. First, it is interpolating the schema into your model file to allow you to change your deployment schema via configuration. Second, it is using these references between models to automatically build the dependency graph. This will enable dbt to deploy models in the correct order when using dbt run. The `ref` function returns a Relation object. ## Advanced ref usage There is also a two-argument variant of the `ref` function. With this variant, you can pass both a package name and model name to `ref` to avoid ambiguity. This functionality is not commonly required for typical dbt usage. > model.sql: select * from {{ ref('package_name', 'model_name') }}" )r!rrrrkrrs rrrzProviderContext.refs0N}}  $**dkk4==YYrtc|jj|j|j|j|j Srj)r!rrrrkrrs rrrzProviderContext.source)-}}##DOOTZZdmm\\rtc|jj|j|j|j|j Srj)r!rrrrkrrs rrrzProviderContext.metric-rxrtc|jj|j|j|j|j Srj)r!r rrrkrrs rrr zProviderContext.function1s0}}%%dootzz4;;PTP]P]^^rtcb|jj|j|jS)aThe `config` variable exists to handle end-user configuration for custom materializations. Configs like `unique_key` can be implemented using the `config` variable in your own materializations. For example, code in the `incremental` materialization like this: {% materialization incremental, default -%} {%- set unique_key = config.get('unique_key') -%} ... is responsible for handling model code that looks like this: {{ config( materialized='incremental', unique_key='id' ) }} ## config.get name: The name of the configuration variable (required) default: The default value to use if this configuration is not provided (optional) The `config.get` function is used to get configurations for a model from the end-user. Configs defined in this way are optional, and a default value can be provided. Example usage: {% materialization incremental, default -%} -- Example w/ no default. unique_key will be None if the user does not provide this configuration {%- set unique_key = config.get('unique_key') -%} -- Example w/ default value. Default to 'id' if 'unique_key' not provided {%- set unique_key = config.get('unique_key', default='id') -%} ... ## config.require name: The name of the configuration variable (required) The `config.require` function is used to get configurations for a model from the end-user. Configs defined using this function are required, and failure to provide them will result in a compilation error. Example usage: {% materialization incremental, default -%} {%- set unique_key = config.require('unique_key') -%} ... )r!r$rr%rs rr ctx_configzProviderContext.ctx_config5s&n}}##DJJ0C0CDDrtc.|jjS)a`execute` is a Jinja variable that returns True when dbt is in "execute" mode. When you execute a dbt compile or dbt run command, dbt: - Reads all of the files in your project and generates a "manifest" comprised of models, tests, and other graph nodes present in your project. During this phase, dbt uses the `ref` statements it finds to generate the DAG for your project. *No SQL is run during this phase*, and `execute == False`. - Compiles (and runs) each node (eg. building models, or running tests). SQL is run during this phase, and `execute == True`. Any Jinja that relies on a result being returned from the database will error during the parse phase. For example, this SQL will return an error: > models/order_payment_methods.sql: {% set payment_method_query %} select distinct payment_method from {{ ref('raw_payments') }} order by 1 {% endset %} {% set results = run_query(relation_query) %} {# Return the first column #} {% set payment_methods = results.columns[0].values() %} The error returned by dbt will look as follows: Encountered an error: Compilation Error in model order_payment_methods (models/order_payment_methods.sql) 'None' has no attribute 'table' This is because Line #11 assumes that a table has been returned, when, during the parse phase, this query hasn't been run. To work around this, wrap any problematic Jinja in an `{% if execute %}` statement: > models/order_payment_methods.sql: {% set payment_method_query %} select distinct payment_method from {{ ref('raw_payments') }} order by 1 {% endset %} {% set results = run_query(relation_query) %} {% if execute %} {# Return the first column #} {% set payment_methods = results.columns[0].values() %} {% else %} {% set payment_methods = [] %} {% endif %} )r!rrs rrrzProviderContext.executensv}}$$$rtc,t|jS)auThe exceptions namespace can be used to raise warnings and errors in dbt userspace. ## raise_compiler_error The `exceptions.raise_compiler_error` method will raise a compiler error with the provided message. This is typically only useful in macros or materializations when invalid arguments are provided by the calling model. Note that throwing an exception will cause a model to fail, so please use this variable with care! Example usage: > exceptions.sql: {% if number < 0 or number > 100 %} {{ exceptions.raise_compiler_error("Invalid `number`. Got: " ~ number) }} {% endif %} ## warn The `exceptions.warn` method will raise a compiler warning with the provided message. If the `--warn-error` flag is provided to dbt, then this warning will be elevated to an exception, which is raised. Example usage: > warn.sql: {% if number < 0 or number > 100 %} {% do exceptions.warn("Invalid `number`. Got: " ~ number) %} {% endif %} )r-rrs rr exceptionszProviderContext.exceptionssHtzz**rtcB|jjjSrj)rkrrrs rrrzProviderContext.databases{{&&///rtcB|jjjSrj)rkrrrs rrrzProviderContext.schemas{{&&---rtcz|jj|j|j|jS)N)rrkr)r!r(r]rkrrs rrvarzProviderContext.vars3}}  II;;!  rtrqc|jS)a(`adapter` is a wrapper around the internal database adapter used by dbt. It allows users to make calls to the database in their dbt models. The adapter methods will be translated into specific SQL statements depending on the type of adapter your project is using. )rrs rr ctx_adapterzProviderContext.ctx_adaptersrtc\|jj|jjdS)N)rnr)rrnrqrrs rrapizProviderContext.apis(00ll))  rtc.|jjSrj)rqrrs rrrzProviderContext.columns||"""rtc|jSrj)rrs rrenvzProviderContext.envs {{rtc.|jjS)aThe `graph` context variable contains information about the nodes in your dbt project. Models, sources, tests, and snapshots are all examples of nodes in dbt projects. ## The graph context variable The graph context variable is a dictionary which maps node ids onto dictionary representations of those nodes. A simplified example might look like: { "model.project_name.model_name": { "config": {"materialzed": "table", "sort": "id"}, "tags": ["abc", "123"], "path": "models/path/to/model_name.sql", ... }, "source.project_name.source_name": { "path": "models/path/to/schema.yml", "columns": { "id": { .... }, "first_name": { .... }, }, ... } } The exact contract for these model and source nodes is not currently documented, but that will change in the future. ## Accessing models The `model` entries in the `graph` dictionary will be incomplete or incorrect during parsing. If accessing the models in your project via the `graph` variable, be sure to use the `execute` flag to ensure that this code only executes at run-time and not at parse-time. Do not use the `graph` variable to build you DAG, as the resulting dbt behavior will be undefined and likely incorrect. Example usage: > graph-usage.sql: /* Print information about all of the models in the Snowplow package */ {% if execute %} {% for node in graph.nodes.values() | selectattr("resource_type", "equalto", "model") | selectattr("package_name", "equalto", "snowplow") %} {% do log(node.unique_id ~ ", materialized: " ~ node.config.materialized, info=true) %} {% endfor %} {% endif %} /* Example output --------------------------------------------------------------- model.snowplow.snowplow_id_map, materialized: incremental model.snowplow.snowplow_page_views, materialized: incremental model.snowplow.snowplow_web_events, materialized: incremental model.snowplow.snowplow_web_page_context, materialized: table model.snowplow.snowplow_web_events_scroll_depth, materialized: incremental model.snowplow.snowplow_web_events_time, materialized: incremental model.snowplow.snowplow_web_events_internal_fixed, materialized: ephemeral model.snowplow.snowplow_base_web_page_context, materialized: ephemeral model.snowplow.snowplow_base_events, materialized: ephemeral model.snowplow.snowplow_sessions_tmp, materialized: incremental model.snowplow.snowplow_sessions, materialized: table */ ## Accessing sources To access the sources in your dbt project programatically, use the "sources" attribute. Example usage: > models/events_unioned.sql /* Union all of the Snowplow sources defined in the project which begin with the string "event_" */ {% set sources = [] -%} {% for node in graph.sources.values() -%} {%- if node.name.startswith('event_') and node.source_name == 'snowplow' -%} {%- do sources.append(source(node.source_name, node.name)) -%} {%- endif -%} {%- endfor %} select * from ( {%- for source in sources %} {{ source }} {% if not loop.last %} union all {% endif %} {% endfor %} ) /* Example compiled SQL --------------------------------------------------------------- select * from ( select * from raw.snowplow.event_add_to_cart union all select * from raw.snowplow.event_remove_from_cart union all select * from raw.snowplow.event_checkout ) */ )r flat_graphrs rrgraphzProviderContext.graphsR}}'''rtrcf|jjd}d|vr|d|d<t|jdrr|jjjdurPd|vrL|dj D]6}d|vs|d}|j jj|}||d<8|S)NT omit_none compiled_code compiled_sqlcontractrr) rto_dictrr alias_typesvaluesrqrtranslate_type)rp model_dctrorig_data_type new_data_types rr ctx_modelzProviderContext.ctx_modeldsJJ&&&6  i '(1/(BIn % DJJ + ##//47Y&#I.557 8&(%+K%8N$(LL$7$7$F$F~$VM*7F;'  8 rtcyrjrrs rr pre_hookszProviderContext.pre_hooksyrrtcyrjrrs rr post_hookszProviderContext.post_hooks}rrtcyrjrrs rrsqlzProviderContext.sqlrrtc6|jjSrj)rq date_functionrs rrsql_nowzProviderContext.sql_nows||))++rtc<dj|}t|)z9This was deprecated in v0.18 in favor of adapter.dispatchaThe "adapter_macro" macro has been deprecated. Instead, use the `adapter.dispatch` method to find a macro and call the result. For more information, see: https://docs.getdbt.com/reference/dbt-jinja-functions/dispatch) adapter_macro was called for: {macro_name})r)r{rC)rprr~rrs rr adapter_macrozProviderContext.adapter_macros( :;A&D&:Q s##rtrrcvd}|jtr t|tj}||vr||}n||}|t |j drt|j dddurdnd}|j r|s||vr|nt|jj|<|j j|jjvrW|jj|j j}|jdk7r|jj||St|)zThe env_var() function. Return the environment variable named 'var'. If there is no such environment variable set, return the default. If the default is None, raise an exception for an undefined variable. NcompiledFTr) startswithr\rSr]rrrrvr'renv_varsfile_idfilesparse_file_typerrF)rprr return_valuer compiling source_files rrenv_varzProviderContext.env_vars  >>+ ,+C0 0$&** #:s8L  "L  # 4::z2DJJ E:dB  zz) %(3JL4K &&s+ ::%%)<)<<"&--"5"5djj6H6H"IK"22h>#,,33C8 $S) )rtc"tjS)zThe `selected_resources` variable contains a list of the resources selected based on the parameters provided to the dbt command. Currently, is not populated for the command `run-operation` that doesn't support `--select`. )rrrs rrrz"ProviderContext.selected_resourcess"444rt parsed_modelrc,|jjdk(r7|jjddk(rd|jjdvs)td|dd|jjd |jj ||S) Nrrzmacro.dbt.statementmaterializationrz>submit_python_job is not intended to be called here, at model aliasz, with macro call_stack r)context_macro_stackdepth call_stackrarqsubmit_python_job)rprrs rrrz!ProviderContext.submit_python_jobs  $ $ * *a /((33A6:OO!T%=%=%H%H%KK!PQ]^eQfPggAEAYAYAdAd@eefg ||--lMJJrtrj)rr5)=rrrr&r4rr r,rsr(r*rrr*r,rdr.r)r2rr>rDrNrZr^rrcrurrrr r$r|rRrrrrrrrrrr rrrrrr rrrrrrrrrrrs@rrr r Is~VV V  V !/ V V. #4S>##.. d3(:#:;  _(:(_OS#&5=m5L "_ c]sm/3 > > m, >  > >  (_SS,_;S;S;;_E$'E/7E EE_6&ZX&Z&ZP]]]]]]_(__X6EF6E6Ep:%:%:%x#+DcN#+#+J0#00... '  Y0  T#s(^  #V ##T#s(^h(tCH~h(h(TW4S>(8Dc3h$89HT$sCx.%9:Xc],,,_ $# $ $_,*3,*#,*#,*,*\5DI55_ Kd K3 K? K Krtr c BeZdZdZdededededee ddf fd Z xZ S) MacroContexta7Internally, macros can be executed like nodes, with some restrictions: - they don't have all values available that nodes do: - 'this', 'pre_hooks', 'post_hooks', and 'sql' are missing - 'schema' does not use any 'model' information - they can't be configured with config() directives rrkrr!r%rNcbt|||||d||j|_y||_yrj)rrsr_search_package)rprrkrr!r%r|s rrrszMacroContext.__init__s6 (DA  !#)#6#6D #1D rt) rrrrr[r&r4rr rrsrrs@rrrrsP222 2  2 ! 2 22rtrc^eZdZUeed<edeefdZedefdZ y) SourceContextrrcv|jjj|j|jSrjrrnrrkrrs rrthiszSourceContext.this'''33DKKLLrtc|jSrjrVrs rr source_nodezSourceContext.source_nodes zzrtN) rrrrArr*r rhrrrrtrrrrsM Mh}-MM-rtrcPeZdZUeed<edeeee ffdZ edeeee ffdZ ede efdZ ede efdZedefdZedefdZede efd Zede efd Zy ) ModelContextrrc|jjtjtjtj fvrgS|jj jDcgc]}|jdc}Scc}wNTr) rrgrXSourceTestUnitrkr,rrphs rrrzModelContext.pre_hookssf :: # # 'V VI04zz/@/@/I/I *+AIII %   *Bc|jjtjtjtj fvrgS|jj jDcgc]}|jdc}Scc}wr) rrgrXrrrrkr-rrs rrrzModelContext.post_hookssf :: # # 'V VI04zz/@/@/J/J *+AIII %   rcFt|jddr]|jjjdk(r:d|jj j xst|j St|jddr|jjSy)Nrclonezselect * from extra_ctes_injected) rvrrkr~whichr relation_namerrrs rrrzModelContext.compiled_codes 4::/ 64;;;K;K;Q;QU\;\#DJJ$=$=$K$K$gsSWSfSfOg#hi i TZZ!6 =::++ +rtcj|jjtjk(r |jSyrj)rlanguagerWrrrs rrrzModelContext.sql$s+ ::  -"3"3 3%% %rtclt|jd|jjjS)Nr)rvrrkrrrs rrrzModelContext.database,s%tzz:t{{/F/F/O/OPPrtclt|jd|jjjS)Nr)rvrrkrrrs rrrzModelContext.schema0s%tzz8T[[-D-D-K-KLLrtc|jjtjk(ry|jj j |j|jS)a`this` makes available schema information about the currently executing model. It's is useful in any context in which you need to write code that references the current model, for example when defining a `sql_where` clause for an incremental model and for writing pre- and post-model hooks that operate on the model in some way. Developers have options for how to use `this`: |------------------|------------------| | dbt Model Syntax | Output | |------------------|------------------| | {{this}} | "schema"."table" | |------------------|------------------| | {{this.schema}} | schema | |------------------|------------------| | {{this.table}} | table | |------------------|------------------| | {{this.name}} | table | |------------------|------------------| Here's an example of how to use `this` in `dbt_project.yml` to grant select rights on a table to a different db user. > example.yml: models: project-name: post-hook: - "grant select on {{ this }} to db_reader" N)rrgrX Operationrrnrrkrs rrrzModelContext.this4sD> :: # #x'9'9 9''33DKKLLrtct|jddrD|jjj |j |jj Sy)z For commands which add information about this node's corresponding production version (via a --state artifact), access the Relation object for that stateful other rN)rvrrrnrrkrrs rrrzModelContext.defer_relationWsJ 4::/ 6??++77 TZZ66 rtN)rrrr;rr*r rrrrrr rrrrrhrrrrtrrrrs,  4S#X/   Dc3h0   x}  Xc]Q#QQMMM Mh}- M MD  7  rtrczeZdZUeed<eddedeedeffd Ze deefdZ xZ S)UnitTestContextrrrrc|jjrE||jjjvr#|jjj|St|||S)aTThe env_var() function. Return the overriden unit test environment variable named 'var'. If there is no unit test override, return the environment variable named 'var'. If there is no such environment variable set, return the default. If the default is None, raise an exception for an undefined variable. )rrrrr)rprrr|s rrrzUnitTestContext.env_varisU ::  C4::+?+?+H+H$H::''005 57?30 0rtc8|jjr|jj|jj}|jj |j d|j jj|jSyrj) rthis_input_node_unique_idrexpectrr]rqrnrr)rp this_nodes rrrzUnitTestContext.thisxsk :: / / ,,TZZ-Q-QRI JJ  y22D 9<<((==i>R>RS Srtrj) rrrrBrr)rr rr*rrrs@rrrrfsX _ 13 1# 1# 1 1hsmrtrc>eZdZUeed<edeefdZy)FunctionContextrrcv|jjj|j|jSrjrrs rrrzFunctionContext.thisrrtN) rrrr9rr*r rhrrrtrrrrs, Mh}-MMrtrrrkrr%rcPt|||t|}|jSrj)rrr)rrkrr%ctxs rrgenerate_parser_model_contextrs& ufh  PC ;;=rt unit_testct||jtj|j}t |||t |}|jSrj)r,fqnrXrrrrr)rrkrr%rs rr!generate_parser_unit_test_contextrsJ#   N )VX} WC ;;=rtrcPt|||td}|jSrj)rrr)rrkrrs rr$generate_generate_name_macro_contextrs& ufh0D0F MC ;;=rtcPt|||td}|jSrj)rrr)rrkrrs rrgenerate_runtime_model_contextrs% ufh0A4 HC ;;=rtrcPt|||t|}|jSrj)rrr)rrkrrrs rrgenerate_runtime_macro_contextrs& ufh0A0C\ RC ;;=rtct|||td}|j}|jrB|jjr+i}i}|jjj D]\}}|j d} t| dk(r| dnd} | d}| +||vr'||} t| tr t| |}|||<^| sa| |vsf||| vsn|| |} t| tr t| |}||| |f<|j D]\\} }} | || |<| dk(s||<|j D]'\}} | ||<|dj|s | |d|<)|S)Nrrrdbt) rrrrmacrosrrrrrr!r) rrkrrctx_dictglobal_macro_overridespackage_macro_overridesr macro_valuemacro_name_split macro_packageoriginal_context_valuemacro_override_values rr"generate_runtime_unit_test_contextrs )VX7N7PRV WC{{}Hy229913>@(1':':'A'A'G'G'I S #J )//4 367G3HA3M,Q/SWM)"-J$x)?)1*)=&4nE"89OQ\"]K5@&z2!X-(="99)1-)@)L&4nE"89OQ\"]KGR' (CD+ S0BYA^A^A` 3 = ']J)=2FH] #J /%'2$  31G0L0L0N C ,J,#7HZ "":..B +  C Ortr cPt|||td}|jSrj)rrr)r rkrrs rr!generate_runtime_function_contextrs& (FH6G6I4 PC ;;=rtceZdZdefdZy)ExposureRefResolverrc0d}t|dk(r|d}n+t|dk(r|\}}nt|j||jdxs|jd}|jjj t |||y Nrrrrrrrr)rrQrrrrrrpr~rrrrs rrrzExposureRefResolver.__call__s} t9>7D Y!^ MGTDJJT: :**Y':6::c? wwT7STrtNrrrrrrrtrrrrs 3 rtrceZdZdefdZy)ExposureSourceResolverrct|dk7rt||j|jjj t |y)Nrrr)rrMrrrlistrs rrrzExposureSourceResolver.__call__s= t9>'4::> > !!$t*-rtNr rrtrrr r rtr ceZdZdefdZy)ExposureMetricResolverrct|dvrt|j||jjj t |y)N)rrrr)rrLrrrr rs rrrzExposureMetricResolver.__call__s> t9F "!tzz= = !!$t*-rtNr rrtrrrrrrtrexposurec||j|}td|||td|||td|||dS)N)rrr)rrr r)rrkrrrs rrgenerate_parse_exposurer"sa &&(6G"      )      )       rtceZdZdefdZdZy)SemanticModelRefResolverrcVd}t|dk(r|d}n+t|dk(r|\}}nt|j||jdxs|jd}|j ||||jj j t|||yr)rrQrrrrrrrs rrrz!SemanticModelRefResolver.__call__As t9>7D Y!^ MGTDJJT: :**Y':6::c? 4'2 wwT7STrtcjt|ts#td|jjdy)Nz*In a semantic model or metrics section in z, the name argument to ref() must be a string)rrrOrrkrs rrrz&SemanticModelRefResolver.validate_argsQs;$$> %rtN)rrrrrrrrtrrrr@s3 rtrsemantic_modelcH|j|}dtd|||iS)Nr)rr)rrkrrrs rrgenerate_parse_semantic_modelsrZs8 &&(6G '       rtc zeZdZdedededeededdf fd Z d Z d Z e dd e d ee de fd ZxZS) TestContextrkrr!r%macro_resolverrNc||_t|_t|||||||j |j j|j|j|_ yrj) rr thread_ctxrrs_build_test_namespacer!rrqrr)rprrkrr!r%rr|s rrrszTestContext.__init__osY-$, (NK ""$--77 dnnUrtciSrjrrs rr_build_namespacezTestContext._build_namespaces rtcg}|jjjd}|r|j|j|j j rO|j j jr/|j|j j j|j}|D]O}|jjj|}|s+|j|j jQt|j|j|j |j|}||_ y)Nget_where_subquery)rmacros_by_namerrr]rrrextendcopyr/r]r r)rpdepends_on_macrosr% lookup_macrosmacro_unique_id lookup_macrors rrr!z!TestContext._build_test_namespaces"00??CCDXY   $ $%7%A%A B :: TZZ%:%:%A%A  $ $TZZ%:%:%A%A B)..0 , IO..5599/JL!(()@)@)G)GH I -   DJJIZ )rtrrcPd}|jtr t|tj}||vr||}n||}||j r||vr|nt |jj|<|j jtjk(r~|j jrh|jj|j j}|j jjd\}}|j!||||St#|)Nr)rr\rSr]rrr'rrrgrXr file_key_namerrr add_env_varrF)rprrrrryaml_keyrs rrrzTestContext.env_vars >>+ ,+C0 0$&** #:s8L  "L  #zz %(3JL4K &&s+ ::++x}}<AYAY"&--"5"5djj6H6H"IK'+zz'?'?'E'Ec'J$Xt++C4@ $S) )rtrj)rrrr&r4rr r,r.rsr#r!r)rrrrs@rrrrnsVV V  V !/ V&V V&)(_*3*#*#**rtrrcRt|||t||}|jSrj)rrr)rrkrr%rrs rrgenerate_test_contextr2s) eVX}P^ _C ;;=rt)rrhr(rtypingrrrrrr r r r r rrtyping_extensionsrrrdbt.adapters.base.columnrdbt.adapters.base.relationrr!dbt.adapters.contracts.connectionrdbt.adapters.exceptionsrdbt.adapters.factoryrrrdbt.artifacts.resourcesrrrrrdbt.clients.jinjarr r!r"dbt.clients.jinja_staticr# dbt.configr$r%r& dbt.constantsr'dbt.context.baser(r)r*dbt.context.configuredr+dbt.context.context_configr,dbt.context.exceptions_jinjar-dbt.context.macro_resolverr.r/dbt.context.macrosr0r1dbt.context.manifestr2dbt.contracts.graph.manifestr3r4dbt.contracts.graph.metricsr5r6dbt.contracts.graph.nodesr7r8r9r:r;r<r=r>r?r@rArBdbt.exceptionsrCrDrErFrGrHrIrJrKrLrMrNrOrPrQrRrSrT dbt.flagsrU+dbt.materializations.incremental.microbatchrVdbt.node_typesrWrX dbt.utilsrYrZdbt_common.clients.jinjar[dbt_common.constantsr\dbt_common.contextr]dbt_common.events.functionsr_dbt_common.exceptionsr`rarbrcdbt_common.utilsrdrerfagateobjectr^rhrABCMetarrr rrr$r)rTrurr ResolveRefrrrrrrrrrrrrrrrrrrrrrr rrrrrrrrrrrrrrr rrrrrr2rrtrrrXsy     '"+D=6  H<<1@@,48HD0;P    ( I2-2259 :9 8 ; ;a$a$HA S[[A H)4l)4X..0 + +F +< +FLXL ::zN2&N2d.$ 0 BB8\) * aAaAH9-9& L!3 L B,B . >"$6"D.,.E.E*!!H!  #  D* D"^0^ 2 L)x)%H%%8%'h''h' CLO KoO Kd2?24 O `?`Fl6MlM      "   #s(^   %2 >F  #s(^    #s(^    #s(^   3-   #s(^ 22 22 #s(^ 2j  #s(^  ,  \\    #s(^ <|4!    #s(^ (O*/O*d      "  "   #s(^ rt