add field to file model, add smigration script

backend-app/file/file_tests/serializers_test.py

@@ -34,6 +34,7 @@ def test_file_serializer(self):
             set(data.keys()),
             data_keys.union({
                 "rebound",
+                "rebound_project",
                 "related_duplicates",
                 'upload_date',
                 "update_date",

backend-app/file/models.py

@@ -2,6 +2,7 @@
     Model,
     CharField,
     DateTimeField,
+    IntegerField,
     BooleanField,
     ForeignKey,
     DO_NOTHING,
@@ -31,6 +32,8 @@ class File(Model):
     project: ForeignKey = ForeignKey("project.Project", on_delete=DO_NOTHING)
     author: ForeignKey = ForeignKey("user.CustomUser", on_delete=DO_NOTHING)
     rebound: ForeignKey = ForeignKey("self", on_delete=DO_NOTHING, null=True)
+    # todo: temp to remap storge bucket on file migration from project to projcet
+    rebound_project = IntegerField(null=True)
     validator: ForeignKey = ForeignKey(
         "user.CustomUser",
         null=True,

frontend-app/src/components/ui/FileMedia/index.jsx

@@ -119,11 +119,11 @@ function FileMedia({ files, slide, pathID }, ref) {
     const setFile = (token) => {
       resetZoom();
       if (!files[slide]) return;
-      var { id, file_type, rebound } = files[slide];
+      var { id, file_type, rebound, rebound_project } = files[slide];
       setTypeVideo(file_type === 'video');
 
       var baseUrl = `${getOriginDomain()}:9000/api/storage`;
-      var queryUrl = `project_${pathID}/${rebound || id}/`;
+      var queryUrl = `project_${rebound_project || pathID}/${rebound || id}/`;
 
       setFileUrl(`${baseUrl}/${queryUrl}?access=${token || tempFileToken}`);
       setMark(rebound ? "DUPLICATE" : "");

scripts/file_project_rebound.py

@@ -0,0 +1,33 @@
+from django.db import connection, transaction
+from file.models import File
+
+
+def migrate_files(
+    target_project: int,
+    dest_project: int,
+    id_mapping: list[tuple[int, int]]
+):
+    with transaction.atomic():
+        files = File.objects.filter(project_id=target_project)
+
+        assert files.update(project_id=dest_project, rebound_project=target_project), "No File to update"
+
+        update_attributes_query = """
+            update attribute_group_attribute
+            set attribute_id = %s
+            where attributegroup_id in %s and attribute_id = %s;
+        """
+
+        ag_id_list = tuple([
+            str(uid) for uid
+            in files.values_list("attributegroup__uid", flat=True).distinct()
+        ])
+
+        query_values = [
+            (dest_id, ag_id_list, target_id)
+            for target_id, dest_id
+            in id_mapping
+        ]
+
+        with connection.cursor() as cursor:
+            cursor.executemany(update_attributes_query, query_values)