cognite-neat 0.85.10__py3-none-any.whl → 0.85.11__py3-none-any.whl

cognite/neat/_version.py

@@ -1 +1 @@
-__version__ = "0.85.10"
+__version__ = "0.85.11"

cognite/neat/rules/models/dms/_rules.py

@@ -2,6 +2,7 @@ import math
 import re
 import sys
 import warnings
+from collections import defaultdict
 from datetime import datetime
 from typing import TYPE_CHECKING, Any, ClassVar, Literal
 
@@ -390,3 +391,72 @@ class DMSRules(BaseRules):
         from ._converter import _DMSRulesConverter
 
         return _DMSRulesConverter(self).as_domain_rules()
+
+    def create_reference(
+        self, reference: "DMSRules", view_extension_mapping: dict[str, str], default_extension: str | None = None
+    ) -> None:
+        """Takes this data models and makes it into an extension of the reference data model.
+
+        The argument view_extension_mapping is a dictionary where the keys are views of this data model,
+        and each value is the view of the reference data model that the view should extend. For example:
+
+        ```python
+        view_extension_mapping = {"Pump": "Asset"}
+        ```
+
+        This would make the view "Pump" in this data model extend the view "Asset" in the reference data model.
+        Note that all the keys in the dictionary must be external ids of views in this data model,
+        and all the values must be external ids of views in the reference data model.
+
+        Args:
+            reference: The reference data model
+            view_extension_mapping: A dictionary mapping views in this data model to views in the reference data model
+            default_extension: The default view in the reference data model that views in this
+                data model should extend if no mapping is provided.
+
+        """
+        if self.reference is not None:
+            raise ValueError("Reference already exists")
+        self.reference = reference
+        view_by_external_id = {view.view.external_id: view for view in self.views}
+        ref_view_by_external_id = {view.view.external_id: view for view in reference.views}
+
+        if invalid_views := set(view_extension_mapping.keys()) - set(view_by_external_id.keys()):
+            raise ValueError(f"Views are not in this dat model {invalid_views}")
+        if invalid_views := set(view_extension_mapping.values()) - set(ref_view_by_external_id.keys()):
+            raise ValueError(f"Views are not in the reference data model {invalid_views}")
+        if default_extension and default_extension not in ref_view_by_external_id:
+            raise ValueError(f"Default extension view not in the reference data model {default_extension}")
+
+        properties_by_view_external_id: dict[str, dict[str, DMSProperty]] = defaultdict(dict)
+        for prop in self.properties:
+            properties_by_view_external_id[prop.view.external_id][prop.view_property] = prop
+
+        ref_properties_by_view_external_id: dict[str, dict[str, DMSProperty]] = defaultdict(dict)
+        for prop in reference.properties:
+            ref_properties_by_view_external_id[prop.view.external_id][prop.view_property] = prop
+
+        for view_external_id, view in view_by_external_id.items():
+            if view_external_id in view_extension_mapping:
+                ref_external_id = view_extension_mapping[view_external_id]
+            elif default_extension:
+                ref_external_id = default_extension
+            else:
+                continue
+
+            ref_view = ref_view_by_external_id[ref_external_id]
+            shared_properties = set(properties_by_view_external_id[view_external_id].keys()) & set(
+                ref_properties_by_view_external_id[ref_external_id].keys()
+            )
+            if shared_properties:
+                if view.implements is None:
+                    view.implements = [ref_view.view]
+                elif isinstance(view.implements, list) and ref_view.view not in view.implements:
+                    view.implements.append(ref_view.view)
+            for prop_name in shared_properties:
+                prop = properties_by_view_external_id[view_external_id][prop_name]
+                ref_prop = ref_properties_by_view_external_id[ref_external_id][prop_name]
+                if ref_prop.container and ref_prop.container_property:
+                    prop.container = ref_prop.container
+                    prop.container_property = ref_prop.container_property
+                prop.reference = ReferenceEntity.from_entity(ref_prop.view, ref_prop.view_property)

cognite/neat/rules/models/dms/_rules_input.py

@@ -146,7 +146,9 @@ class DMSPropertyInput:
             "Value Type": value_type,
             "Property (linage)": self.property_ or self.view_property,
             "Class (linage)": (
-                ClassEntity.load(self.class_, prefix=default_space, version=default_version) if self.class_ else None
+                ClassEntity.load(self.class_ or self.view, prefix=default_space, version=default_version)
+                if self.class_ or self.view
+                else None
             ),
             "Name": self.name,
             "Description": self.description,

cognite/neat/rules/models/entities.py

@@ -499,6 +499,13 @@ class ReferenceEntity(ClassEntity):
     prefix: str
     property_: str | None = Field(None, alias="property")
 
+    @classmethod
+    def from_entity(cls, entity: Entity, property_: str) -> "ReferenceEntity":
+        if isinstance(entity, ClassEntity):
+            return cls(prefix=str(entity.prefix), suffix=entity.suffix, version=entity.version, property=property_)
+        else:
+            return cls(prefix=str(entity.prefix), suffix=entity.suffix, property=property_)
+
     def as_view_id(self) -> ViewId:
         if isinstance(self.prefix, _UndefinedType) or isinstance(self.suffix, _UnknownType):
             raise ValueError("Prefix is not defined or suffix is unknown")

