i$ddlZddlmZmZmZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZej(eZGddeeeeZGd d eeeeZy) N)GenericListSequence)SemanticManifestT SemanticModel)SemanticModelReference) EntityType) FileContextSemanticManifestValidationRuleSemanticModelContextSemanticModelValidationHelpersValidationErrorValidationIssuevalidate_safelyceZdZdZeeddedeefdZ eedde deefd Z y ) SemanticModelValidityWindowRulez[Checks validity windows in semantic models to ensure they comply with runtime requirements.zKchecking correctness of the time dimension validity parameters in the modelwhats_being_donesemantic_manifestreturncvg}|jD]'}|jtj|)|S)zKChecks the validity param definitions in every semantic model in the model.semantic_model)semantic_modelsextendr_validate_semantic_modelrissuesrs t/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt_semantic_interfaces/validations/semantic_models.pyvalidate_manifestz1SemanticModelValidityWindowRule.validate_manifestsC)+/?? sN MM9RRbpRq r s z]checking the semantic model's validity parameters for compatibility with runtime requirementsrc g}|jDcgc]}|j|}}|s|Sttj|j t |j}d}|Dcgc]}|j}}|Dcgc]2}|js|jjs'|j4}}|Dcgc]2}|js|jjs'|j4}}t|} t|} t|dk(r@| dk(r;| dk(r6t|d|jd|dd |d  } |j| nt|d k7r>t|d|jd t|d|d| } |j| nC| dk7s| dk7r9g}t|d|jd|d|d|d| } |j| |jD cgc]1} | jtj tj"fvs0| 3} } t%|jD cgc]} | jtj&u c} sGt|d|jd| D cgc]} | jc} d } |j| | r,t|d|jd } |j| |j(rU|j(Dcgc]}|j}}t|d|jd|d|d } |j| |Scc}wcc}wcc}wcc}wcc} wcc} wcc} wcc}w)z_Runs assertions on semantic models with validity parameters set on one or more time dimensions.metadatasemantic_model_name file_contextrzSemantic models using dimension validity params to define a validity window must have exactly two time dimensions with validity params specified - one marked `is_start` and the other marked `is_end`.zSemantic model zA has a single validity param dimension that defines its window: `rz4`. This is not a currently supported configuration! z If you have one column defining a window, as in a daily snapshot table, you can define a separate dimension and increment the time value in the `expr` field as a work-around.contextmessagez has z4 dimensions defined with validity params. They are: z. There must be either zero or two! If you wish to define a validity window for this semantic model, please follow these requirements: z| has two validity param dimensions defined, but does not have exactly one each marked with is_start and is_end! Dimensions: z. is_start dimensions: z. is_end dimensions: z. a has validity param dimensions defined, but does not have an entity with type `natural` set. The natural key for this semantic model is what we use to process a validity window join. Primary or unique entities, if any, might be suitable for use as natural keys: (z).z has validity param dimensions defined and also has one or more entities designated as `primary` or `unique`. This is not yet supported, as we do not currently process joins against these key types for semantic models with validity windows specified.z has both measures and validity param dimensions defined. This is not currently supported! Please remove either the measures or the validity params. Measure names: z". Validity param dimension names: .) dimensionsvalidity_paramsr r from_metadatar$rnameis_startis_endlenrappendentitiestyper PRIMARYUNIQUEanyNATURALmeasures)rrdimvalidity_param_dimsr+ requirementsvalidity_param_dimension_namesstart_dim_names end_dim_namesnum_start_dims num_end_dimserrorentityprimary_or_unique_entitiesmeasure measure_namess rrz8SemanticModelValidityWindowRule._validate_semantic_model"s)+.<.G.Gks3K^K^Kjskk"M&$22Nq ABBv#n%uvE MM% $ % *#%n&9&9%:%DW@X?YZ22P1QR%%1N4E MM% q LA$5 O#%n&9&9%:;Z566MoM^^s$oR ~7E MM% "0!8!8& FKKJL^L^`j`q`qKr8OC\uhh8OC\h  hr!rceZdZdZeeddedeefdZ eedde deefd Z y ) SemanticModelDefaultsRulez#Checks defaults in semantic models.z8running model validation ensuring the defaults are validrrrcvg}|jD]'}|jtj|)|S)Nr)rrrR$_validate_default_agg_time_dimensionrs rr z+SemanticModelDefaultsRule.validate_manifestsA)+/?? yN MM3XXhvXw x y r!zDchecking validity of the semantic model's default agg_time_dimensionrc g}|j|jjgS|jj}tj||sj|j t t tj|jt|jd|d|jd|S) N)time_dimension_namerr#r%r'z5Default aggregation time dimension was specified as 'zC' which doesn't exist as a time dimension in semantic model named 'z'.r*) defaultsagg_time_dimensionr time_dimension_in_modelr6rr r r1r$rr2)rrdefault_agg_time_dimensions rrTz>SemanticModelDefaultsRule._validate_default_agg_time_dimensions)+  " " *n.E.E.X.X.`I%3%<%<%O%O"-EE :>  MM0%0%>%>H_H_%`'=R`ReRe'fTTnSopRR`ReReQffhj   r!N) rKrLrMrNrOrrrrr rrTrPr!rrRrRsq-&`a->8OC\b&lm]xXgOhnr!rR)loggingtypingrrr!dbt_semantic_interfaces.protocolsrr"dbt_semantic_interfaces.referencesr"dbt_semantic_interfaces.type_enumsr 5dbt_semantic_interfaces.validations.validator_helpersr r r r rrr getLoggerrKloggerrrRrPr!rrcsu**NE9   8 ${&DEV&WY`arYs{|$ >?P QSZ[lSm$r!