statikk 0.1.8__tar.gz → 0.1.9__tar.gz

{statikk-0.1.8 → statikk-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: statikk
-Version: 0.1.8
+Version: 0.1.9
 Summary: statikk is a single table application (STA) library for DynamoDb.
 Home-page: https://github.com/terinia/statikk
 Author: Balint Biro

{statikk-0.1.8 → statikk-0.1.9}/src/statikk/engine.py

@@ -1,6 +1,6 @@
 import os
 from datetime import datetime
-from typing import Any, Dict, Type, Optional, List, Union
+from typing import Any, Dict, Type, Optional, List, Union, get_type_hints, get_origin, get_args
 
 import boto3
 from botocore.config import Config
@@ -623,6 +623,26 @@ class Table:
                     data = self._serialize_item(enriched_item)
                     batch.delete_item(Key=data)
 
+    def inspect_optional_field(self, model_class, field_name):
+        field_type = model_class.model_fields[field_name].annotation
+
+        is_optional = False
+        inner_type = field_type
+
+        if get_origin(field_type) is Union:
+            args = get_args(field_type)
+            if len(args) == 2 and args[1] is type(None):
+                is_optional = True
+                inner_type = args[0]
+
+        elif hasattr(field_type, "__origin__") and field_type.__origin__ is Union:
+            args = getattr(field_type, "__args__", [])
+            if len(args) == 2 and args[1] is type(None):
+                is_optional = True
+                inner_type = args[0]
+
+        return (is_optional, inner_type)
+
     def reconstruct_hierarchy(self, items: list[dict]) -> Optional[dict]:
         """
         Reconstructs a hierarchical dictionary structure from a flat list of dictionaries
@@ -704,7 +724,9 @@ class Table:
                 if field_name.startswith("_"):
                     continue
 
-                field_type = field_info.annotation
+                is_optional, inner_type = self.inspect_optional_field(parent_model_class, field_name)
+
+                field_type = inner_type if is_optional else field_info.annotation
 
                 if field_type == child_model_class:
                     matching_fields.append((field_name, "single"))

{statikk-0.1.8 → statikk-0.1.9}/src/statikk/models.py

@@ -204,8 +204,19 @@ class DatabaseModel(BaseModel, TrackingMixin, extra=Extra.allow):
 
     @property
     def should_delete(self) -> bool:
+        if self._is_any_parent_marked_for_deletion():
+            return True
+
         return self._should_delete or self.is_parent_changed()
 
+    def _is_any_parent_marked_for_deletion(self) -> bool:
+        current = self._parent
+        while current is not None:
+            if current._should_delete:
+                return True
+            current = current._parent
+        return False
+
     @classmethod
     def query(
         cls: Type[T],

{statikk-0.1.8 → statikk-0.1.9}/src/statikk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: statikk
-Version: 0.1.8
+Version: 0.1.9
 Summary: statikk is a single table application (STA) library for DynamoDb.
 Home-page: https://github.com/terinia/statikk
 Author: Balint Biro

{statikk-0.1.8 → statikk-0.1.9}/tests/test_engine.py

@@ -775,8 +775,20 @@ def test_nested_raw_models():
 
 
 def test_nested_hierarchies():
+    class OptionalModel(DatabaseModel):
+        xax: str = "xax"
+
+        @classmethod
+        def index_definitions(cls) -> dict[str, IndexFieldConfig]:
+            return {"main-index": IndexFieldConfig(sk_fields=["xax"])}
+
+        @classmethod
+        def is_nested(cls) -> bool:
+            return True
+
     class TriplyNested(DatabaseModel):
         faz: str = "faz"
+        optional: Optional[OptionalModel] = None
 
         @classmethod
         def index_definitions(cls) -> dict[str, IndexFieldConfig]:
@@ -833,10 +845,10 @@ def test_nested_hierarchies():
                 sort_key=Key(name="gsi_sk"),
             )
         ],
-        models=[ModelHierarchy, NestedModel, DoublyNestedModel, TriplyNested],
+        models=[ModelHierarchy, NestedModel, DoublyNestedModel, TriplyNested, OptionalModel],
     )
     _create_dynamodb_table(table)
-    triple_nested_model = TriplyNested(faz="faz")
+    triple_nested_model = TriplyNested(faz="faz", optional=OptionalModel(xax="xax"))
     doubly_nested = DoublyNestedModel(bar="bar", items=[triple_nested_model])
     double_nested_no_write = DoublyNestedModel(bar="far", items=[TriplyNested(faz="faz")])
     nested = NestedModel(foo="foo", doubly_nested=[doubly_nested, double_nested_no_write])
@@ -851,6 +863,15 @@ def test_nested_hierarchies():
     assert hierarchy.nested.doubly_nested[0].gsi_pk == "foo_id"
     assert hierarchy.nested.doubly_nested[0].gsi_sk == "ModelHierarchy|state|NestedModel|foo|DoublyNestedModel|bar"
     assert hierarchy.nested.doubly_nested[0].items[0].gsi_pk == "foo_id"
+    assert (
+        hierarchy.nested.doubly_nested[0].items[0].gsi_sk
+        == "ModelHierarchy|state|NestedModel|foo|DoublyNestedModel|bar|TriplyNested|faz"
+    )
+    assert hierarchy.nested.doubly_nested[0].items[0].optional.gsi_pk == "foo_id"
+    assert (
+        hierarchy.nested.doubly_nested[0].items[0].optional.gsi_sk
+        == "ModelHierarchy|state|NestedModel|foo|DoublyNestedModel|bar|TriplyNested|faz|OptionalModel|xax"
+    )
     hierarchy.nested.doubly_nested[0].items[0].mark_for_delete()
     hierarchy.save()
     hierarchy = ModelHierarchy.query_hierarchy(hash_key=Equals("foo_id"))