Disable PortfolioRole when revoking an invite

2019-10-28 13:14:39 -04:00
parent 7c0179a108
commit a0bccc64a9
1 changed files with 6 additions and 0 deletions
--- a/atst/domain/invitations.py
+++ b/atst/domain/invitations.py
@@ -138,6 +138,12 @@ class PortfolioInvitations(BaseInvitations):
    model = PortfolioInvitation
    role_domain_class = PortfolioRoles

+    @classmethod
+    def revoke(cls, token):
+        invite = super().revoke(token)
+        PortfolioRoles.disable(invite.role)
+        return invite
+

 class ApplicationInvitations(BaseInvitations):
    model = ApplicationInvitation