diff --git a/src/registrar/admin.py b/src/registrar/admin.py index 40d4befb5..1b62f74d6 100644 --- a/src/registrar/admin.py +++ b/src/registrar/admin.py @@ -3579,6 +3579,14 @@ class WaffleFlagAdmin(FlagAdmin): model = models.WaffleFlag fields = "__all__" + # Hack to get the dns_prototype_flag to auto populate when you navigate to + # the waffle flag page. + def changelist_view(self, request, extra_context=None): + if extra_context is None: + extra_context = {} + extra_context["dns_prototype_flag"] = flag_is_active_for_user(request.user, "dns_prototype_flag") + return super().changelist_view(request, extra_context=extra_context) + class DomainGroupAdmin(ListHeaderAdmin, ImportExportModelAdmin): list_display = ["name", "portfolio"] diff --git a/src/registrar/views/domain.py b/src/registrar/views/domain.py index 851a7c266..34aac37f5 100644 --- a/src/registrar/views/domain.py +++ b/src/registrar/views/domain.py @@ -459,15 +459,7 @@ class DomainDNSView(DomainBaseView): def get_context_data(self, **kwargs): """Adds custom context.""" context = super().get_context_data(**kwargs) - object = self.get_object() context["dns_prototype_flag"] = flag_is_active_for_user(self.request.user, "dns_prototype_flag") - - context["is_valid_domain_for_prototype"] = True - x = object.name == "igorville.gov" - print(f"what is the object? {object.name} and equal? {x}") - if settings.IS_PRODUCTION: - context["is_valid_domain_for_prototype"] = object.name == "igorville.gov" - return context