iGddlmZmZmZmZddlmZmZmZddl m Z m Z ddl m Z mZddlmZmZmZmZmZmZmZmZGddeeeeZy) )DictGenericListSequence) DimensionSemanticManifestT SemanticModel)DimensionReferenceSemanticModelElementReference) DimensionTypeTimeGranularity)DimensionInvariants FileContextSemanticManifestValidationRuleSemanticModelElementContextSemanticModelElementTypeValidationErrorValidationIssuevalidate_safelyc eZdZdZeeddedeefdZ eedde d e e e fd edeefd Zeed d ed e e efdedeefdZy)DimensionConsistencyRulezChecks for consistent dimension properties in the semantic models in a model. * Dimensions with the same name should be of the same type. * Dimensions with the same name should be either all partitions or not. z7running model validation ensuring dimension consistency)whats_being_donesemantic_manifestreturnci}i}g}|jD]I}|tj||dz }|jD]}|tj |||z }K|S)NT)semantic_modeldimension_to_invariantupdate_invariant_dict) dimensiontime_dims_to_granularityr)semantic_modelsr_validate_semantic_model dimensions_validate_dimension)rrr issuesrrs t/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt_semantic_interfaces/validations/dimension_const.pyvalidate_manifestz*DimensionConsistencyRule.validate_manifest sQSNP (*/?? N .GG-F\txH F,66  2FF'-E#1G   zqchecking that time dimensions of the same name that are not primary have the same time granularity specificationsrr rc zg}ttj|jt |j |j t j}|jtjk(r|j|vr|jr|jj||j<|jy|jj||jk7rS||j}|jt|d|j d|j d|j d|S) aCheck that time dimensions of the same name and aren't primary have the same time granularity. Args: dimension: the dimension to check time_dims_to_granularity: a dict from the dimension to the time granularity it should have semantic_model: the associated semantic model. Used for generated issue messages Throws: MdoValidationError if there is an inconsistent dimension in the semantic model. metadatasemantic_model_name element_name file_contextsemantic_model_element element_typezaTime granularity must be the same for time dimensions with the same name. Problematic dimension: z in semantic model with name: `z`. Expected granularity is .contextmessage)rr from_metadatar+r namer DIMENSIONtyper TIME reference type_paramstime_granularityappendr)rr rr%r5expected_granularitys r&r$z,DimensionConsistencyRule._validate_dimension4s($)+-$22N>]// /""*BByG\G\@I@U@U@f@f()<)<=))5!-->>BZ[d[n[nBoo+CIDWDW+X(MM'$+'66?nn5EF . 3 344OPdPiPiOjjk%m r(zTchecking that the semantic model has dimensions consistent with the given invariantsrrc.g}|jD]}|j|j}|7|r4t|j|j xsd}|||j<WX|j xsd}t tj|jt|j|jtj}|j|jk7rQ|jt|d|jd|jd|jd|jd |j |k7s<|jt|d|jd |jd |j d|d |S)aChecks that the given semantic model has dimensions consistent with the given invariants. Args: semantic_model: the semantic model to check dimension_to_invariant: a dict from the dimension name to the properties it should have update_invariant_dict: whether to insert an entry into the dict if the given dimension name doesn't exist. Throws: MdoValidationError if there is an inconsistent dimension in the semantic model. Fr*r,r/zIn semantic model `z `, type conflict for dimension `z` - already in model as type `z ` but got ``r4z4, conflicting is_partition attribute for dimension `z+` - already in model with is_partition as `z``)r#getr<rr: is_partitionrrr7r+r r8rr9r?r)rrrr%rdimension_invariantrDr5s r&r"z1DimensionConsistencyRule._validate_semantic_modelcs")+'22) I"8"<"+>?%11:UL1(66@W@WX'D(6(;(;)..(6?? G#''9>>9 # '"5n6I6I5JK%NN++IJ]JbJbIcd%NN+1!.#//<? # '"5n6I6I5JK&&/&9&9%:;22E2R2R1ST(>!-C) V r(N)__name__ __module__ __qualname____doc__ staticmethodrrrrr'rrr r r r$rboolr"r(r&rrs &_`->8OC\a$8(("&'9?'J"K(&( / " (  (To:%: $%79L%L M: $: / " ::r(rN)typingrrrr!dbt_semantic_interfaces.protocolsrrr "dbt_semantic_interfaces.referencesr r "dbt_semantic_interfaces.type_enumsr r 5dbt_semantic_interfaces.validations.validator_helpersrrrrrrrrrrLr(r&rRsM00 N   H=>OPRYZkRlHr(