Merge pull request #2592 from IFRCGo/feat/add-simplified-eap-to-global-pdf-export

 Add simplified eap to global pdf export

Author: susilnem

api/migrations/0227_alter_export_export_type.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.19 on 2025-11-18 05:22
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0226_nsdinitiativescategory_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='export',
+            name='export_type',
+            field=models.CharField(choices=[('dref-applications', 'DREF Application'), ('dref-operational-updates', 'DREF Operational Update'), ('dref-final-reports', 'DREF Final Report'), ('old-dref-final-reports', 'Old DREF Final Report'), ('per', 'Per'), ('simplified-eap', 'Simplified EAP')], max_length=255, verbose_name='Export Type'),
+        ),
+    ]

api/models.py

@@ -2560,6 +2560,7 @@ class ExportType(models.TextChoices):
         FINAL_REPORT = "dref-final-reports", _("DREF Final Report")
         OLD_FINAL_REPORT = "old-dref-final-reports", _("Old DREF Final Report")
         PER = "per", _("Per")
+        SIMPLIFIED_EAP = "simplified-eap", _("Simplified EAP")
 
     export_id = models.IntegerField(verbose_name=_("Export Id"))
     export_type = models.CharField(verbose_name=_("Export Type"), max_length=255, choices=ExportType.choices)

api/serializers.py

@@ -15,6 +15,7 @@
 from api.utils import CountryValidator, RegionValidator
 from deployments.models import EmergencyProject, Personnel, PersonnelDeployment
 from dref.models import Dref, DrefFinalReport, DrefOperationalUpdate
+from eap.models import SimplifiedEAP
 from lang.models import String
 from lang.serializers import ModelSerializer
 from local_units.models import DelegationOffice
@@ -2566,6 +2567,11 @@ def create(self, validated_data):
         elif export_type == Export.ExportType.PER:
             overview = Overview.objects.filter(id=export_id).first()
             title = f"{overview.country.name}-preparedness-{overview.get_phase_display()}"
+        elif export_type == Export.ExportType.SIMPLIFIED_EAP:
+            simplified_eap = SimplifiedEAP.objects.filter(id=export_id).first()
+            title = (
+                f"{simplified_eap.eap_registration.national_society.name}-{simplified_eap.eap_registration.disaster_type.name}"
+            )
         else:
             title = "Export"
         user = self.context["request"].user

api/tasks.py

@@ -104,6 +104,8 @@ def generate_url(url, export_id, user, title):
                 page.wait_for_selector("#pdf-preview-ready", state="attached", timeout=timeout)
                 if export.export_type == Export.ExportType.PER:
                     file_name = f'PER {title} ({datetime.now().strftime("%Y-%m-%d %H-%M-%S")}).pdf'
+                elif export.export_type == Export.ExportType.SIMPLIFIED_EAP:
+                    file_name = f'SIMPLIFIED EAP {title} ({datetime.now().strftime("%Y-%m-%d %H-%M-%S")}).pdf'
                 else:
                     file_name = f'DREF {title} ({datetime.now().strftime("%Y-%m-%d %H-%M-%S")}).pdf'
                 file = ContentFile(

eap/test_views.py

@@ -1,12 +1,14 @@
 import os
 import tempfile
+from unittest import mock
 
 from django.conf import settings
 from django.contrib.auth.models import Group, Permission
 from django.core import management
 
 from api.factories.country import CountryFactory
 from api.factories.disaster_type import DisasterTypeFactory
+from api.models import Export
 from deployments.factories.user import UserFactory
 from eap.factories import (
     EAPRegistrationFactory,
@@ -1151,3 +1153,64 @@ def test_status_transition(self):
         # Check is the activated timeline is added
         self.eap_registration.refresh_from_db()
         self.assertIsNotNone(self.eap_registration.activated_at)
+
+
+class TestSimplifiedEapPdfExport(APITestCase):
+    def setUp(self):
+        super().setUp()
+        self.country = CountryFactory.create(name="country1", iso3="XXX")
+        self.national_society = CountryFactory.create(name="national_society1", iso3="YYY")
+        self.disaster_type = DisasterTypeFactory.create(name="disaster1")
+        self.partner1 = CountryFactory.create(name="partner1", iso3="ZZZ")
+        self.partner2 = CountryFactory.create(name="partner2", iso3="AAA")
+
+        self.user = UserFactory.create()
+
+        self.eap_registration = EAPRegistrationFactory.create(
+            eap_type=EAPType.SIMPLIFIED_EAP,
+            country=self.country,
+            national_society=self.national_society,
+            disaster_type=self.disaster_type,
+            partners=[self.partner1.id, self.partner2.id],
+            created_by=self.user,
+            modified_by=self.user,
+        )
+
+        self.simplified_eap = SimplifiedEAPFactory.create(
+            eap_registration=self.eap_registration,
+            created_by=self.user,
+            modified_by=self.user,
+            national_society_contact_title="NS Title Example",
+        )
+        self.url = "/api/v2/pdf-export/"
+
+    @mock.patch("api.serializers.generate_url.delay")
+    def test_create_simplified_eap_export(self, mock_generate_url):
+        data = {
+            "export_type": Export.ExportType.SIMPLIFIED_EAP,
+            "export_id": self.simplified_eap.id,
+            "is_pga": False,
+        }
+
+        self.authenticate(self.user)
+
+        with self.capture_on_commit_callbacks(execute=True):
+            response = self.client.post(self.url, data, format="json")
+        self.assert_201(response)
+        export = Export.objects.first()
+        self.assertIsNotNone(export)
+
+        expected_url = (
+            f"{settings.GO_WEB_INTERNAL_URL}/" f"{Export.ExportType.SIMPLIFIED_EAP}/" f"{self.simplified_eap.id}/export/"
+        )
+        self.assertEqual(export.url, expected_url)
+        self.assertEqual(response.data["status"], Export.ExportStatus.PENDING)
+
+        self.assertEqual(mock_generate_url.called, True)
+        title = f"{self.national_society.name}-{self.disaster_type.name}"
+        mock_generate_url.assert_called_once_with(
+            export.url,
+            export.id,
+            self.user.id,
+            title,
+        )