industrial-model 0.1.23__py3-none-any.whl → 0.1.24__py3-none-any.whl

industrial_model/cognite_adapters/filter_mapper.py

@@ -1,9 +1,11 @@
 from datetime import date, datetime
+from typing import Any
 
 import cognite.client.data_classes.filters as cdf_filters
 from cognite.client.data_classes.data_modeling import MappedProperty, View
 
 from industrial_model.cognite_adapters.utils import get_property_ref
+from industrial_model.models.entities import InstanceId
 from industrial_model.statements import (
     BoolExpression,
     Expression,
@@ -54,11 +56,8 @@ class FilterMapper:
         property_ref = get_property_ref(expression.property, root_view)
 
         value_ = expression.value
-        if isinstance(value_, datetime):
-            value_ = datetime_to_ms_iso_timestamp(value_)
 
-        if isinstance(value_, date):
-            value_ = value_.strftime("%Y-%m-%d")
+        value_ = self._handle_type_value_convertion(value_)
 
         if expression.operator == "==":
             return cdf_filters.Equals(property_ref, value_)
@@ -96,3 +95,14 @@ class FilterMapper:
         assert isinstance(view_definiton, MappedProperty)
         assert view_definiton.source
         return self._view_mapper.get_view(view_definiton.source.external_id)
+
+    def _handle_type_value_convertion(self, value_: Any) -> Any:
+        if isinstance(value_, datetime):
+            return datetime_to_ms_iso_timestamp(value_)
+        elif isinstance(value_, date):
+            return value_.strftime("%Y-%m-%d")
+        elif isinstance(value_, InstanceId):
+            return value_.model_dump(mode="json", by_alias=True)
+        elif isinstance(value_, list):
+            return [self._handle_type_value_convertion(v) for v in value_]
+        return value_

{industrial_model-0.1.23.dist-info → industrial_model-0.1.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: industrial-model
-Version: 0.1.23
+Version: 0.1.24
 Summary: Industrial Model ORM
 Author-email: Lucas Alves <lucasrosaalves@gmail.com>
 Classifier: Programming Language :: Python

{industrial_model-0.1.23.dist-info → industrial_model-0.1.24.dist-info}/RECORD

@@ -5,7 +5,7 @@ industrial_model/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 industrial_model/utils.py,sha256=oh4AxwxXaWgIC2uolkCbvkgo0ququHB6yAPVIXy45Ts,663
 industrial_model/cognite_adapters/__init__.py,sha256=cKPW5y-wWiyWYLPDaNjhqc5zdsgdLc40vunIlz2Debg,6246
 industrial_model/cognite_adapters/aggregation_mapper.py,sha256=qglWUnyhgSPG005HzIRyjYkt9V2WKsab5y4ruwuKLcA,2503
-industrial_model/cognite_adapters/filter_mapper.py,sha256=xXTzxCYKU1-6TxG5W_bb8zNyWi2bfIM9RiujyOHUujg,3900
+industrial_model/cognite_adapters/filter_mapper.py,sha256=R_OX8JOM33vzBVFvfo0iDmSJl5rhGjsTBLyKDoesHO0,4349
 industrial_model/cognite_adapters/models.py,sha256=2j2IS01uPkQEp9WdVk8seYzEqGcDdWFnpzXhusHB2zk,945
 industrial_model/cognite_adapters/optimizer.py,sha256=EQfCe-4mSeXhYa2kcqJerPXYsCRgMJPA1AIf1o1KNH0,2415
 industrial_model/cognite_adapters/query_mapper.py,sha256=QdFVkfmL4tWcCy_oypOluT0Fo68obPk2RjBPvVA9mUs,5950
@@ -27,6 +27,6 @@ industrial_model/queries/models.py,sha256=wn4I8bTUGCT5keF6mUA7bEYNgSsOxeN-zibY5C
 industrial_model/queries/params.py,sha256=ehgCoR5n6E-tkEuoymZ2lkLcSzMaBAx_HnyJ7sWpqz0,964
 industrial_model/statements/__init__.py,sha256=rjLRo2KoazHQaOpmPkxbI3_Nm8NCkJxjpuqow6IZVSc,4221
 industrial_model/statements/expressions.py,sha256=4ZZOcZroI5-4xRw4PXIRlufi0ARndE5zSbbxLDpR2Ec,4816
-industrial_model-0.1.23.dist-info/METADATA,sha256=HoA3Ba1kVrvGjtZQEqGvFHL2NnvV8G1S1uIcq7Of53c,6858
-industrial_model-0.1.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-industrial_model-0.1.23.dist-info/RECORD,,
+industrial_model-0.1.24.dist-info/METADATA,sha256=F8lZBHy_GghR4NABWGDlVrQdITuncMGQ0FF89ZCCHQg,6858
+industrial_model-0.1.24.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+industrial_model-0.1.24.dist-info/RECORD,,