statikk 0.1.9__tar.gz → 0.1.10__tar.gz

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

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

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

@@ -307,15 +307,20 @@ class Table:
         subtree_copy._parent = new_parent
         for node in subtree_copy.dfs_traverse_hierarchy():
             for idx in self.indexes:
-                new_index_values = self._compose_index_values(node, idx)
-                new_sort_key_value = new_index_values[idx.sort_key.name]
-                setattr(node, idx.sort_key.name, new_sort_key_value)
-                new_hash_key_value = new_index_values[idx.hash_key.name]
-                setattr(node, idx.hash_key.name, new_hash_key_value)
+                self.build_model_indexes(node)
             setattr(node, FIELD_STATIKK_PARENT_ID, new_parent.id)
 
         return subtree_copy
 
+    def build_model_indexes(self, model: T) -> T:
+        for idx in self.indexes:
+            new_index_values = self._compose_index_values(model, idx)
+            new_sort_key_value = new_index_values[idx.sort_key.name]
+            setattr(model, idx.sort_key.name, new_sort_key_value)
+            new_hash_key_value = new_index_values[idx.hash_key.name]
+            setattr(model, idx.hash_key.name, new_hash_key_value)
+        return model
+
     def batch_write(self):
         """
         Returns a context manager for batch writing items to the database. This method handles all the buffering of the

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

@@ -217,6 +217,9 @@ class DatabaseModel(BaseModel, TrackingMixin, extra=Extra.allow):
             current = current._parent
         return False
 
+    def build_model_indexes(self) -> T:
+        return self._table.build_model_indexes(self)
+
     @classmethod
     def query(
         cls: Type[T],

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

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

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

@@ -951,3 +951,32 @@ def test_update_parent():
     hierarchy.save()
     hierarchy = Model.query_hierarchy(hash_key=Equals("model-1"))
     assert len(hierarchy.nested) == 0
+
+
+def test_rebuild_model_indexes():
+    class MyDatabaseModel(DatabaseModel):
+        foo: str = "foo"
+        bar: str = "bar"
+
+        @classmethod
+        def index_definitions(cls) -> dict[str, IndexFieldConfig]:
+            return {"main-index": IndexFieldConfig(pk_fields=["foo"], sk_fields=["bar"])}
+
+    mock_dynamodb().start()
+    table = Table(
+        name="my-dynamodb-table",
+        key_schema=KeySchema(hash_key="id"),
+        indexes=[
+            GSI(
+                name="main-index",
+                hash_key=Key(name="gsi_pk"),
+                sort_key=Key(name="gsi_sk"),
+            )
+        ],
+        models=[MyDatabaseModel],
+    )
+    _create_dynamodb_table(table)
+    my_database_model = MyDatabaseModel(foo="foo", bar="bar")
+    my_database_model.build_model_indexes()
+    assert my_database_model.gsi_pk == "foo"
+    assert my_database_model.gsi_sk == "MyDatabaseModel|bar"

{statikk-0.1.9 → statikk-0.1.10}/tests/test_models.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel
-from statikk.models import DatabaseModel
+from statikk.models import DatabaseModel, IndexFieldConfig
 
 
 class SimpleObject(BaseModel):