industrial-model 0.1.17__tar.gz → 0.1.19__tar.gz

{industrial_model-0.1.17 → industrial_model-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: industrial-model
-Version: 0.1.17
+Version: 0.1.19
 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.17 → industrial_model-0.1.19}/README.md

@@ -209,6 +209,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.17 → industrial_model-0.1.19}/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-0.1.17 → industrial_model-0.1.19}/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-0.1.17 → industrial_model-0.1.19}/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-0.1.17 → industrial_model-0.1.19}/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-0.1.17 → industrial_model-0.1.19}/industrial_model/models/entities.py

@@ -101,5 +101,8 @@ class PaginatedResult(RootModel, Generic[TViewInstance]):
     has_next_page: bool
     next_cursor: str | None
 
+    def first_or_default(self) -> TViewInstance | None:
+        return self.data[0] if self.data else None
+
 
 ValidationMode = Literal["raiseOnError", "ignoreOnError"]

{industrial_model-0.1.17 → industrial_model-0.1.19}/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.17 → industrial_model-0.1.19}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "industrial-model"
-version = "0.1.17"
+version = "0.1.19"
 description = "Industrial Model ORM"
 readme = "README.md"
 requires-python = ">=3.11"

{industrial_model-0.1.17 → industrial_model-0.1.19}/uv.lock

@@ -215,7 +215,7 @@ wheels = [
 
 [[package]]
 name = "industrial-model"
-version = "0.1.17"
+version = "0.1.19"
 source = { editable = "." }
 dependencies = [
     { name = "anyio" },