From 7fb186e24b32cc9582f9f4484bbfd60ce874e011 Mon Sep 17 00:00:00 2001
From: zandercymatics <141044360+zandercymatics@users.noreply.github.com>
Date: Thu, 22 Aug 2024 14:11:28 -0600
Subject: [PATCH] Display logic in admin

---
 src/registrar/admin.py                 | 26 ++++++++++++++++++++++++++
 src/registrar/models/domain_request.py | 12 ++++++++++++
 2 files changed, 38 insertions(+)

diff --git a/src/registrar/admin.py b/src/registrar/admin.py
index 9e5fb71aa..995a00fde 100644
--- a/src/registrar/admin.py
+++ b/src/registrar/admin.py
@@ -7,6 +7,7 @@ from django import forms
 from django.db.models import Value, CharField, Q
 from django.db.models.functions import Concat, Coalesce
 from django.http import HttpResponseRedirect
+from django.conf import settings
 from django.shortcuts import redirect
 from django_fsm import get_available_FIELD_transitions, FSMField
 from registrar.models.domain_information import DomainInformation
@@ -306,6 +307,7 @@ class DomainRequestAdminForm(forms.ModelForm):
 
         return cleaned_data
 
+
     def _check_for_valid_rejection_reason(self, rejection_reason) -> bool:
         """
         Checks if the rejection_reason field is not none.
@@ -1914,6 +1916,19 @@ class DomainRequestAdmin(ListHeaderAdmin, ImportExportModelAdmin):
             else:
                 obj.action_needed_reason_email = default_email
 
+        if obj.status in DomainRequest.get_statuses_that_send_emails():
+            if not settings.IS_PRODUCTION:
+                profile_flag = flag_is_active(None, "profile_feature")
+                if profile_flag and hasattr(obj, "creator"):
+                    recipient = obj.creator
+                elif not profile_flag and hasattr(obj, "submitter"):
+                    recipient = obj.submitter
+                else 
+                    recipient = None
+
+                if recipient and recipient.email:
+                    self._check_for_valid_email(request, recipient.email)
+
         # == Handle status == #
         if obj.status == original_obj.status:
             # If the status hasn't changed, let the base function take care of it
@@ -1926,6 +1941,17 @@ class DomainRequestAdmin(ListHeaderAdmin, ImportExportModelAdmin):
             if should_save:
                 return super().save_model(request, obj, form, change)
 
+    def _check_for_valid_email(self, request, email):
+        """Certain emails are whitelisted in non-production environments, 
+        so we should display that information using this function.
+        
+        """
+
+        allowed = models.AllowedEmail.is_allowed_email(email)
+        error_message = f"Could not send email. The email '{email}' does not exist within the whitelist."
+        if not allowed:
+            messages.warning(request, error_message)
+
     def _handle_status_change(self, request, obj, original_obj):
         """
         Checks for various conditions when a status change is triggered.
diff --git a/src/registrar/models/domain_request.py b/src/registrar/models/domain_request.py
index 966c880d7..812f8e582 100644
--- a/src/registrar/models/domain_request.py
+++ b/src/registrar/models/domain_request.py
@@ -577,6 +577,18 @@ class DomainRequest(TimeStampedModel):
         blank=True,
     )
 
+    @classmethod
+    def get_statuses_that_send_emails(cls):
+        """Returns a list of statuses that send an email to the user"""
+        excluded_statuses = [
+            cls.DomainRequestStatus.INELIGIBLE,
+            cls.DomainRequestStatus.IN_REVIEW
+        ]
+        return [
+            status for status in cls.DomainRequestStatus
+            if status not in excluded_statuses
+        ]
+
     def sync_organization_type(self):
         """
         Updates the organization_type (without saving) to match