cognite-neat 0.82.2__py3-none-any.whl → 0.82.3__py3-none-any.whl

cognite/neat/_version.py

@@ -1 +1 @@
-__version__ = "0.82.2"
+__version__ = "0.82.3"

cognite/neat/rules/models/information/_converter.py

@@ -1,8 +1,12 @@
 import re
 from collections import Counter, defaultdict
 from collections.abc import Collection
+from datetime import date, datetime
 from typing import TYPE_CHECKING, Literal
 
+from cognite.client.data_classes import data_modeling as dms
+
+from cognite.neat.rules.models import data_types
 from cognite.neat.rules.models._base import (
     ExtensionCategory,
     SchemaCompleteness,
@@ -15,6 +19,7 @@ from cognite.neat.rules.models.entities import (
     ClassEntity,
     ContainerEntity,
     DMSUnknownEntity,
+    MultiValueTypeInfo,
     ReferenceEntity,
     UnknownEntity,
     ViewEntity,
@@ -172,6 +177,8 @@ class _InformationRulesConverter:
             value_type = DMSUnknownEntity()
         elif isinstance(prop.value_type, ClassEntity):
             value_type = prop.value_type.as_view_entity(default_space, default_version)
+        elif isinstance(prop.value_type, MultiValueTypeInfo):
+            value_type = self.convert_multi_value_type(prop.value_type)
         else:
             raise ValueError(f"Unsupported value type: {prop.value_type.type_}")
 
@@ -271,3 +278,24 @@ class _InformationRulesConverter:
             return suffix[: suffix_number.start()] + str(int(suffix_number.group()) + 1)
         else:
             return f"{suffix}2"
+
+    @staticmethod
+    def convert_multi_value_type(value_type: MultiValueTypeInfo) -> DataType:
+        if not all(isinstance(type_, DataType) for type_ in value_type.types):
+            raise ValueError("Only MultiValueType with DataType types is supported")
+        # We check above that there are no ClassEntity types in the MultiValueType
+        py_types = {type_.python for type_ in value_type.types}  # type: ignore[union-attr]
+        if dms.Json in py_types and len(py_types) > 1:
+            raise ValueError("MultiValueType with Json and other types is not supported")
+        elif dms.Json in py_types:
+            return data_types.Json()
+        elif not (py_types - {bool}):
+            return data_types.Boolean()
+        elif not (py_types - {int, bool}):
+            return data_types.Integer()
+        elif not (py_types - {float, int, bool}):
+            return data_types.Double()
+        elif not (py_types - {datetime, date}):
+            return data_types.DateTime()
+
+        return data_types.String()

{cognite_neat-0.82.2.dist-info → cognite_neat-0.82.3.dist-info}/METADATA

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

{cognite_neat-0.82.2.dist-info → cognite_neat-0.82.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 cognite/neat/__init__.py,sha256=v-rRiDOgZ3sQSMQKq0vgUQZvpeOkoHFXissAx6Ktg84,61
-cognite/neat/_version.py,sha256=RTYfZgQQ2I7IHQ7CnMLKfzWJLJYwV_HoEeJ8bvw2WlY,23
+cognite/neat/_version.py,sha256=3x2wKMbffUEEJlJ3cDVbyt1M_iEH-TBgkPoWxClY4WU,23
 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
@@ -242,7 +242,7 @@ cognite/neat/rules/models/dms/_validation.py,sha256=5mk9L99FSwC8Ok7weEjnFJ_OZnmq
 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/information/__init__.py,sha256=HR6g8xgyU53U7Ck8pPdbT70817Q4NC1r1pCRq5SA8iw,291
-cognite/neat/rules/models/information/_converter.py,sha256=r0a2uyzv8m82xzAkYt_-ZXdMN5u46SA_mn95Oo7ng-s,11424
+cognite/neat/rules/models/information/_converter.py,sha256=Xxlg_F6wWXNOnnq4ZJDgj1blDL-mZdFVlYutktZ9r68,12757
 cognite/neat/rules/models/information/_rules.py,sha256=G_FsNMRy5h4u9c3WfCAo2isrGlflgjsgRGpVLQSUJ5Q,13210
 cognite/neat/rules/models/information/_rules_input.py,sha256=ExCjcD0pvsThXYDf3uWYLzSLqN_2OtXFggbW_RB8hr4,10343
 cognite/neat/rules/models/information/_serializer.py,sha256=yti9I_xJruxrib66YIBInhze___Io-oPTQH6uWDumPE,3503
@@ -307,8 +307,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.82.2.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
-cognite_neat-0.82.2.dist-info/METADATA,sha256=MoKqP6Krsaj7ZfMrDfvMfmCPOgHRhZyjQtd1Vp2BBZc,9290
-cognite_neat-0.82.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-cognite_neat-0.82.2.dist-info/entry_points.txt,sha256=61FPqiWb25vbqB0KI7znG8nsg_ibLHBvTjYnkPvNFso,50
-cognite_neat-0.82.2.dist-info/RECORD,,
+cognite_neat-0.82.3.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
+cognite_neat-0.82.3.dist-info/METADATA,sha256=vFK-g0VZN6HPCgAo_yEgu5XPKpAIbq1Gyhj2Tz8091w,9290
+cognite_neat-0.82.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+cognite_neat-0.82.3.dist-info/entry_points.txt,sha256=61FPqiWb25vbqB0KI7znG8nsg_ibLHBvTjYnkPvNFso,50
+cognite_neat-0.82.3.dist-info/RECORD,,