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