From 08c8e8c9a4d65677de6a493446a605d70efee631 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 10 十二月 2024 16:32:07 +0800
Subject: [PATCH] 12.10 16
---
app/models/token_model.py | 9 ++++++---
1 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/app/models/token_model.py b/app/models/token_model.py
index 2b50bfd..fad57e0 100644
--- a/app/models/token_model.py
+++ b/app/models/token_model.py
@@ -7,6 +7,7 @@
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):
@@ -59,7 +60,7 @@
if not isinstance(user_id, int) or user_id <= 0:
return
db_token = None
- print(token)
+ # print(token)
try:
# 鏌ヨ鐜版湁璁板綍
db_token = db.query(TokenModel).filter_by(user_id=user_id).first()
@@ -88,5 +89,7 @@
db.rollback() # 鍥炴粴浜嬪姟
-def get_token(db: Session, user_id: int) -> Type[TokenModel] | None:
- return db.query(TokenModel).filter_by(user_id=user_id).first()
+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