fix user type

apps/account/models/user.py

@@ -83,6 +83,16 @@ class User(AbstractUser):
         return self.fullname
 
 
+    @property
+    def user_type_based_on_groups(self):
+        if self.groups.filter(name="Student Group").exists():
+            return self.UserType.STUDENT
+        elif self.groups.filter(name="Professor Group").exists():
+            return self.UserType.PROFESSOR
+        else:
+            return self.UserType.CLIENT
+        
+        
     class Meta:
         ordering = ("-id",)
         verbose_name = "All Users"

apps/account/views/user.py

@@ -168,6 +168,7 @@ class UserLoginView(CreateAPIView):
             "fullname": user.fullname,
             "email": user.email,
             "token": token.key,
+            "user_type": user.user_type_based_on_groups,
             "avatar": request.build_absolute_uri(user.avatar.url) if user.avatar else None,
         }, status=status.HTTP_201_CREATED)