statikk 0.1.6__tar.gz → 0.1.7__tar.gz

{statikk-0.1.6 → statikk-0.1.7}/PKG-INFO

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

{statikk-0.1.6 → statikk-0.1.7}/src/statikk/engine.py

@@ -232,9 +232,13 @@ class Table:
 
         Returns the enriched database model instance.
         """
+
         with self.batch_write() as batch:
             for item in model.split_to_simple_objects():
-                batch.put(item)
+                if item._should_delete:
+                    batch.delete(item)
+                else:
+                    batch.put(item)
 
     def update_item(
         self,

{statikk-0.1.6 → statikk-0.1.7}/src/statikk/models.py

@@ -162,6 +162,7 @@ class DatabaseModel(BaseModel, TrackingMixin, extra=Extra.allow):
     id: str = Field(default_factory=lambda: str(uuid4()))
     _parent: Optional[DatabaseModel] = None
     _model_types_in_hierarchy: dict[str, Type[DatabaseModel]] = {}
+    _should_delete: bool = False
 
     @classmethod
     def type(cls) -> str:
@@ -256,6 +257,9 @@ class DatabaseModel(BaseModel, TrackingMixin, extra=Extra.allow):
             return self._parent.should_write_to_database()
         return True
 
+    def mark_for_delete(self):
+        self._should_delete = True
+
     @classmethod
     def scan(
         cls,

{statikk-0.1.6 → statikk-0.1.7}/src/statikk.egg-info/PKG-INFO

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

{statikk-0.1.6 → statikk-0.1.7}/tests/test_engine.py

@@ -788,7 +788,7 @@ def test_nested_hierarchies():
 
     class DoublyNestedModel(DatabaseModel):
         bar: str
-        items: list[TriplyNested]
+        items: list[TriplyNested] = []
 
         @classmethod
         def index_definitions(cls) -> dict[str, IndexFieldConfig]:
@@ -803,7 +803,7 @@ def test_nested_hierarchies():
 
     class NestedModel(DatabaseModel):
         foo: str
-        doubly_nested: list[DoublyNestedModel]
+        doubly_nested: list[DoublyNestedModel] = []
 
         @classmethod
         def index_definitions(cls) -> dict[str, IndexFieldConfig]:
@@ -836,7 +836,8 @@ def test_nested_hierarchies():
         models=[ModelHierarchy, NestedModel, DoublyNestedModel, TriplyNested],
     )
     _create_dynamodb_table(table)
-    doubly_nested = DoublyNestedModel(bar="bar", items=[TriplyNested(faz="faz")])
+    triple_nested_model = TriplyNested(faz="faz")
+    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])
     model_hierarchy = ModelHierarchy(foo_id="foo_id", state="state", nested=nested)
@@ -850,5 +851,9 @@ 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"
+    hierarchy.nested.doubly_nested[0].items[0].mark_for_delete()
+    hierarchy.save()
+    hierarchy = ModelHierarchy.query_hierarchy(hash_key=Equals("foo_id"))
+    assert len(hierarchy.nested.doubly_nested[0].items) == 0
     hierarchy.delete()
     assert list(table.scan()) == []