{cognite_neat-0.85.10.dist-info → cognite_neat-0.85.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-neat
-Version: 0.85.10
+Version: 0.85.11
 Summary: Knowledge graph transformation
 Home-page: https://cognite-neat.readthedocs-hosted.com/
 License: Apache-2.0

{cognite_neat-0.85.10.dist-info → cognite_neat-0.85.11.dist-info}/RECORD

@@ -1,6 +1,6 @@
 cognite/neat/__init__.py,sha256=AiexNcHdAHFbrrbo9c65gtil1dqx_SGraDH1PSsXjKE,126
 cognite/neat/_shared.py,sha256=RSaHm2eJceTlvb-hMMe4nHgoHdPYDfN3XcxDXo24k3A,1530
-cognite/neat/_version.py,sha256=r9a11H4Pc6Steq_whDR9D6B9Q8eBS2t0VVcst7vsaQA,24
+cognite/neat/_version.py,sha256=bVP2BELxIGHP-amVHOwETotJ5zyQqFcbe7c26GvSm-c,24
 cognite/neat/app/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cognite/neat/app/api/asgi/metrics.py,sha256=nxFy7L5cChTI0a-zkCiJ59Aq8yLuIJp5c9Dg0wRXtV0,152
 cognite/neat/app/api/configuration.py,sha256=2U5M6M252swvQPQyooA1EBzFUZNtcTmuSaywfJDgckM,4232
@@ -236,13 +236,13 @@ cognite/neat/rules/models/data_types.py,sha256=xAjm8UVEQlSylfkUJWiWP2NaK6dLYRDAB
 cognite/neat/rules/models/dms/__init__.py,sha256=Wzyqzz2ZIjpUbDg04CMuuIAw-f2A02DayNeqO9R-2Hw,491
 cognite/neat/rules/models/dms/_converter.py,sha256=QaJT0z0Yo4gkG9tHPZkU8idF8PK7c-tDahbyIT-WJQU,5959
 cognite/neat/rules/models/dms/_exporter.py,sha256=DRL5ahcBEdBltHLA2rzLyUfqG5SZkqINkqmYGDkQVg0,24785
-cognite/neat/rules/models/dms/_rules.py,sha256=XTIEWG49VjNs_bICGlgMd6uk4hseY1H6UVe2KMfHAWQ,16152
-cognite/neat/rules/models/dms/_rules_input.py,sha256=apDDTQll9UAyYL5gS2vDxHsujWrGBilTp7lK2kzJWO8,13467
+cognite/neat/rules/models/dms/_rules.py,sha256=M7qv4uTZ-qpnIFQtoExEDppz0m0beNPm01djZxaj0yo,20118
+cognite/neat/rules/models/dms/_rules_input.py,sha256=M5AqwW-p6z_qAum-1HE_lmONSmg-vF1ak7Y_AiM8vnU,13525
 cognite/neat/rules/models/dms/_schema.py,sha256=balykJXJy93HJzBZN-KBYrRJJvGsaEtQeX7-z0tbX7w,49566
 cognite/neat/rules/models/dms/_serializer.py,sha256=iqp2zyyf8jEcU-R3PERuN8nu248xIqyxiWj4owAn92g,6406
 cognite/neat/rules/models/dms/_validation.py,sha256=5mk9L99FSwC8Ok7weEjnFJ_OZnmqMWUc6XFMTfkqfDw,14549
 cognite/neat/rules/models/domain.py,sha256=wZ-DeIPFnacbNlxSrRuLzUpnhHdTpzNc22z0sDfisi4,2880
-cognite/neat/rules/models/entities.py,sha256=qZa_PJOjFk3yTu5NyTSbAjsrU0HUxVnme_7YruBJoRQ,20460
+cognite/neat/rules/models/entities.py,sha256=u8jbE7n99FKKND6U9ogoUfhjtataDtATprAyRgBNwrw,20823
 cognite/neat/rules/models/information/__init__.py,sha256=HR6g8xgyU53U7Ck8pPdbT70817Q4NC1r1pCRq5SA8iw,291
 cognite/neat/rules/models/information/_converter.py,sha256=nsfOCe13c3gqbbF1mrQHofemzYKpvg_NKjJZDwSXo5E,13960
 cognite/neat/rules/models/information/_rules.py,sha256=v8ptTif8X55wYaNHvsx65jK7PP2ClSYWyFWUyI2rbyY,13487
@@ -310,8 +310,8 @@ cognite/neat/workflows/steps_registry.py,sha256=fkTX14ZA7_gkUYfWIlx7A1XbCidvqR23
 cognite/neat/workflows/tasks.py,sha256=dqlJwKAb0jlkl7abbY8RRz3m7MT4SK8-7cntMWkOYjw,788
 cognite/neat/workflows/triggers.py,sha256=_BLNplzoz0iic367u1mhHMHiUrCwP-SLK6_CZzfODX0,7071
 cognite/neat/workflows/utils.py,sha256=gKdy3RLG7ctRhbCRwaDIWpL9Mi98zm56-d4jfHDqP1E,453
-cognite_neat-0.85.10.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
-cognite_neat-0.85.10.dist-info/METADATA,sha256=8UOfc9V1ro6ldjvcGGTZek0bLiddHUjhdm4pbAfNoFw,9494
-cognite_neat-0.85.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-cognite_neat-0.85.10.dist-info/entry_points.txt,sha256=61FPqiWb25vbqB0KI7znG8nsg_ibLHBvTjYnkPvNFso,50
-cognite_neat-0.85.10.dist-info/RECORD,,
+cognite_neat-0.85.11.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
+cognite_neat-0.85.11.dist-info/METADATA,sha256=dofrglhVYBscSaExRgcBzxsXGN7a1cNkk1srGCk692w,9494
+cognite_neat-0.85.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+cognite_neat-0.85.11.dist-info/entry_points.txt,sha256=61FPqiWb25vbqB0KI7znG8nsg_ibLHBvTjYnkPvNFso,50
+cognite_neat-0.85.11.dist-info/RECORD,,