iddlZddlmZmZmZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZmZmZej&eZGddeeeeZy) N)GenericListSequence)SemanticManifestT SemanticModel)SemanticModelReference) EntityType) FileContextSemanticManifestValidationRuleSemanticModelContextValidationErrorValidationIssuevalidate_safelyceZdZdZededefdZeeddede e fdZ eedde de e fd Z y ) PrimaryEntityRuleahChecks that the primary entity has been properly defined in a semantic model. * If a semantic model contains dimensions, the primary entity must be available. * The primary entity could be defined by the primary_entity field, or by one of the entities defined in a semantic model. * There should only be one primary entity in the model. semantic_modelreturnc2t|jdkDS)Nr)len dimensions)rs s/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt_semantic_interfaces/validations/primary_entity.py_model_requires_primary_entityz0PrimaryEntityRule._model_requires_primary_entitys>,,-11zECheck that a semantic model has properly configured primary entities.c ttj|jt |j }t d|jD}t|dkDrt|dkDrR|Dcgc]}|j }}td|j dt|d d j|| fS|d}|j@td |j d |j d|jjd| fStj|r6|j*t|dk(rtd|j d| fSycc}w)N)metadata)semantic_model_name) file_contextrc3ZK|]#}|jtjus |%ywN)typer PRIMARY).0entitys r z1PrimaryEntityRule._check_model..)s&+ FKK:K]K]<]F+ s!++rzFSemantic models can have only one primary entity. The semantic model `z` has z: z, )messagecontextzThe semantic model `z` has an entity named `zG` with type primary but it also has the `primary_entity` field set to `z+`. Both should not be present in the model.zThe semantic model z contains dimensions, but it does not define a primary entity. Either add an entity with type PRIMARY or set a value for the primary_entity key.)r r from_metadatarrnametupleentitiesrr joinprimary_entity_reference element_namerr)rr'entities_with_primary_typeprimary_entityprimary_entity_namesentity_with_primary_types r _check_modelzPrimaryEntityRule._check_model s'$22N?A!(   (B!'D $66B#2>3F3F2GH 8 = =>?--;-T-T-a-a,bc78 !(    < <^ L77?./14 -n.A.A-BC./$   Y(ns<E'zUCheck that semantic models in the manifest have properly configured primary entities.semantic_manifestc\g}|jD]}|tj|z }|Sr)semantic_modelsrr4)r5issuesrs rvalidate_manifestz#PrimaryEntityRule.validate_manifest]s=)+/?? EN '44^D DF E rN)__name__ __module__ __qualname____doc__ staticmethodrboolrrrrr4rr9r(rrrrs2}222\]9]9x7P9^9vlm->8OC\nrr)loggingtypingrrr!dbt_semantic_interfaces.protocolsrr"dbt_semantic_interfaces.referencesr"dbt_semantic_interfaces.type_enumsr 5dbt_semantic_interfaces.validations.validator_helpersr r r r rr getLoggerr:loggerrr(rrrHsR**NE9   8 $Q67HI7SdKeQr