industrial-model 0.1.14__tar.gz → 0.1.16__tar.gz

{industrial_model-0.1.14 → industrial_model-0.1.16}/PKG-INFO

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

{industrial_model-0.1.14 → industrial_model-0.1.16}/industrial_model/__init__.py

@@ -4,6 +4,7 @@ from .models import (
     AggregatedViewInstance,
     InstanceId,
     PaginatedResult,
+    RootModel,
     TAggregatedViewInstance,
     TViewInstance,
     TWritableViewInstance,
@@ -32,6 +33,7 @@ __all__ = [
     "Engine",
     "AsyncEngine",
     "PaginatedResult",
+    "RootModel",
     "ViewInstanceConfig",
     "WritableViewInstance",
 ]

{industrial_model-0.1.14 → industrial_model-0.1.16}/industrial_model/models/schemas.py

@@ -57,17 +57,21 @@ def get_parent_and_children_nodes(
 
 
 def _get_type_properties(
-    cls: type[BaseModel], visited_count: defaultdict[type, int] | None = None
+    cls: type[BaseModel],
+    parent_type: type[BaseModel] | None = None,
+    visited_count: defaultdict[type, int] | None = None,
 ) -> dict[str, Any] | None:
     if visited_count is not None:
-        if visited_count[cls] > 2:
+        if visited_count[cls] > 1:
             return None
-        visited_count[cls] += 1
+        elif parent_type == cls:
+            visited_count[cls] += 1
 
     hints = get_type_hints(cls)
     origins = {
         key: _get_field_type(
             type_hint,
+            cls,
             visited_count or defaultdict(lambda: 0),
         )
         for key, type_hint in hints.items()
@@ -81,13 +85,15 @@ def _get_type_properties(
 
 
 def _get_field_type(
-    type_hint: type, visited_count: defaultdict[type, int]
+    type_hint: type,
+    parent_type: type[BaseModel],
+    visited_count: defaultdict[type, int],
 ) -> tuple[bool, dict[str, Any] | None]:
     should_iter = _type_is_list_or_union(type_hint)
 
     if not should_iter:
         return _get_field_relations(
-            [_cast_base_model(type_hint)], visited_count
+            [_cast_base_model(type_hint)], parent_type, visited_count
         )
 
     entries: list[type[BaseModel] | None] = []
@@ -96,11 +102,12 @@ def _get_field_type(
             return _get_field_type(arg, visited_count)
         entries.append(_cast_base_model(arg))
 
-    return _get_field_relations(entries, visited_count)
+    return _get_field_relations(entries, parent_type, visited_count)
 
 
 def _get_field_relations(
     entries: list[type[TBaseModel] | None],
+    parent_type: type[BaseModel],
     visited_count: defaultdict[type, int],
 ) -> tuple[bool, dict[str, Any] | None]:
     entry_type = next((type_ for type_ in entries if type_ is not None), None)
@@ -108,7 +115,7 @@ def _get_field_relations(
     if not entry_type:
         return False, None
 
-    properties = _get_type_properties(entry_type, visited_count)
+    properties = _get_type_properties(entry_type, parent_type, visited_count)
 
     return True, properties
 

{industrial_model-0.1.14 → industrial_model-0.1.16}/pyproject.toml

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

{industrial_model-0.1.14 → industrial_model-0.1.16}/uv.lock

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