mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-14 06:55:08 +02:00
Review feedback: name changes
This commit is contained in:
parent
58977f14f2
commit
f66df2c931
3 changed files with 5 additions and 5 deletions
|
@ -10,7 +10,7 @@ from .user_domain_role import UserDomainRole
|
|||
|
||||
|
||||
class DomainInvitation(TimeStampedModel):
|
||||
SENT = "sent"
|
||||
INVITED = "invited"
|
||||
RETRIEVED = "retrieved"
|
||||
|
||||
email = models.EmailField(
|
||||
|
@ -27,17 +27,17 @@ class DomainInvitation(TimeStampedModel):
|
|||
|
||||
status = FSMField(
|
||||
choices=[
|
||||
(SENT, SENT),
|
||||
(INVITED, INVITED),
|
||||
(RETRIEVED, RETRIEVED),
|
||||
],
|
||||
default=SENT,
|
||||
default=INVITED,
|
||||
protected=True, # can't alter state except through transition methods!
|
||||
)
|
||||
|
||||
def __str__(self):
|
||||
return f"Invitation for {self.email} on {self.domain} is {self.status}"
|
||||
|
||||
@transition(field="status", source=SENT, target=RETRIEVED)
|
||||
@transition(field="status", source=INVITED, target=RETRIEVED)
|
||||
def retrieve(self):
|
||||
"""When an invitation is retrieved, create the corresponding permission."""
|
||||
|
||||
|
|
|
@ -41,7 +41,7 @@ class User(AbstractUser):
|
|||
invitations that match their email address.
|
||||
"""
|
||||
for invitation in DomainInvitation.objects.filter(
|
||||
email=self.email, status=DomainInvitation.SENT
|
||||
email=self.email, status=DomainInvitation.INVITED
|
||||
):
|
||||
invitation.retrieve()
|
||||
invitation.save()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue