feat(emergency): Add new stage for appeal dref type

- update on stage logic generation
- rename enums of the EventSource
- update test cases
- add source field from auto generated source on event
- generate new migrations for the emergency, dref and deployments

Author: susilnem

api/admin.py

@@ -619,7 +619,7 @@ def create_events(self, request, queryset):
                 dtype=getattr(report, "dtype"),
                 disaster_start_date=getattr(report, "created_at"),
                 auto_generated=True,
-                source=models.Event.EventSource.REPORT_ADMIN,
+                source=models.Event.EventSource.FIELD_REPORT_ADMIN,
             )
             if getattr(report, "countries").exists():
                 for country in report.countries.all():

api/drf_views.py

@@ -89,6 +89,7 @@
     Event,
     EventContact,
     EventFeaturedDocument,
+    EventLink,
     EventSeverityLevelHistory,
     EventStage,
     Export,
@@ -759,7 +760,7 @@ def get_queryset(self, *args, **kwargs):
         if self.action == "response_activity_events":
             return (
                 qset.filter(parent_event__isnull=True)
-                .filter(Q(auto_generated=False) | Q(source=Event.EventSource.NEW_REPORT))
+                .filter(Q(auto_generated=False) | Q(source=Event.EventSource.NEW_FIELD_REPORT))
                 .select_related("dtype")
             )
         return (
@@ -1361,7 +1362,7 @@ class SupportedActivityViewset(viewsets.ReadOnlyModelViewSet):
 #             summary=report.description or "",
 #             disaster_start_date=report.start_date,
 #             auto_generated=True,
-#             source=Event.EventSource.NEW_REPORT,
+#             source=Event.EventSource.NEW_FIELD_REPORT,
 #             visibility=report.visibility,
 #             **{TRANSLATOR_ORIGINAL_LANGUAGE_FIELD_NAME: django_get_language()},
 #         )
@@ -1658,10 +1659,10 @@ def get_queryset(self):
                     # If there is an active appeal of DREF type, but no approved DREF yet,
                     # we consider the emergency to be in the Emergency Appeal stage.
                     # Reaches here only if no approved DREF/ops-update/final-report exists
-                    # So an active appeal type DREF appeal with no approved DREF = Emergency Appeal
+                    # So an active appeal type DREF appeal with no approved DREF = DREF_APPEAL_ONLY stage.
                     When(
                         Exists(active_dref_appeal_qs),
-                        then=Value(EventStage.EMERGENCY_APPEAL),
+                        then=Value(EventStage.DREF_APPEAL_ONLY),
                     ),
                     When(
                         Exists(FieldReport.objects.filter(event=OuterRef("pk"))),
@@ -1676,7 +1677,10 @@ def get_queryset(self):
                 # to avoid extra queries in serializer.
                 stage_appeal_id=Case(
                     When(
-                        stage=EventStage.EMERGENCY_APPEAL,
+                        stage__in=[
+                            EventStage.EMERGENCY_APPEAL,
+                            EventStage.DREF_APPEAL_ONLY,
+                        ],
                         then=Subquery(latest_appeal_qs.values("id")[:1]),
                     ),
                     default=Value(None),
@@ -1737,6 +1741,10 @@ def get_queryset(self):
                     "countries",
                     queryset=Country.objects.select_related("region"),
                 ),
+                Prefetch(
+                    "districts",
+                    queryset=District.objects.select_related("country"),
+                ),
                 Prefetch(
                     "key_figures",
                     queryset=KeyFigure.objects.all(),
@@ -1745,5 +1753,13 @@ def get_queryset(self):
                     "contacts",
                     queryset=EventContact.objects.all(),
                 ),
+                Prefetch(
+                    "links",
+                    queryset=EventLink.objects.all(),
+                ),
+                Prefetch(
+                    "featured_documents",
+                    queryset=EventFeaturedDocument.objects.order_by("-id"),
+                ),
             )
         )

api/enums.py

@@ -21,4 +21,5 @@
     "profile_org_types": models.Profile.OrgTypes,
     "supporting_type": models.CountrySupportingPartner.SupportingPartnerType,
     "event_source": models.Event.EventSource,
+    "emergency_stage": models.EventStage,
 }

api/management/commands/index_and_notify.py

@@ -1035,7 +1035,7 @@ def handle(self, *args, **options):
         condR = Q(real_data_update__gte=time_diff)  # instead of modified at
         cond2 = ~Q(previous_update__gte=time_diff_1_day)  # negate (~) no previous_update in the last day, so send once a day
         condF = Q(
-            source=Event.EventSource.NEW_REPORT
+            source=Event.EventSource.NEW_FIELD_REPORT
         )  # exclude those events that were generated from field reports, to avoid 2x notif.
         condE = Q(status=CronJobStatus.ERRONEOUS)
 

api/management/commands/ingest_mdb.py

@@ -248,7 +248,7 @@ def handle(self, *args, **options):
                 "dtype": report_dtype,
                 "disaster_start_date": datetime.utcnow().replace(tzinfo=timezone.utc),
                 "auto_generated": True,
-                "source": Event.EventSource.REPORT_INGEST,
+                "source": Event.EventSource.FIELD_REPORT_DMIS_INGEST,
             }
             event = Event(**event_record)
             event.save()

api/management/commands/migrate_event_source.py

@@ -1,27 +1,93 @@
-# Generated by Django 4.2.29 on 2026-05-12 08:28
+# Generated by Django 4.2.30 on 2026-05-21 04:49
 
 from django.db import migrations, models
 
 
