issue #1177 - Give failed runs another go

SACGF · Oct 8, 2024 · e429b8f · e429b8f
1 parent 7ec9def
commit e429b8f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/snpdb/migrations/0151_one_off_fix_csv_export_quoting.py b/snpdb/migrations/0151_one_off_fix_csv_export_quoting.py
@@ -5,6 +5,12 @@
 from manual.operations.manual_operations import ManualOperation
 
 
+def _delete_failed_csv_exports(apps, _schema_editor):
+    CachedGeneratedFile = apps.get_model("snpdb", "CachedGeneratedFile")
+    AFFECTED_GENERATORS = ["export_sample_to_downloadable_file", "export_cohort_to_downloadable_file"]
+    CachedGeneratedFile.objects.filter(generator__in=AFFECTED_GENERATORS, exception__contains='NoneType').delete()
+
+
 def _test_for_csv_export(apps):
     CachedGeneratedFile = apps.get_model("snpdb", "CachedGeneratedFile")
 
@@ -19,6 +25,7 @@ class Migration(migrations.Migration):
     ]
 
     operations = [
+        migrations.RunPython(_delete_failed_csv_exports),
         ManualOperation(task_id=ManualOperation.task_id_manage(["one_off_fix_csv_export_quoting"]),
                         test=_test_for_csv_export)
     ]