li<ddlZddlmZmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZdd lmZGd d eZej,d ej.d ej0dej2dej4dej6dej8dej:dej<dej>dej@dejBdi Z"eGddeZ#eGddeZ$eGddee Z%eGdde%Z&eGd d!e%Z'eGd"d#e%Z(e e'e&e(fZ)y)$N) dataclassfield)AnyDictListOptionalUnion)SerializableType)FileHash)MAXIMUM_SEED_SIZE)StrEnum dbtClassMixin) SourceKeyc<eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zy ) ParseFileTypemacromodelsnapshotanalysis singular_test generic_testseeddocsschemahookfixturefunctionN)__name__ __module__ __qualname__MacroModelSnapshotAnalysis SingularTest GenericTestSeed DocumentationSchemaHookFixtureFunctionX/var/www/html/strategist-ai/venv_dbt/lib/python3.12/site-packages/dbt/contracts/files.pyrrs> E EHH"L K DM F DGHr/r MacroParser ModelParserSnapshotParserAnalysisParserSingularTestParserGenericTestParser SeedParserDocumentationParser SchemaParser HookParser FixtureParserFunctionParserceZdZUeed<eed<eed<eed<edefdZedefdZedefdZ edefd Z de fd Z y ) FilePath searched_path relative_pathmodification_time project_rootreturnc|jSN) absolute_pathselfs r0 search_keyzFilePath.search_key4s!!!r/ctjj|j|j|j SrE)ospathjoinrBr?r@rGs r0 full_pathzFilePath.full_path9s-ww||D--t/A/A4CUCUVVr/cTtjj|jSrE)rKrLabspathrNrGs r0rFzFilePath.absolute_path>swwt~~..r/cjtjj|j|jSrE)rKrLrMr?r@rGs r0original_file_pathzFilePath.original_file_pathBs#ww||D..0B0BCCr/cbtj|jjtkDS)zCReturn whether the file this represents is over the seed size limit)rKstatrNst_sizer rGs r0seed_too_largezFilePath.seed_too_largeFs!wwt~~&..1BBBr/N) rr r!str__annotations__floatpropertyrIrNrFrRboolrVr.r/r0r>r>-s "C""W3WW/s//DCDDCCr/r>cveZdZd dZedefdZedefdZedefdZedZ edZ y) RemoteFilerCNc||dk(rd|_n|dk(rd|_ntd|d|j|_y)Nsqlz.sqlpythonz.pyz!Invalid language for remote File zfrom remote system)path_end RuntimeErrorrL)rHlanguages r0__init__zRemoteFile.__init__MsE u "DM  !!DM!B8*MN N(8 r/c|jSrErLrGs r0r?zRemoteFile.searched_pathV yyr/c|jSrErfrGs r0r@zRemoteFile.relative_pathZrgr/c|jSrErfrGs r0rFzRemoteFile.absolute_path^rgr/c|jSrErfrGs r0rRzRemoteFile.original_file_pathbrgr/c|jSrErfrGs r0rAzRemoteFile.modification_timefrgr/)rCN) rr r!rdrZrWr?r@rFrRrAr.r/r0r]r]Ks9sssr/r]ceZdZUdZeeefed<eed<dZ e e ed<dZ e e ed<dZe e ed<edZed Zd Zed ee effd Zdd ed e effd ZxZS)BaseSourceFilezDefine a source file in dbtrLchecksumN project_nameparse_file_typecontentsct|jtry|jd|jjS)Nz://) isinstancerLr]rorRrGs r0file_idzBaseSourceFile.file_idxs6 dii ,##$C (D(D'EFFr/c.|jjSrE)rLrRrGs r0rRz!BaseSourceFile.original_file_path~syy+++r/c&|j}|SrE)to_dict)rHdcts r0 _serializezBaseSourceFile._serializeslln r/rxc|ddk(rtj|}|S|ddk(rtj|}|Stj|}|S)Nrprr)SchemaSourceFile from_dictFixtureSourceFile SourceFile)clsrxsfs r0 _deserializezBaseSourceFile._deserializese  !X -!++C0B  " #y 0",,S1B %%c*B r/contextct|||}t|j}|D]}t ||ts||r||=!d|vr|d=|S)Nrq)super__post_serialize__listkeysrs)rHrxrdct_keyskey __class__s r0rz!BaseSourceFile.__post_serialize__sdg(g6 # C#c(D)#c(H   J r/rE)rr r!__doc__r r>r]rXr rorrWrprrqrZrtrRry classmethodrintrr __classcell__rs@r0rmrmks% *$ %%"&L(3-&/3OXm,3"Hhsm" GG ,,tCH~ d Xd^  r/rmc eZdZUeeZeeed<eeZ eeed<eeZ eeed<eeZ eeed<eeZ eeed<e deddfd Zd Ze d ed ed eddfdZy)r~default_factorynodesrmacrosenv_vars functionsrLrCcb||tj|j}d|_|S)z2Parse seeds over the size limit with just the path)rLrn)r rLrRrq)rrLrHs r0big_seedzSourceFile.big_seeds+x}}T5L5L'MN  r/cX||jvr|jj|yyrE)rappendrHvalues r0add_nodezSourceFile.add_nodes%  " JJ  e $ #r/rqrorccV|t|tj|||}|S)N)rLrnrorq)r]r from_contents)rrqrorcrHs r0remotezSourceFile.remotes/H%++H5%    r/N)rr r!rrrrrWrXrrrrrr>rrrr.r/r0r~r~sT2E492D1D$s)1d3FDI35Hd3i5 6ItCy6H % c r/r~cLeZdZUeeZeeefe d<eeZ eeefe d<ee Z e ee d<ee Ze ee d<ee Ze ee d<ee Ze ee d<ee Ze ee d<ee Ze ee d <eeZeeefe d <ee Ze ee d <ee Ze ee d <ee Ze ee d <ee Ze ee d<ee Ze ee d<eeZeeefe d<ee Ze ee d<eeZeeefe d<eeZeeefe d<eeZeeefe d<eeZeeefe d<dZ e!eeefe d<dZ"e!eeefe d<e#dZ$e#dZ%e#dZ&e#dZ'd8dede!effd Z(d Z)d!Z*d"Z+d#Z,d$ed%efd&Z-d'Z.d(Z/d)Z0d8d*Z1d8d+e!eeeffd,Z2d-Z3d.Z4d/Z5d0ed1ed2ed+dfd3Z6d0ed1ed+e!efd4Z7d0ed1ed5ed+dfd6Z8d0ed1ed+e!efd7Z9xZ:S)9r{rdfy data_testssources exposuresrmetrics snapshotsgenerated_metricsmetrics_from_measuresgroupsndpsemantic_models unit_tests saved_queriesmcpsoprunrendered_configsunrendered_databasesunrendered_schemasNpp_dict pp_test_indexc|jSrE)rrGs r0dict_from_yamlzSchemaSourceFile.dict_from_yaml xxr/c|jSrE)rrGs r0 node_patcheszSchemaSourceFile.node_patchesrr/c|jSrE)rrGs r0 macro_patcheszSchemaSourceFile.macro_patchesrr/c|jSrE)rrGs r0source_patcheszSchemaSourceFile.source_patchesrr/rxrcFt|||}dD] }||vs||= |S)N)rr)rr)rHrxrrrs r0rz#SchemaSourceFile.__post_serialize__s8g(g6/ CczH  r/c:|jj|yrE)rr)rHyaml_key unique_ids r0 append_patchzSchemaSourceFile.append_patchs   +r/c|d}|d}||jvri|j|<||j|vrg|j||<|j||j|y)Nnamer)rr)rHnode_unique_id test_fromrrs r0add_testzSchemaSourceFile.add_testsp  doo %#%DOOC ts+ +)+DOOC  & T")).9r/cf||jvr#||j|vr|j||=yyyrErrHrrs r0 remove_testszSchemaSourceFile.remove_testss: t &tx00OOH-d31 'r/ch||jvr#||j|vr|j||SgSrErrs r0 get_testszSchemaSourceFile.get_tests s8 t &tx00x066 r/semantic_model_namemetric_unique_idc|jr|j||jvrg|j|<|j|j|yrE)rfix_metrics_from_measuresrr)rHrrs r0add_metrics_from_measuresz*SchemaSourceFile.add_metrics_from_measuressS  ! !  * * , d&@&@ @>@D & &': ; ""#67>>?OPr/c|j}g|_|D]i}|jd}|d}d|jvs(|jdD]0}d|vs|dD]!}|d|k(s |j|d|02ky)N.rmeasuresr)rsplitrr)rHrrparts metric_name sem_modelmeasures r0rz*SchemaSourceFile.fix_metrics_from_measuress!22!# !2 & $**3/E)K D$7$77!%!4!45F!G&I!Y.'0'<&G&v+= $ > >y?PRb c %&& &r/cd}d}|jjD]8}|j|D]$}|j||D] }||k(s |}|}$&:||fSrE)rr)rHtest_unique_idr block_namerrrs r0get_key_and_name_for_testz*SchemaSourceFile.get_key_and_name_for_test5s ??'') C, !%!5d!;I N2#&%)    *%%r/cg}|jjD]7}|j|D]#}|j|j||%9|SrE)rrextend)rHtest_idsrrs r0get_all_test_idsz!SchemaSourceFile.get_all_test_idsAs\??'') tCy>-2$,G4S>Gd3FDI340Cc0!&t!r]rmr~r{r} AnySourceFiler.r/r0rs (33,1+> G  ,, 42 !6. ?,   C}C C:  > /]$4/ /d  > R;~R; R;j ** *& 4EEF r/