pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Invitation revoking & role disabling to base class

Browse Source
This commit is contained in:
graham-dds 2019-10-28 13:27:57 -04:00
parent ccaabcaab0
commit eb22d5ec1e
1 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
atst/domain/invitations.py
View File

@ -114,7 +114,9 @@ class BaseInvitations(object):
@classmethod @classmethod
def revoke(cls, token): def revoke(cls, token):
invite = cls._get(token) invite = cls._get(token)
return cls._update_status(invite, InvitationStatus.REVOKED) invite = cls._update_status(invite, InvitationStatus.REVOKED)
cls.role_domain_class.disable(invite.role)
return invite
@classmethod @classmethod
def resend(cls, inviter, token, user_info=None): def resend(cls, inviter, token, user_info=None):
@ -138,19 +140,7 @@ class PortfolioInvitations(BaseInvitations):
model = PortfolioInvitation model = PortfolioInvitation
role_domain_class = PortfolioRoles role_domain_class = PortfolioRoles
@classmethod
def revoke(cls, token):
invite = super().revoke(token)
PortfolioRoles.disable(invite.role)
return invite
class ApplicationInvitations(BaseInvitations): class ApplicationInvitations(BaseInvitations):
model = ApplicationInvitation model = ApplicationInvitation
role_domain_class = ApplicationRoles role_domain_class = ApplicationRoles
@classmethod
def revoke(cls, token):
invite = super().revoke(token)
ApplicationRoles.disable(invite.role)
return invite