Add yes and default field to screening status

hypha/apply/funds/admin.py

@@ -113,6 +113,7 @@ class ScreeningStatusPermissionHelper(PermissionHelper):
 class ScreeningStatusAdmin(ModelAdmin):
     model = ScreeningStatus
     menu_icon = 'tag'
+    list_display = ('title', 'yes', 'default')
     permission_helper_class = ScreeningStatusPermissionHelper
 
 

hypha/apply/funds/admin_forms.py

 from collections import Counter
 
-from wagtail.admin.forms import WagtailAdminPageForm
+from django.apps import apps
+from wagtail.admin.forms import WagtailAdminModelForm, WagtailAdminPageForm
 
 from .workflow import WORKFLOWS
 
@@ -82,3 +83,20 @@ class RoundBasePageAdminForm(WagtailAdminPageForm):
             self.add_error('start_date', 'Please select start date.')
 
         return cleaned_data
+
+
+class ScreeningStatusAdminForm(WagtailAdminModelForm):
+
+    def clean(self):
+        cleaned_data = super().clean()
+        default = cleaned_data['default']
+        yes = cleaned_data['yes']
+        ScreeningStatus = apps.get_model('funds', 'ScreeningStatus')
+        default_screening = ScreeningStatus.objects.filter(
+            default=True, yes=yes
+        ).first()
+        if default:
+            # Can't set two defaults for yes/no.
+            default_screening.default = False
+            default_screening.save()
+        return cleaned_data

hypha/apply/funds/migrations/0080_add_yes_and_default_fields_to_screening_status.py

+# Generated by Django 2.2.16 on 2020-11-02 06:21
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('funds', '0079_add_reviewer_settings_for_submission_access'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='screeningstatus',
+            name='default',
+            field=models.BooleanField(default=False, help_text='Only one Yes and No screening status can be set as default.', verbose_name='Default Yes/No'),
+        ),
+        migrations.AddField(
+            model_name='screeningstatus',
+            name='yes',
+            field=models.BooleanField(default=False, help_text='Tick mark for Yes otherwise No.', verbose_name='Yes/No'),
+        ),
+    ]

hypha/apply/funds/models/screening.py

 from django.db import models
 
+from ..admin_forms import ScreeningStatusAdminForm
+
 
 class ScreeningStatus(models.Model):
     title = models.CharField(max_length=128)
+    yes = models.BooleanField(
+        default=False, verbose_name="Yes/No",
+        help_text='Tick mark for Yes otherwise No.'
+    )
+    default = models.BooleanField(
+        default=False, verbose_name="Default Yes/No",
+        help_text='Only one Yes and No screening status can be set as default.'
+    )
+
+    base_form_class = ScreeningStatusAdminForm
 
     class Meta:
         verbose_name_plural = "screening statuses"