fix: update migration scripts to work with the events implementation

Author: Angeluz-07

commons/data_access_layer/cosmos_db.py

@@ -220,7 +220,8 @@ def delete(self, id: str, event_context: EventContext,
             'deleted': generate_uuid4()
         }, event_context, peeker=peeker, visible_only=True, mapper=mapper)
 
-    def delete_permanently(self, id: str, partition_key_value: str) -> None:
+    def delete_permanently(self, id: str, event_context: EventContext) -> None:
+        partition_key_value = self.find_partition_key_value(event_context)
         self.container.delete_item(id, partition_key_value)
 
     def find_partition_key_value(self, event_context: EventContext):

commons/data_access_layer/database.py

@@ -35,13 +35,15 @@ def delete(self, id):
 
 class EventContext():
     def __init__(self, container_id: str, action: str, description: str = None,
-                 user_id: str = None, tenant_id: str = None, session_id: str = None):
+                 user_id: str = None, tenant_id: str = None, session_id: str = None,
+                 app_id: str = None):
         self._container_id = container_id
         self._action = action
         self._description = description
         self._user_id = user_id
         self._tenant_id = tenant_id
         self._session_id = session_id
+        self._app_id = app_id
 
     @property
     def container_id(self):
@@ -66,3 +68,7 @@ def tenant_id(self):
     @property
     def session_id(self):
         return self._session_id
+
+    @property
+    def app_id(self):
+        return self._app_id

migrations/__init__.py

@@ -2,40 +2,14 @@
 from migrate_anything import configure
 from migrate_anything.storage import Storage
 
-from time_tracker_api import create_app
-
+from commons.data_access_layer.database import EventContext
+from commons.data_access_layer.cosmos_db import cosmos_helper, init_app, \
+    CosmosDBRepository
 
-class CustomStorage(object):
-    def __init__(self, file):
-        self.file = file
-
-    def save_migration(self, name, code):
-        with open(self.file, "a", encoding="utf-8") as file:
-            file.write("{},{}\n".format(name, code))
-
-    def list_migrations(self):
-        try:
-            with open(self.file, encoding="utf-8") as file:
-                return [
-                    line.split(",")
-                    for line in file.readlines()
-                    if line.strip()  # Skip empty lines
-                ]
-        except FileNotFoundError:
-            return []
-
-    def remove_migration(self, name):
-        migrations = [
-            migration for migration in self.list_migrations() if migration[0] != name
-        ]
-
-        with open(self.file, "w", encoding="utf-8") as file:
-            for row in migrations:
-                file.write("{},{}\n".format(*row))
+from time_tracker_api import create_app
 
 
 app = create_app('time_tracker_api.config.CLIConfig')
-from commons.data_access_layer.cosmos_db import cosmos_helper, init_app, CosmosDBRepository
 
 if cosmos_helper is None:
     init_app(app)
@@ -59,19 +33,38 @@ def __init__(self, collection_id, app_id):
         self.repository = CosmosDBRepository.from_definition(migrations_definition)
 
     def save_migration(self, name, code):
-        self.repository.create({"id": name,
-                                "name": name,
-                                "code": code,
-                                "app_id": self.app_id})
+        event_ctx = self.create_event_context('create')
+        self.repository.create(
+            data={
+                "id": name,
+                "name": name,
+                "code": code,
+                "app_id": self.app_id
+            },
+            event_context=event_ctx,
+        )
 
     def list_migrations(self):
-        migrations = self.repository.find_all(self.app_id)
+        event_ctx = self.create_event_context('read-many')
+        migrations = self.repository.find_all(event_context=event_ctx)
         return [
             [item['name'], item['code']] for item in migrations
         ]
 
     def remove_migration(self, name):
-        self.repository.delete_permanently(name, self.app_id)
-
+        event_ctx = self.create_event_context('delete-permanently')
+        self.repository.delete_permanently(id=name, event_context=event_ctx)
+
+    def create_event_context(
+        self,
+        action: str = None,
+        description: str = None
+    ) -> EventContext:
+        return EventContext(
+            container_id=self.collection_id,
+            action=action,
+            description=description,
+            app_id=self.app_id
+        )
 
 configure(storage=CosmosDBStorage("migration", "time-tracker-api"))