From 2b5942b11acc8565d2eeb505c4f7e52b9f39f6ef Mon Sep 17 00:00:00 2001 From: Birin Sanchez Date: Tue, 1 Oct 2019 17:55:53 +0200 Subject: [PATCH] Remove consumed invitations when deleting an invitee account so the Family pack owner can create a new invitation. Signed-off-by: Birin Sanchez --- limitmonitor/common.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/limitmonitor/common.py b/limitmonitor/common.py index 07c3a24..d4c7e87 100644 --- a/limitmonitor/common.py +++ b/limitmonitor/common.py @@ -8,6 +8,8 @@ from purist.limitmonitor import ServicesContainer from limitmonitor.models import ExternalBundle, Limit from limitmonitor.tunnel import TunnelManager from limitmonitor.subscription import SubscriptionManager, SubscriptionStatus +from invitation.models import Invitation + logger = logging.getLogger(__name__) @@ -167,6 +169,14 @@ def delete_account(user, purge_n=0, purge=False): # Remove WC PII user.remove_woocommerce_pii() + # If the account is of type invitee free up that invitation slot + if user.account_type == AccountType.INVITED: + try: + inv = Invitation.objects.get(guest=user) + inv.delete() + except Invitation.DoesNotExist as e: + logger.error(repr(e)) + if purge: ldap_user = user.get_ldap() ldap_user.delete() -- 2.22.0