Merge pull request #2723 from IFRCGo/feat/add-emergency-api-endpoint

DEP: Add emergency api endpoint

Author: susilnem

api/admin.py

@@ -26,7 +26,6 @@
 
 import api.models as models
 from api.admin_classes import RegionRestrictedAdmin
-from api.event_sources import SOURCES
 from api.management.commands.index_and_notify import Command as Notify
 from lang.admin import TranslationAdmin, TranslationInlineModelAdmin
 from notifications.models import RecordType, SubscriptionType
@@ -158,38 +157,6 @@ def queryset(self, request, queryset):
             return queryset.filter(is_featured=False)
 
 
-class EventSourceFilter(admin.SimpleListFilter):
-    title = _("source")
-    parameter_name = "event_source"
-
-    def lookups(self, request, model_admin):
-        return (
-            ("input", _("Manual input")),
-            ("gdacs", _("GDACs scraper")),
-            ("who", _("WHO scraper")),
-            ("report_ingest", _("Field report ingest")),
-            ("report_admin", _("Field report admin")),
-            ("appeal_admin", _("Appeals admin")),
-            ("unknown", _("Unknown automated")),
-        )
-
-    def queryset(self, request, queryset):
-        if self.value() == "input":
-            return queryset.filter(auto_generated=False)
-        elif self.value() == "gdacs":
-            return queryset.filter(auto_generated_source=SOURCES["gdacs"])
-        elif self.value() == "who":
-            return queryset.filter(auto_generated_source__startswith="www.who.int")
-        elif self.value() == "report_ingest":
-            return queryset.filter(auto_generated_source=SOURCES["report_ingest"])
-        elif self.value() == "report_admin":
-            return queryset.filter(auto_generated_source=SOURCES["report_admin"])
-        elif self.value() == "appeal_admin":
-            return queryset.filter(auto_generated_source=SOURCES["appeal_admin"])
-        elif self.value() == "unknown":
-            return queryset.filter(auto_generated=True).filter(auto_generated_source__isnull=True)
-
-
 class DisasterTypeAdmin(CompareVersionAdmin, TranslationAdmin, admin.ModelAdmin):
     search_fields = ("name",)
 
@@ -246,9 +213,9 @@ def level_updated_at(self, obj):
         "cc_status",
         "glide",
         "auto_generated",
-        "auto_generated_source",
+        "source",
     )
