diff --git a/account/utils.py b/account/utils.py index 4dc6acc..357f2d1 100644 --- a/account/utils.py +++ b/account/utils.py @@ -23,7 +23,7 @@ from guardian.models import Guardian from account.models import UserDelete from base.messages import ERROR_CODE from django.utils import timezone - +from base.constants import NUMBER # Define delete # user account condition, # Define delete @@ -91,6 +91,7 @@ def junior_account_update(user_tb): junior_data.is_active = False junior_data.is_verified = False junior_data.guardian_code = '{}' + junior_data.guardian_code_status = str(NUMBER['1']) junior_data.save() def guardian_account_update(user_tb): diff --git a/junior/serializers.py b/junior/serializers.py index 003e563..0f014fe 100644 --- a/junior/serializers.py +++ b/junior/serializers.py @@ -319,6 +319,7 @@ class RemoveJuniorSerializer(serializers.ModelSerializer): if instance: instance.is_invited = False instance.guardian_code = '{}' + instance.guardian_code_status = str(NUMBER['1']) instance.save() return instance diff --git a/junior/views.py b/junior/views.py index b41d2aa..471db0f 100644 --- a/junior/views.py +++ b/junior/views.py @@ -185,6 +185,7 @@ class AddJuniorAPIView(viewsets.ModelViewSet): junior = Junior.objects.filter(auth=user).first() guardian = Guardian.objects.filter(user=self.request.user).first() junior.guardian_code = [guardian.guardian_code] + junior.guardian_code_status = str(NUMBER['two']) junior.save() JuniorGuardianRelationship.objects.get_or_create(guardian=guardian, junior=junior, relationship=str(self.request.data['relationship']))