industrial-model 0.1.34__py3-none-any.whl → 0.1.35__py3-none-any.whl

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/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.dist-info → industrial_model-0.1.35.dist-info}/METADATA

@@ -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.dist-info → industrial_model-0.1.35.dist-info}/RECORD

@@ -19,8 +19,8 @@ industrial_model/engines/__init__.py,sha256=7aGHrUm2MxIq39vR8h0xu3i1zNOuT9H9U-q4
 industrial_model/engines/async_engine.py,sha256=-sQv2vn93bBmTZOxY_C1r40YP7IMl1ND2gCN2GE9BqY,2188
 industrial_model/engines/engine.py,sha256=Pwg8A3PFVdRExiLGyhcs8hhiGyKgKi6piOmiLgvCL50,3810
 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=XugtGFCJrHnxR4UX7FWFumB2tNZGB7mNjM4skznnmdQ,4704
+industrial_model/models/base.py,sha256=qi44m0KW9KuXKTSJqSLea1poErBeAqm6Hv08VdRYHkU,2023
+industrial_model/models/entities.py,sha256=-Jf7bNCijbS3_wZohlog3bRsAkXvkUeAQW9DamJqGFo,5700
 industrial_model/models/schemas.py,sha256=EoLK9GYdS-0DQ98myTP3wOU9YpWIJOfDSLOYZUy3xEY,4559
 industrial_model/queries/__init__.py,sha256=lA83zOxMRgBgkseWJgK9kCr1vD8D8iSWs9NGGRnoYKk,355
 industrial_model/queries/models.py,sha256=VK69c4L0b0miPrKvOQBB8A01SPxZXYThrquv6gw5OGY,2544
@@ -28,6 +28,6 @@ industrial_model/queries/params.py,sha256=50qY5BO5onLsXorhcv-7qCKhJaMO94UzhKLCmZ
 industrial_model/queries/utils.py,sha256=uP6PLh9IVHDK6J8x444zHWPmyV4PkxdLO-PMc6qWItc,1505
 industrial_model/statements/__init__.py,sha256=3Et9PM7AMbIS7Fh9051Q0u3qUpX6588d1yuNrvdd4wI,5429
 industrial_model/statements/expressions.py,sha256=4ZZOcZroI5-4xRw4PXIRlufi0ARndE5zSbbxLDpR2Ec,4816
-industrial_model-0.1.34.dist-info/METADATA,sha256=9rtluwI-0-GwC1dqVlIUaCSPL8jdx6DNEVSIXhJ0qVU,6858
-industrial_model-0.1.34.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-industrial_model-0.1.34.dist-info/RECORD,,
+industrial_model-0.1.35.dist-info/METADATA,sha256=sKHvIearHLTa1i63nxbkpD2U7guTDl3FVSNPZIKDx2Q,6858
+industrial_model-0.1.35.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+industrial_model-0.1.35.dist-info/RECORD,,