From 4343779abf7efcb011c2a4befccac53379eb2ce7 Mon Sep 17 00:00:00 2001
From: alireza <alireza@mortezaei.com>
Date: Sun, 16 Feb 2025 11:00:25 +0330
Subject: [PATCH] fix user type

---
 apps/account/models/user.py | 10 ++++++++++
 apps/account/views/user.py  |  1 +
 2 files changed, 11 insertions(+)

diff --git a/apps/account/models/user.py b/apps/account/models/user.py
index f4c34fd..c50434d 100644
--- a/apps/account/models/user.py
+++ b/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"
diff --git a/apps/account/views/user.py b/apps/account/views/user.py
index cb16290..ac54b61 100644
--- a/apps/account/views/user.py
+++ b/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)