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)