zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-24 03:30:50 +02:00
Code Issues Projects Releases Packages Wiki Activity

Go back to using get_or_create

Browse source
This commit is contained in:
Neil Martinsen-Burrell 2023-04-04 15:34:51 -05:00
parent 63979afe32
commit 57f3428025
No known key found for this signature in database
GPG key ID: 6A3C818CC10D0184
1 changed files with 4 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/registrar/models/domain_invitation.py
View file

@ -62,15 +62,10 @@ class DomainInvitation(TimeStampedModel):
)
# and create a role for that user on this domain
try:
UserDomainRole.objects.get(
user=user, domain=self.domain, role=UserDomainRole.Roles.ADMIN
)
except UserDomainRole.DoesNotExist:
UserDomainRole.objects.create(
user=user, domain=self.domain, role=UserDomainRole.Roles.ADMIN
)
else:
_, created = UserDomainRole.objects.get_or_create(
user=user, domain=self.domain, role=UserDomainRole.Roles.ADMIN
)
if not created:
# something strange happened and this role already existed when
# the invitation was retrieved. Log that this occurred.
logger.warn(