-class Migration(migrations.Migration):
+def migrate_sources(apps, _):
+    """
+    Populate the source field for Event records:
+    - Non auto generated events are mapped to MANUAL_INPUT.
+    - Events that are auto generated and have auto_generated_source are mapped to the corresponding enum value.
+    - Events that are auto generated but do not have auto_generated_source are mapped to MANUAL_INPUT.
+    - For WHO, the who_guid field is populated with the id extracted from the auto_generated_source field.
+    """
+
+    Event = apps.get_model("api", "Event")
+
+    for obj in Event.objects.iterator():
+        update_fields = ["source"]
+
+        if not obj.auto_generated:
+            source = 100  # MANUAL_INPUT
+
+        else:
+            raw_source = (obj.auto_generated_source or "").lower()
+            if "gdacs" in raw_source:
+                source = 110 # GDACS
+
+            elif "who.int" in raw_source:
+                source = 120
+
+                parts = raw_source.strip().split(".")
+                who_id = parts[-1].strip()
 
+                try:
+                    obj.who_guid = int(who_id)
+                    update_fields.append("who_guid")
+                except (ValueError, TypeError):
+                    pass
+            elif "field report dmis ingest" in raw_source:
+                source = 130 # FIELD_REPORT_DMIS_INGEST
+            elif "field report admin" in raw_source:
+                source = 140 # FIELD_REPORT_ADMIN
+            elif "appeal" in raw_source:
+                source = 150 # APPEAL_ADMIN
+            elif "new field report" in raw_source:
+                source = 160 # NEW_FIELD_REPORT
+            else:
+                # Using title for backward compatibility, as some events have auto_generated_source empty but title containing "GDACS"
+                if obj.title and obj.title.lower().startswith("gdacs"):
+                    source = 110  # GDACS
+                else:
+                    source = 100  # MANUAL_INPUT
+                    obj.auto_generated = False
+                    update_fields.append("auto_generated")
+        obj.source = source
+        obj.save(update_fields=update_fields)
+
+
+class Migration(migrations.Migration):
     dependencies = [
-        ('api', '0231_alter_export_export_type'),
+        ("api", "0231_alter_export_export_type"),
     ]
 
     operations = [
-        migrations.RemoveField(
-            model_name='event',
-            name='auto_generated_source',
-        ),
         migrations.AddField(
-            model_name='event',
-            name='source',
-            field=models.IntegerField(choices=[(100, 'Manual input'), (110, 'GDACs scraper'), (120, 'WHO scraper'), (130, 'Field report DMIS ingest'), (140, 'Field report admin'), (150, 'Appeal admin'), (160, 'New field report'), (170, 'DREF')], default=100, verbose_name='Event source'),
+            model_name="event",
+            name="source",
+            field=models.IntegerField(
+                choices=[
+                    (100, "Manual input"),
+                    (110, "GDACs scraper"),
+                    (120, "WHO scraper"),
+                    (130, "Field report DMIS ingest"),
+                    (140, "Field report admin"),
+                    (150, "Appeal admin"),
+                    (160, "New field report"),
+                    (170, "DREF"),
+                ],
+                default=100,
+                verbose_name="Event source",
+            ),
         ),
         migrations.AddField(
-            model_name='event',
-            name='who_guid',
-            field=models.IntegerField(blank=True, null=True, verbose_name='Who guid'),
+            model_name="event",
+            name="who_guid",
+            field=models.IntegerField(blank=True, null=True, verbose_name="Who guid"),
+        ),
+        migrations.RunPython(migrate_sources, reverse_code=migrations.RunPython.noop),
+        migrations.RemoveField(
+            model_name="event",
+            name="auto_generated_source",
         ),
     ]

api/migrations/0232_remove_event_auto_generated_source_event_source_and_more.py

@@ -768,6 +768,7 @@ class EventStage(models.IntegerChoices):
     DREF_OPERATIONAL_UPDATE = 3, _("DREF Operational Update")
     DREF_FINAL_REPORT = 4, _("DREF Final Report")
     FIELD_REPORT = 5, _("Field Report")
+    DREF_APPEAL_ONLY = 6, _("DREF")
 
 
 # NOTE: If ever in future we need to create an api to update the event table
@@ -778,7 +779,6 @@ class Event(models.Model):
 
     class EventSource(models.IntegerChoices):
 
-        # TODO(Susilnem): use upper case
         MANUAL_INPUT = 100, _("Manual input")
         """MANUAL_INPUT: Event data manually entered by a user through the event administration interface."""
 
@@ -788,17 +788,17 @@ class EventSource(models.IntegerChoices):
         WHO = 120, _("WHO scraper")
         """WHO: Event data automatically ingested from the (WHO) scraper."""
 
-        REPORT_INGEST = 130, _("Field report DMIS ingest")
-        """REPORT_INGEST: Event data imported through the DMIS field report."""
+        FIELD_REPORT_DMIS_INGEST = 130, _("Field report DMIS ingest")
+        """FIELD_REPORT_DMIS_INGEST: Event data imported through the DMIS field report."""
 
-        REPORT_ADMIN = 140, _("Field report admin")
-        """REPORT_ADMIN: Event data created or modified via the field report administration interface."""
+        FIELD_REPORT_ADMIN = 140, _("Field report admin")
+        """FIELD_REPORT_ADMIN: Event data created or modified via the field report administration interface."""
 
         APPEAL_ADMIN = 150, _("Appeal admin")
         """APPEAL_ADMIN: Event data created or managed through the appeal administration interface."""
 
-        NEW_REPORT = 160, _("New field report")
-        """NEW_REPORT: Event data originating from newly created field reports."""
+        NEW_FIELD_REPORT = 160, _("New field report")
+        """NEW_FIELD_REPORT: Event data originating from newly created field reports."""
 
         DREF = 170, _("DREF")
         """DREF: Event originating records."""