From ce42e48b377635693a58ffd2e6bbdecccf4e4e71 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Mon, 4 Sep 2023 17:48:44 +0800
Subject: [PATCH] fix login api

---
 app/api/endpoints/login.py | 3 +++
 app/schemas/token.py       | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/app/api/endpoints/login.py b/app/api/endpoints/login.py
index a24d0438..4f484e86 100644
--- a/app/api/endpoints/login.py
+++ b/app/api/endpoints/login.py
@@ -56,6 +56,9 @@ async def login_access_token(
             user.id, expires_delta=access_token_expires
         ),
         token_type="bearer",
+        super_user=user.is_superuser,
+        user_name=user.name,
+        avatar=user.avatar
     )
 
 
diff --git a/app/schemas/token.py b/app/schemas/token.py
index b1dbf230..488f73eb 100644
--- a/app/schemas/token.py
+++ b/app/schemas/token.py
@@ -6,6 +6,9 @@ from pydantic import BaseModel
 class Token(BaseModel):
     access_token: str
     token_type: str
+    super_user: bool
+    user_name: str
+    avatar: Optional[str] = None
 
 
 class TokenPayload(BaseModel):