industrial-model 0.1.18__py3-none-any.whl → 0.1.20__py3-none-any.whl

industrial_model/cognite_adapters/__init__.py

@@ -111,7 +111,7 @@ class CogniteAdapter:
         operation = self._upsert_mapper.map(entries)
 
         for node_chunk in operation.chunk_nodes():
-            logger.info(
+            logger.debug(
                 f"Upserting {len(node_chunk)} nodes (replace={replace})"
             )
             self._cognite_client.data_modeling.instances.apply(
@@ -120,7 +120,7 @@ class CogniteAdapter:
             )
 
         for edge_chunk in operation.chunk_edges():
-            logger.info(
+            logger.debug(
                 f"Upserting {len(edge_chunk)} edges (replace={replace})"
             )
             self._cognite_client.data_modeling.instances.apply(
@@ -129,11 +129,16 @@ class CogniteAdapter:
             )
 
         for edges_to_remove_chunk in operation.chunk_edges_to_delete():
-            logger.info(f"Deleting {len(edges_to_remove_chunk)} edges")
+            logger.debug(f"Deleting {len(edges_to_remove_chunk)} edges")
             self._cognite_client.data_modeling.instances.delete(
                 edges=[item.as_tuple() for item in edges_to_remove_chunk],
             )
 
+    def delete(self, nodes: list[TViewInstance]) -> None:
+        self._cognite_client.data_modeling.instances.delete(
+            nodes=[item.as_tuple() for item in nodes],
+        )
+
     def _query_dependencies_pages(
         self,
         cognite_query: CogniteQuery,

industrial_model/cognite_adapters/query_result_mapper.py

@@ -1,6 +1,6 @@
 from collections import defaultdict
 from enum import StrEnum
-from typing import Any, Literal, TypedDict
+from typing import Any, TypedDict
 
 from cognite.client.data_classes.data_modeling import (
     Edge,

industrial_model/cognite_adapters/utils.py

@@ -142,6 +142,7 @@ def _get_leaf_cursors(
             cursor_key == view_external_id
             or not cursor_value
             or len(query_result[cursor_key]) != MAX_LIMIT
+            or cursor_key not in nodes_children
         ):
             continue
 

industrial_model/engines/async_engine.py

@@ -47,3 +47,6 @@ class AsyncEngine:
         self, entries: list[TWritableViewInstance], replace: bool = False
     ) -> None:
         return await run_async(self._engine.upsert, entries, replace)
+
+    async def delete_async(self, nodes: list[TViewInstance]) -> None:
+        return await run_async(self._engine.delete, nodes)

industrial_model/engines/engine.py

@@ -63,6 +63,11 @@ class Engine:
 
         return self._cognite_adapter.upsert(entries, replace)
 
+    def delete(self, nodes: list[TViewInstance]) -> None:
+        self._cognite_adapter.delete(
+            nodes,
+        )
+
     def _validate_data(
         self,
         entity: type[TViewInstance],

industrial_model/models/schemas.py

@@ -99,7 +99,7 @@ def _get_field_type(
     entries: list[type[BaseModel] | None] = []
     for arg in get_args(type_hint):
         if _type_is_list_or_union(arg):
-            return _get_field_type(arg, visited_count)
+            return _get_field_type(arg, parent_type, visited_count)
         entries.append(_cast_base_model(arg))
 
     return _get_field_relations(entries, parent_type, visited_count)

{industrial_model-0.1.18.dist-info → industrial_model-0.1.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: industrial-model
-Version: 0.1.18
+Version: 0.1.20
 Summary: Industrial Model ORM
 Author-email: Lucas Alves <lucasrosaalves@gmail.com>
 Classifier: Programming Language :: Python
@@ -229,6 +229,22 @@ class AggregateByNamePerson(AggregatedViewInstance):
 aggregate_result = engine.aggregate(aggregate(AggregateByNamePerson, "count"))
 
 
+
+# 9. Deletion
+
+class Entity(ViewInstance):
+    view_config = ViewInstanceConfig(
+        view_external_id="Person"
+    )
+    name: str
+
+
+statement = select(Entity).where(Entity.external_id == "Lucas")
+person = engine.query_all_pages(statement)[0]
+
+
+engine.delete([person])
+
 ```
 
 ---

{industrial_model-0.1.18.dist-info → industrial_model-0.1.20.dist-info}/RECORD

@@ -3,29 +3,29 @@ industrial_model/config.py,sha256=LseCZ8JvewFejjElPUQWdmIQQUU2wnd43McDid_Rixs,47
 industrial_model/constants.py,sha256=wtFdxkR9gojqekwXeyVCof6VUkJYuqjjgAgVVKrcxaw,450
 industrial_model/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 industrial_model/utils.py,sha256=oh4AxwxXaWgIC2uolkCbvkgo0ququHB6yAPVIXy45Ts,663
-industrial_model/cognite_adapters/__init__.py,sha256=MxX7Z6VntqmP8luG1CzbMBFu_K0ls8iU2diLsrEjLCk,5492
+industrial_model/cognite_adapters/__init__.py,sha256=K-F0BjbF9yM_YilQ7U16t7-70aAsIWUaaU_uQOul418,5680
 industrial_model/cognite_adapters/aggregation_mapper.py,sha256=rXt2BZ2jS3afD9d3B2t8aizL9TWnMnqCCNdzTn_XR3Y,2362
 industrial_model/cognite_adapters/filter_mapper.py,sha256=X2BVp305v1Wt00DTZEnCooOG_g6Jo7bkj-NyfRtECAE,3952
 industrial_model/cognite_adapters/models.py,sha256=2j2IS01uPkQEp9WdVk8seYzEqGcDdWFnpzXhusHB2zk,945
 industrial_model/cognite_adapters/optimizer.py,sha256=BbbsA3r9ZKJt6mN1wVRZc046ov-vbml9_ESfurTNRUg,2477
 industrial_model/cognite_adapters/query_mapper.py,sha256=3fEcaLsGjLKIh-g1BbMcffQ6rp99JeCW555iJo8JW44,6260
-industrial_model/cognite_adapters/query_result_mapper.py,sha256=e2W2QPDUusZh0cgpvIKF21_dP7QyZjLKrp_H_OQ4Ddk,9516
+industrial_model/cognite_adapters/query_result_mapper.py,sha256=0IhW8YVQM_EBMQM_NXlzl27CuzSUWGX4g7K9VMUVgPw,9507
 industrial_model/cognite_adapters/sort_mapper.py,sha256=RJUAYlZGXoYzK0PwX63cibRF_L-MUq9g2ZsC2EeNIF4,696
 industrial_model/cognite_adapters/upsert_mapper.py,sha256=tWEiBJQeeNz1HDu0AoBIfCw_LL156Zg4h7ORKlZ__uw,4870
-industrial_model/cognite_adapters/utils.py,sha256=rztCtS10ZTQmXfBv0nLgDiQIMWemhdSFK-SwrbRjVxM,4792
+industrial_model/cognite_adapters/utils.py,sha256=AyQxK8tDy-tAC6pji1ZzfAMsaVX2epiBhdwdCXRCPl4,4840
 industrial_model/cognite_adapters/view_mapper.py,sha256=qvT25gwIgPdAqZf-oQEh0GylUi5KEPwjLNmIl5ISQGA,1474
 industrial_model/engines/__init__.py,sha256=7aGHrUm2MxIq39vR8h0xu3i1zNOuT9H9U-q4lV3nErQ,102
-industrial_model/engines/async_engine.py,sha256=XgDKCHh4gi2mPBknp7MM4mRsuA_224NesIZ7xq1YQic,1545
-industrial_model/engines/engine.py,sha256=DaA34P1aL_qvL691OG_-g_yCC-U2ZJr9tF32Pl5vpP4,3098
+industrial_model/engines/async_engine.py,sha256=AkEHiHaO0wYiU8fwEG8H9EzNTlLpn9ygRPt-p4cQl7A,1675
+industrial_model/engines/engine.py,sha256=pZOvBbvIfa2aX2Ad8KnGntKmbZQmmvHqWzFvxEDc8xU,3224
 industrial_model/models/__init__.py,sha256=AzJ0CyPK5PvUCX45FFtybl13tkukUvk2UAF_90s_LQ8,742
 industrial_model/models/base.py,sha256=iGhDjXqA5ULEQIFHtkMi7WYJl0nQq1wi8_zqOr-Ep78,1649
 industrial_model/models/entities.py,sha256=1ZmijxOXEAWocf3HTJ9a7ci5cJEjzA3hbt0QvVP8Gb8,2817
-industrial_model/models/schemas.py,sha256=-ZnDO-yJL2PR4QmsduzDsC-rjS7CZnqPS-wYAo65Mbo,4612
+industrial_model/models/schemas.py,sha256=PXbCZTN5RHTLsufYYS3fwPcBzIBTJ48m9-hxBYXx-Yk,4625
 industrial_model/queries/__init__.py,sha256=7aheTE5qs03rxWm9fmGWptbz_p9OIXXYD8if56cqs18,227
 industrial_model/queries/models.py,sha256=iiHQ7-cfg0nukEv5PoCx9QPF-w1gVSnoNbXBOK9Mzeo,1185
 industrial_model/queries/params.py,sha256=ehgCoR5n6E-tkEuoymZ2lkLcSzMaBAx_HnyJ7sWpqz0,964
 industrial_model/statements/__init__.py,sha256=15LI9POLUdh6i3Zw0Aek0fVFfcTj8P8ZEa1Cli3pgXQ,3095
 industrial_model/statements/expressions.py,sha256=Sar1cIvy3sYi7tkWJN3ylHlZ252oN2mZJpZ1TX9jN3s,4940
-industrial_model-0.1.18.dist-info/METADATA,sha256=s9T_m788nocZAMzq-wbf1bTi9VLDW8vdCryL-DO4BtU,6144
-industrial_model-0.1.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-industrial_model-0.1.18.dist-info/RECORD,,
+industrial_model-0.1.20.dist-info/METADATA,sha256=ZkydcSIJvDKweNn0nW8NeT9AeLjxHbO9iuR8Wn_Dyr4,6419
+industrial_model-0.1.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+industrial_model-0.1.20.dist-info/RECORD,,