Display long-run warning for migration 0050 only…

where appropriate
kobotoolbox · May 24, 2023 · 9d9ec11 · 9d9ec11
1 parent 7646a69
commit 9d9ec11
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/kpi/migrations/0050_add_indexes_to_import_and_export_tasks.py b/kpi/migrations/0050_add_indexes_to_import_and_export_tasks.py
@@ -47,6 +47,18 @@ def manually_drop_indexes_instructions(apps, schema_editor):
         """
     )
 
+def warning_long_run(apps, schema_editor):
+    print(
+        """
+        This might take a while. If it is too slow, you may want to
+        interrupt this migration, cancel any outstanding `CREATE…` or `DROP
+        INDEX` queries on `kpi_importtask` and `kpi_exporttask`, re-run the
+        migration with `SKIP_HEAVY_MIGRATIONS=True`, and then follow the
+        printed instructions to set up the indexes concurrently (without
+        downtime) using raw SQL.
+        """
+    )
+
 
 class Migration(migrations.Migration):
 
@@ -55,17 +67,8 @@ class Migration(migrations.Migration):
     ]
 
     if not settings.SKIP_HEAVY_MIGRATIONS:
-        print(
-            """
-            This might take a while. If it is too slow, you may want to
-            interrupt this migration, cancel any outstanding `CREATE…` or `DROP
-            INDEX` queries on `kpi_importtask` and `kpi_exporttask`, re-run the
-            migration with `SKIP_HEAVY_MIGRATIONS=True`, and then follow the
-            printed instructions to set up the indexes concurrently (without
-            downtime) using raw SQL.
-            """
-        )
         operations = [
+            migrations.RunPython(warning_long_run, warning_long_run),
             migrations.AddIndex(
                 model_name='importtask',
                 index=django.contrib.postgres.indexes.BTreeIndex(