zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-23 19:20:47 +02:00
Code Issues Projects Releases Packages Wiki Activity

changes

Browse source
This commit is contained in:
asaki222 2024-10-31 14:55:17 -04:00
parent 963edc1c05
commit cb771f6208
No known key found for this signature in database
GPG key ID: 2C4F802060E06EA4
2 changed files with 18 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/registrar/views/__init__.py
View file

@ -12,6 +12,7 @@ from .domain import (
DomainUsersView,
DomainAddUserView,
DomainInvitationDeleteView,
DomainInvitationCancelView,
DomainDeleteUserView,
)
from .user_profile import UserProfileView, FinishProfileSetupView

34
src/registrar/views/domain.py
View file

@ -1054,26 +1054,26 @@ class DomainAddUserView(DomainFormBaseView):
# # The order of the superclasses matters here. BaseDeleteView has a bug where the
# # "form_valid" function does not call super, so it cannot use SuccessMessageMixin.
# # The workaround is to use SuccessMessageMixin first.
# class DomainInvitationDeleteView(SuccessMessageMixin, DomainInvitationPermissionDeleteView):
# object: DomainInvitation # workaround for type mismatch in DeleteView
class DomainInvitationDeleteView(SuccessMessageMixin, DomainInvitationPermissionDeleteView):
object: DomainInvitation # workaround for type mismatch in DeleteView
# def post(self, request, *args, **kwargs):
# """Override post method in order to error in the case when the
# domain invitation status is RETRIEVED"""
# self.object = self.get_object()
# form = self.get_form()
# if form.is_valid() and self.object.status == self.object.DomainInvitationStatus.INVITED:
# return self.form_valid(form)
# else:
# # Produce an error message if the domain invatation status is RETRIEVED
# messages.error(request, f"Invitation to {self.object.email} has already been retrieved.")
# return HttpResponseRedirect(self.get_success_url())
def post(self, request, *args, **kwargs):
"""Override post method in order to error in the case when the
domain invitation status is RETRIEVED"""
self.object = self.get_object()
form = self.get_form()
if form.is_valid() and self.object.status == self.object.DomainInvitationStatus.INVITED:
return self.form_valid(form)
else:
# Produce an error message if the domain invatation status is RETRIEVED
messages.error(request, f"Invitation to {self.object.email} has already been retrieved.")
return HttpResponseRedirect(self.get_success_url())
# def get_success_url(self):
# return reverse("domain-users", kwargs={"pk": self.object.domain.id})
def get_success_url(self):
return reverse("domain-users", kwargs={"pk": self.object.domain.id})
# def get_success_message(self, cleaned_data):
# return f"Canceled invitation to {self.object.email}."
def get_success_message(self, cleaned_data):
return f"Canceled invitation to {self.object.email}."
class DomainInvitationCancelView(SuccessMessageMixin, DomainInvitationUpdateView):