industrial-model 0.1.34__tar.gz → 0.1.35__tar.gz

{industrial_model-0.1.34 → industrial_model-0.1.35}/PKG-INFO

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

{industrial_model-0.1.34 → industrial_model-0.1.35}/industrial_model/models/base.py

@@ -55,3 +55,14 @@ class RootModel(BaseModel):
                 return key
 
         return None
+
+    def get_field_alias(self, field_name_or_alias: str) -> str | None:
+        entry = self.__class__.model_fields.get(field_name_or_alias)
+        if entry:
+            return entry.alias
+
+        for _, field_info in self.__class__.model_fields.items():
+            if field_info.alias == field_name_or_alias:
+                return field_info.alias
+
+        return None

{industrial_model-0.1.34 → industrial_model-0.1.35}/industrial_model/models/entities.py

@@ -1,5 +1,5 @@
 from abc import abstractmethod
-from datetime import date, datetime
+from datetime import UTC, date, datetime
 from typing import (
     Any,
     ClassVar,
@@ -9,7 +9,7 @@ from typing import (
     TypeVar,
 )
 
-from pydantic import PrivateAttr
+from pydantic import PrivateAttr, computed_field
 
 from industrial_model.statements import Column
 
@@ -39,6 +39,18 @@ class EdgeContainer(InstanceId):
     type: InstanceId
     start_node: InstanceId
     end_node: InstanceId
+    last_updated_time: int
+    created_time: int
+
+    @computed_field  # type: ignore[prop-decorator]
+    @property
+    def created_time_as_datetime(self) -> datetime:
+        return datetime.fromtimestamp(self.created_time / 1000, tz=UTC)
+
+    @computed_field  # type: ignore[prop-decorator]
+    @property
+    def last_updated_time_as_datetime(self) -> datetime:
+        return datetime.fromtimestamp(self.last_updated_time / 1000, tz=UTC)
 
 
 TInstanceId = TypeVar("TInstanceId", bound=InstanceId)
@@ -60,6 +72,19 @@ class ViewInstance(InstanceId):
     def get_view_external_id(cls) -> str:
         return cls.view_config.get("view_external_id") or cls.__name__
 
+    def get_edge_metadata(self, property: Column | str | Any) -> list[EdgeContainer]:
+        assert isinstance(property, Column | str), (
+            f"Expected property to be Column, or str, got {type(property).__name__}"
+        )
+        identifier = property.property if isinstance(property, Column) else property
+
+        edge_map_key = self.get_field_alias(identifier) or self.get_field_name(
+            identifier
+        )
+        if not edge_map_key:
+            return []
+        return self._edges.get(edge_map_key, [])
+
     def generate_model_id(
         self,
         fields: list[str] | list[Column] | list[Any],

{industrial_model-0.1.34 → industrial_model-0.1.35}/pyproject.toml

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