li #ddlZddlmZddlmZddlmZmZmZm Z m Z ddl m Z m Z mZddlmZmZmZmZmZej*eZGddee ee ZGd d ee ee ZGd d ee ee Zy) N) defaultdict) dataclass) DefaultDictDictGenericListSequence)MetricSemanticManifestT SemanticModel) FileContextSemanticManifestValidationRuleValidationErrorValidationIssuevalidate_safelyc eZdZdZeeddedeeefde e fdZ eedde de e fd Z y ) MetricLabelsRulez1Checks that the labels are unique across metrics.z)Checking that a metric has a unique labelmetricexisting_labelsreturnc |j|vrVttj|jd|jd|j d||jdfS|j|j ||j<y)NCan't use label `z` for metric `z#` as it's already used for metric ``contextmessagelabelrr from_metadatametadatanamerrs o/var/www/html/strategist-ai/venv_dbt/lib/python3.12/site-packages/dbt_semantic_interfaces/validations/labels.py _check_metriczMetricLabelsRule._check_metrics <8OC\Br&rc ,eZdZdZeeddedeeefde e fdZ eeddede e fdZ eed dede e fd Z eed dede e fd Zeed dede e fdZy)SemanticModelLabelsRulez9Checks that the labels are unique across semantic models.z1checking that a semantic model has a unique labelsemantic_modelrrc |j|vrVttj|jd|jd|j d||jdfS|j|j ||j<y)Nrz` for semantic model `z+` as it's already used for semantic model `rrrrr5rs r$_check_semantic_modelz-SemanticModelLabelsRule._check_semantic_model3s   ? 2'55n6M6MN/0D0D/EE\]k]p]p\qr@@OP^PdPd@e?ffgi  ! ! -4B4G4GON00 1r&zJchecking that a semantic model's dimension labels are unique within itselfc pg}td}|jD].}|j||jdz||j<0|jD]W\}}|dkDs |j t t j|jd|d|d|jY|S)NcyNrrrr&r$zJSemanticModelLabelsRule._check_semantic_model_dimensions..Ir&zDDimension labels must be unique within a semantic model. The label `` was used for z dimensions on semantic model `r) r dimensionsritemsappendrr r r!r")r5r* label_counts dimensionrcounts r$ _check_semantic_model_dimensionsz8SemanticModelLabelsRule._check_semantic_model_dimensionsEs)+.9).D '22 RI*0.^r=r&r>zAEntity labels must be unique within a semantic model. The label `r?z entities on semantic model `r) rentitiesrrArBrr r r!r")r5r*rCentityrrEs r$_check_semantic_model_entitiesz6SemanticModelLabelsRule._check_semantic_model_entitiesZs)+.9).D $-- LF||'-9&,,-G!-K V\\* L)..0 LE5qy # + 9 9.:Q:Q R"cdicjk$g%B>CVCVBW!Y  r&zHchecking that a semantic model's measure labels are unique within itselfc pg}td}|jD].}|j||jdz||j<0|jD]W\}}|dkDs |j t t j|jd|d|d|jY|S)Ncyr;rrr&r$r<zHSemanticModelLabelsRule._check_semantic_model_measures..sr=r&r>zBMeasure labels must be unique within a semantic model. The label `r?z measures on semantic model `r) rmeasuresrrArBrr r r!r")r5r*rCmeasurerrEs r$_check_semantic_model_measuresz6SemanticModelLabelsRule._check_semantic_model_measuresos)+.9).D %.. NG}}(.:7==.IA.M W]]+ N)..0 LE5qy # + 9 9.:Q:Q R"dejdkl$g%B>CVCVBW!Y  r&z8checking labels on semantic models and their sub objectsr'cg}i}|jD]g}|tj||z }|tj|z }|tj |z }|tj |z }i|S)Nr7)r5)semantic_modelsr4r8rFrKrP)r'r*labels_to_semantic_modelsr5s r$r,z)SemanticModelLabelsRule.validate_manifests)+46!/?? lN -CC-?XD F -NN^lNm mF -LL\jLk kF -LL\jLk kF  l r&N)r-r.r/r0r1rr rr2r rr8rFrKrPr r,rr&r$r4r40s'CHI%8 8OC\ Q r&r4c eZdZdZeGddZeeddede e efde e fdZ eed d ede e fd Zy ) EntityLabelsRulezDChecks that the entity labels are consistent across semantic models.c&eZdZUdZeed<eed<y)EntityLabelsRule.EntityInfozAClass used in validating of entity labels across semantic models.semantic_model_namerN)r-r.r/r0r2__annotations__rr&r$ EntityInforWsO   r&rZzNChecking entities of the same name have the same label (or None for the label)r5rrcg}|jD]}|j|j|vr9tj |j|j||j<W||jj|jk7s~|j t tj|jd|jd|jd|jd||jjd |S)N)rXrzZEntities with the same name must have the same label or the label must be `None`. Entity `z` on semantic model `z ` has label `z)` but the same entity on semantic model `rr) rIrr"rUrZrBrr r r!rX)r5rr*rJs r$rKz/EntityLabelsRule._check_semantic_model_entitiess )+$-- F||';;o53C3N3N,:,?,?v||4O4OFKK0%V[[1776<<GMM'$/$=$=n>U>U$V%//5{{m;PQ_QdQdPef & ~. / < P PQQR%T " r&z 8OC\ U r&rU)logging collectionsr dataclassesrtypingrrrrr !dbt_semantic_interfaces.protocolsr r r 5dbt_semantic_interfaces.validations.validator_helpersr rrrr getLoggerr-loggerrr4rUrr&r$rfs#!==VV   8 $56GH'RcJd