From a791022ff1311e1fb76930c398d6ff91036d0456 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 11 十二月 2024 17:57:52 +0800
Subject: [PATCH] 新增加标签功能

---
 app/api/auth.py |    9 ++++++---
 1 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/app/api/auth.py b/app/api/auth.py
index f850882..ba86a89 100644
--- a/app/api/auth.py
+++ b/app/api/auth.py
@@ -16,7 +16,7 @@
 from app.models.user import UserCreate, LoginData
 from app.models.user_model import UserModel
 from app.service.auth import authenticate_user, create_access_token, is_valid_password, save_register_user, \
-    update_user_token, UserAppDao
+    update_user_token, UserAppDao, update_user_info
 from app.service.bisheng import BishengService
 from app.service.v2.app_register import AppRegisterDao
 from app.service.difyService import DifyService
@@ -83,13 +83,13 @@
     access_token = create_access_token(data={"sub": user.username, "user_id": user.id})
 
     upsert_token(db, user.id, access_token, bisheng_token, ragflow_token)
-
+    # print(111)
     return Response(code=200, msg="Login successful", data={
         "access_token": access_token,
         "token_type": "bearer",
         "username": user.username,
         "nickname": "",
-        "user": user.to_login_json()
+        # "user": user.to_login_json()
     })
 
 
@@ -120,9 +120,12 @@
             logger.error("鏈煡娉ㄥ唽搴旂敤---")
             continue
         try:
+            name = login_data.username
             user_app = await UserAppDao(db).get_data_by_id(user.id, app["id"])
             if user_app:
                 name  = user_app.username
+            else:
+                await update_user_info(db, user.id)
             token = await service.login(name, login_data.password)
             token_dict[app["id"]] = token
         except Exception as e:

--
Gitblit v1.8.0