From 1aefca21f00d3d9fbee3119ae009da4d629126ff Mon Sep 17 00:00:00 2001 From: zandercymatics <141044360+zandercymatics@users.noreply.github.com> Date: Thu, 14 Mar 2024 12:37:02 -0600 Subject: [PATCH] Use "is_production" instead --- src/registrar/views/utility/permission_views.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/registrar/views/utility/permission_views.py b/src/registrar/views/utility/permission_views.py index f086747f8..fde01253e 100644 --- a/src/registrar/views/utility/permission_views.py +++ b/src/registrar/views/utility/permission_views.py @@ -2,8 +2,8 @@ import abc # abstract base class -from django.conf import settings from django.views.generic import DetailView, DeleteView, TemplateView +from registrar.context_processors import is_production from registrar.models import Domain, DomainRequest, DomainInvitation from registrar.models.user_domain_role import UserDomainRole @@ -39,7 +39,8 @@ class DomainPermissionView(DomainPermission, DetailView, abc.ABC): """ if "IS_PRODUCTION" not in request.session: # Pass the production flag to the context - request.session["IS_PRODUCTION"] = settings.IS_PRODUCTION + production_status = is_production(request) + request.session.update(production_status) return super().dispatch(request, *args, **kwargs) # Adds context information for user permissions