-    list_filter = [IsFeaturedFilter, EventSourceFilter]
+    list_filter = [IsFeaturedFilter, "source"]
     actions = ["create_field_reports"]
     search_fields = (
         "name",
@@ -369,7 +336,7 @@ def changeform_view(self, request, object_id=None, form_url="", extra_context=No
             self.readonly_fields = (
                 "appeals",
                 "field_reports",
-                "auto_generated_source",
+                "source",
                 "parent_event",
                 "created_at",
                 "updated_at",
@@ -378,9 +345,10 @@ def changeform_view(self, request, object_id=None, form_url="", extra_context=No
             self.readonly_fields = (
                 "appeals",
                 "field_reports",
-                "auto_generated_source",
+                "source",
                 "created_at",
                 "updated_at",
+                "who_guid",
             )
 
         # Set severity level from GET parameter
@@ -651,7 +619,7 @@ def create_events(self, request, queryset):
                 dtype=getattr(report, "dtype"),
                 disaster_start_date=getattr(report, "created_at"),
                 auto_generated=True,
-                auto_generated_source=SOURCES["report_admin"],
+                source=models.Event.EventSource.REPORT_ADMIN,
             )
             if getattr(report, "countries").exists():
                 for country in report.countries.all():
@@ -760,7 +728,7 @@ def create_events(self, request, queryset):
                 dtype=getattr(appeal, "dtype"),
                 disaster_start_date=getattr(appeal, "start_date"),
                 auto_generated=True,
-                auto_generated_source=SOURCES["appeal_admin"],
+                source=models.Event.EventSource.APPEAL_ADMIN,
             )
             if appeal.country is not None:
                 event.countries.add(appeal.country)

api/drf_views.py

@@ -89,6 +89,7 @@
     Export,
     ExternalPartner,
     FieldReport,
+    KeyFigure,
     MainContact,
     Profile,
     Region,
@@ -122,6 +123,7 @@
     CountrySupportingPartnerSerializer,
     CountryTableauSerializer,
     DeploymentsByEventSerializer,
+    DetailEmergencySerializer,
     DetailEventSerializer,
     DisasterTypeSerializer,
     DistrictSerializer,
@@ -751,7 +753,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(auto_generated_source="New field report"))
+                .filter(Q(auto_generated=False) | Q(source=Event.EventSource.NEW_REPORT))
                 .select_related("dtype")
             )
         return (
@@ -1344,7 +1346,7 @@ class SupportedActivityViewset(viewsets.ReadOnlyModelViewSet):
 #             summary=report.description or "",
 #             disaster_start_date=report.start_date,
 #             auto_generated=True,
-#             auto_generated_source=SOURCES["new_report"],
+#             source=Event.EventSource.NEW_REPORT,
 #             visibility=report.visibility,
 #             **{TRANSLATOR_ORIGINAL_LANGUAGE_FIELD_NAME: django_get_language()},
 #         )
@@ -1542,3 +1544,47 @@ class CountrySupportingPartnerViewSet(viewsets.ModelViewSet):
 
     def get_queryset(self):
         return CountrySupportingPartner.objects.select_related("country")
+
+
+class EmergencyViewset(ReadOnlyVisibilityViewset):
+    queryset = Event.objects.all()
+    lookup_field = "id"
+    serializer_class = DetailEmergencySerializer
+    filterset_class = EventFilter
+    visibility_model_class = Event
+
+    def get_queryset(self):
+        return (
+            super()
+            .get_queryset()
+            .select_related(
+                "dtype",
+                "parent_event",
+            )
+            .prefetch_related(
+                "regions",
+                "countries",
+                "countries_for_preview",
+                Prefetch("key_figures", queryset=KeyFigure.objects.all()),
+                Prefetch("contacts", queryset=EventContact.objects.all()),
+            )
+            .annotate(
+                first_field_report_id=Subquery(
+                    FieldReport.objects.filter(event=OuterRef("pk"))
+                    .order_by(
+                        "fr_num",
+                        "updated_at",
+                    )
+                    .values("id")[:1]
+                ),
+                latest_field_report_id=Subquery(
+                    FieldReport.objects.filter(event=OuterRef("pk"))
+                    .order_by(
+                        "-fr_num",
+                        "-updated_at",
+                    )
+                    .values("id")[:1]
+                ),
+                appeal_id=Subquery(Appeal.objects.filter(event=OuterRef("pk")).order_by("-created_at").values("id")[:1]),
+            )
+        )

api/enums.py

@@ -20,4 +20,5 @@
     "action_category": models.ActionCategory,
     "profile_org_types": models.Profile.OrgTypes,
     "supporting_type": models.CountrySupportingPartner.SupportingPartnerType,
+    "event_source": models.Event.EventSource,
 }

api/factories/event.py

@@ -12,6 +12,7 @@
     AppealHistory,
     AppealType,
     Event,
+    EventContact,
     EventFeaturedDocument,
     EventLink,
 )
@@ -72,7 +73,7 @@ def regions(self, create, extracted, **kwargs):
     previous_update = fuzzy.FuzzyDateTime(datetime.datetime(2008, 1, 1, tzinfo=pytz.utc))
 
     auto_generated = fuzzy.FuzzyChoice([True, False])
-    auto_generated_source = fuzzy.FuzzyText(length=50)
+    source = fuzzy.FuzzyChoice(Event.EventSource.values)
 
     is_featured = fuzzy.FuzzyChoice([True, False])
     is_featured_region = fuzzy.FuzzyChoice([True, False])
@@ -133,3 +134,9 @@ class AppealHistoryFactory(factory.django.DjangoModelFactory):
 
     class Meta:
         model = AppealHistory
+
+
+class EventContactFactory(factory.django.DjangoModelFactory):
+
+    class Meta:
+        model = EventContact

api/filter_set.py

@@ -2,7 +2,6 @@
 from django.contrib.auth.models import User
 from django.db import models
 
