include UPDATE query in migration, rename filename for issue number, re

#11396
archesproject · Aug 23, 2024 · 9012775 · 9012775
1 parent 303ba4b
commit 9012775
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/...odels/migrations/10805_nodegroup_graph.py → ...odels/migrations/11396_nodegroup_graph.py b/...odels/migrations/10805_nodegroup_graph.py → ...odels/migrations/11396_nodegroup_graph.py
@@ -10,6 +10,19 @@ class Migration(migrations.Migration):
         ("models", "10804_core_search_filters"),
     ]
 
+    sql = """
+    UPDATE node_groups
+    SET graphid = sub.graphid
+    FROM (
+        SELECT DISTINCT ng.nodegroupid, n.graphid
+        FROM nodes n
+        JOIN node_groups ng ON n.nodegroupid = ng.nodegroupid
+    ) AS sub
+    WHERE node_groups.nodegroupid = sub.nodegroupid;
+    """
+
+    reverse_sql = """"""
+
     operations = [
         migrations.AddField(
             model_name="nodegroup",
@@ -22,4 +35,8 @@ class Migration(migrations.Migration):
                 to="models.graphmodel",
             ),
         ),
+        migrations.RunSQL(
+            sql,
+            reverse_sql,
+        ),
     ]