From 2f84e3cc4bff981590ab37bd95da64bd72c1f2e0 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期一, 16 十二月 2024 09:40:00 +0800 Subject: [PATCH] 12-16 --- app/models/token_model.py | 13 ++++--------- 1 files changed, 4 insertions(+), 9 deletions(-) diff --git a/app/models/token_model.py b/app/models/token_model.py index fad57e0..43dd2d0 100644 --- a/app/models/token_model.py +++ b/app/models/token_model.py @@ -5,9 +5,7 @@ from sqlalchemy.orm import Session from Log import logger -from app.config.const import RAGFLOW from app.models.base_model import Base -from app.service.auth import UserAppDao class TokenModel(Base): @@ -67,16 +65,16 @@ if db_token: # 璁板綍瀛樺湪锛岃繘琛屾洿鏂� db_token.token = access_token - for k, v in token.items(): - setattr(db_token, k.replace("app", "token"), v) + # for k, v in token.items(): + # setattr(db_token, k.replace("app", "token"), v) else: # 璁板綍涓嶅瓨鍦紝杩涜鎻掑叆 db_token = TokenModel( user_id=user_id, token=access_token, ) - for k, v in token.items(): - setattr(db_token, k.replace("app", "token"), v) + # for k, v in token.items(): + # setattr(db_token, k.replace("app", "token"), v) db.add(db_token) # 鎻愪氦浜嬪姟 @@ -89,7 +87,4 @@ db.rollback() # 鍥炴粴浜嬪姟 -async def get_token(db: Session, user_id: int): - # return db.query(TokenModel).filter_by(user_id=user_id).first() - return {i.app_type.replace("app", "token"): i.access_token for i in await UserAppDao(db).get_user_datas(user_id)} -- Gitblit v1.8.0