-from api.event_sources import SOURCES
 from api.models import (
     Admin2,
     Appeal,
@@ -182,10 +181,8 @@ class EventFilter(filters.FilterSet):
     countries__in = ListFilter(field_name="countries__id")
     regions__in = ListFilter(field_name="regions__id")
     id = filters.NumberFilter(field_name="id", lookup_expr="exact")
-    auto_generated_source = filters.ChoiceFilter(
-        label="Auto generated source choices",
-        choices=[(v, v) for v in SOURCES.values()],
-    )
+    source = filters.ChoiceFilter(choices=Event.EventSource.choices, lookup_expr="exact")
+
     is_subscribed = filters.BooleanFilter(label="is_subscribed", method="get_is_subcribed_event")
 
     class Meta:

api/management/commands/create_events.py

@@ -1,6 +1,5 @@
 from django.core.management.base import BaseCommand
 
-from api.event_sources import SOURCES
 from api.models import Appeal, Event
 
 
@@ -18,7 +17,7 @@ def handle(self, *args, **options):
                 "dtype": appeal.dtype,
                 "disaster_start_date": appeal.start_date,
                 "auto_generated": True,
-                "auto_generated_source": SOURCES["appeal_admin"],
+                "source": Event.EventSource.APPEAL_ADMIN,
             }
             event = Event.objects.create(**fields)
             if appeal.country is not None:

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(
-            auto_generated_source="New field report"
+            source=Event.EventSource.NEW_REPORT
         )  # exclude those events that were generated from field reports, to avoid 2x notif.
         condE = Q(status=CronJobStatus.ERRONEOUS)
 

api/management/commands/ingest_gdacs.py

@@ -3,7 +3,6 @@
 from dateutil.parser import parse
 from django.core.management.base import BaseCommand
 
-from api.event_sources import SOURCES
 from api.logger import logger
 from api.models import Country, CronJob, CronJobStatus, DisasterType, Event, GDACSEvent
 
@@ -104,7 +103,7 @@ def handle(self, *args, **options):
                         "summary": data["description"],
                         "disaster_start_date": data["publication_date"],
                         "auto_generated": True,
-                        "auto_generated_source": SOURCES["gdacs"],
+                        "source": Event.EventSource.GDACS,
                         "ifrc_severity_level": data["alert_level"],
                     }
                     event = Event.objects.create(**fields)

api/management/commands/ingest_mdb.py

@@ -13,7 +13,6 @@
 from django.core.management.base import BaseCommand
 from django.utils import timezone
 
-from api.event_sources import SOURCES
 from api.fixtures.dtype_map import PK_MAP
 from api.logger import logger
 from api.models import (
@@ -249,7 +248,7 @@ def handle(self, *args, **options):
                 "dtype": report_dtype,
                 "disaster_start_date": datetime.utcnow().replace(tzinfo=timezone.utc),
                 "auto_generated": True,
-                "auto_generated_source": SOURCES["report_ingest"],
+                "source": Event.EventSource.REPORT_INGEST,
             }
             event = Event(**event_record)
             event.save()

api/management/commands/ingest_who.py

@@ -11,7 +11,8 @@ class Command(BaseCommand):
     help = "Add new event (=emergency) entries from WHO API"
 
     def handle(self, *args, **options):
-        guids = [e.auto_generated_source for e in Event.objects.filter(auto_generated_source__startswith="www.who.int")]
+        # Change this at the top
+        guids = list(Event.objects.filter(source=Event.EventSource.WHO).values_list("who_guid", flat=True))
 
         logger.info("Querying WHO RSS feed for new emergency data")
         # get latest
@@ -48,10 +49,15 @@ def handle(self, *args, **options):
                     "category": row.pop("category"),
                     "pubDate": row.pop("pubDate"),
                 }
-                if data["guid"] in guids:
-                    continue
                 if data["guid"] in ["WeDontWantThis", "NeitherThis"]:
                     continue
+                try:
+                    who_guid = int(data["guid"].strip().split(".")[-1])
+                except (ValueError, IndexError):
+                    who_guid = None
+
+                if who_guid is None or who_guid in guids:
+                    continue
 
                 title = data["title"]  # for csr link
                 short = title.replace(" (ex-China)", "")
@@ -131,7 +137,8 @@ def handle(self, *args, **options):
                     "summary": summary,
                     "disaster_start_date": date,
                     "auto_generated": True,
-                    "auto_generated_source": data["guid"],
+                    "source": Event.EventSource.WHO,
+                    "who_guid": who_guid,
                     "ifrc_severity_level": alert_level,
                 }
                 # TODO: fields['name'] sometimes exceeds 100 maxlength, so will need some altering if this will be used