Warning message citizenship

programs/admin.py

@@ -141,6 +141,7 @@ class WarningMessageAdmin(ModelAdmin):
     filter_horizontal = (
         "programs",
         "counties",
+        "legal_statuses",
     )
 
     def get_str(self, obj):

programs/migrations/0100_warningmessage_legal_statuses.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.15 on 2024-12-19 22:04
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("programs", "0099_merge_20241108_1521"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="warningmessage",
+            name="legal_statuses",
+            field=models.ManyToManyField(blank=True, related_name="warning_messages", to="programs.legalstatus"),
+        ),
+    ]

programs/models.py

@@ -976,6 +976,7 @@ class WarningMessage(models.Model):
     external_name = models.CharField(max_length=120, blank=True, null=True, unique=True)
     calculator = models.CharField(max_length=120, blank=False, null=False)
     counties = models.ManyToManyField(County, related_name="warning_messages", blank=True)
+    legal_statuses = models.ManyToManyField(LegalStatus, related_name="warning_messages", blank=True)
 
     message = models.ForeignKey(
         Translation, related_name="warning_messages", blank=False, null=False, on_delete=models.PROTECT

screener/serializers.py

@@ -1,8 +1,9 @@
-from datetime import datetime, timedelta, date
+from datetime import datetime, timedelta
+from programs.models import WarningMessage
 from screener.models import Screen, HouseholdMember, IncomeStream, Expense, Message, Insurance, WhiteLabel
 from authentication.serializers import UserOffersSerializer
 from rest_framework import serializers
-from translations.serializers import TranslationSerializer
+from translations.serializers import ModelTranslationSerializer, TranslationSerializer
 from validations.serializers import ValidationSerializer
 
 
@@ -257,6 +258,18 @@ class NavigatorSerializer(serializers.Serializer):
     languages = serializers.ListField()
 
 
+class WarningMessageSerializer(serializers.ModelSerializer):
+    message = ModelTranslationSerializer()
+    legal_statuses = serializers.SerializerMethodField()
+
+    class Meta:
+        model = WarningMessage
+        fields = ("message", "legal_statuses")
+
+    def get_legal_statuses(self, obj: WarningMessage):
+        return [m.status for m in obj.legal_statuses.all()]
+
+
 class EligibilitySerializer(serializers.Serializer):
     description_short = TranslationSerializer()
     name = TranslationSerializer()
@@ -281,7 +294,7 @@ class EligibilitySerializer(serializers.Serializer):
     documents = TranslationSerializer(many=True)
     multiple_tax_units = serializers.BooleanField()
     estimated_value_override = TranslationSerializer()
-    warning_messages = TranslationSerializer(many=True)
+    warning_messages = WarningMessageSerializer(many=True)
 
     class Meta:
         fields = "__all__"

screener/views.py

@@ -24,6 +24,7 @@
     EligibilitySerializer,
     MessageSerializer,
     ResultsSerializer,
+    WarningMessageSerializer,
 )
 from programs.programs.policyengine.policy_engine import calc_pe_eligibility
 from programs.util import DependencyError, Dependencies
@@ -214,6 +215,7 @@ def eligibility_results(screen: Screen, batch=False):
             *translations_prefetch_name("documents__", Document.objects.translated_fields),
             "warning_messages",
             "warning_messages__counties",
+            "warning_messages__legal_statuses",
             *translations_prefetch_name("warning_messages__", WarningMessage.objects.translated_fields),
             "translation_overrides",
             "translation_overrides__counties",
@@ -340,7 +342,7 @@ def sort_first(program):
                 )
 
                 if warning_calculator.calc():
-                    warnings.append(warning)
+                    warnings.append(WarningMessageSerializer(warning).data)
 
         if not skip and program.active:
             legal_status = [status.status for status in program.legal_status_required.all()]
@@ -386,7 +388,7 @@ def sort_first(program):
                     "new": new,
                     "low_confidence": program.low_confidence,
                     "documents": [serialized_document(document) for document in program.documents.all()],
-                    "warning_messages": [default_message(w.message) for w in warnings],
+                    "warning_messages": warnings,
                 }
             )