From a43319dc02718701144128aa9f1bf5c199eaf731 Mon Sep 17 00:00:00 2001
From: sks444 <krishnasingh.ss30@gmail.com>
Date: Tue, 17 Nov 2020 11:53:50 +0530
Subject: [PATCH] Fix data migration to create default screening statuses if
 not exist

---
 ...2_migrate_screening_status_to_screening_statuses.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/hypha/apply/funds/migrations/0082_migrate_screening_status_to_screening_statuses.py b/hypha/apply/funds/migrations/0082_migrate_screening_status_to_screening_statuses.py
index fab5a3315..0f2d35214 100644
--- a/hypha/apply/funds/migrations/0082_migrate_screening_status_to_screening_statuses.py
+++ b/hypha/apply/funds/migrations/0082_migrate_screening_status_to_screening_statuses.py
@@ -10,10 +10,18 @@ def make_many_screening_statuses(apps, schema_editor):
     """
     ApplicationSubmission = apps.get_model('funds', 'ApplicationSubmission')
 
-    for submission in ApplicationSubmission.objects.all():
+    for submission in ApplicationSubmission.objects.filter(screening_status__isnull=False):
         if submission.screening_status:
             submission.screening_statuses.add(submission.screening_status)
 
+    ScreeningStatus = apps.get_model('funds', 'ScreeningStatus')
+
+    if not ScreeningStatus.objects.filter(yes=True).exists():
+        ScreeningStatus.objects.create(title='Accept', yes=True, default=True)
+
+    if not ScreeningStatus.objects.filter(yes=False).exists():
+        ScreeningStatus.objects.create(title='Dismiss', yes=False, default=True)
+
 
 class Migration(migrations.Migration):
 
-- 
GitLab