From fd46f49c204c28aa55f2188dec92f6f69b054114 Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Mon, 3 Mar 2025 13:43:44 +0800
Subject: [PATCH] fix

---
 account/api.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/account/api.py b/account/api.py
index 0c40b7f..e339e54 100644
--- a/account/api.py
+++ b/account/api.py
@@ -2,7 +2,7 @@ from django.contrib.auth import authenticate, login, logout
 from ninja import Router
 from ninja.errors import HttpError
 from .schemas import UserRegistrationSchema, UserLoginSchema
-from .models import User
+from .models import RoleChoices, User
 
 router = Router()
 
@@ -29,7 +29,7 @@ def user_login(request, payload: UserLoginSchema):
     user = authenticate(username=payload.username, password=payload.password)
     if user is not None:
         login(request, user)
-        return request.user.get_username()
+        return {"username": user.username, "role": user.role}
     else:
         raise HttpError(401, "Invalid credentials")
 
@@ -42,4 +42,7 @@ def user_logout(request):
 @router.get("/profile")
 def current_user_profile(request):
     # 暂时这样写
-    return request.user.get_username()
+    if request.user.is_authenticated:
+        return {"username": request.user.get_username(), "role": request.user.role}
+    else:
+        return {"username": "", "role": RoleChoices.NORMAL}