guardian code

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):

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
 

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']))