问题优化

zhaoqingang

2024-12-13 f2c43713b16e821f4a1ae97804c60f18734d5f6a

 app/models/token_model.py

@@ -5,7 +5,6 @@
from sqlalchemy.orm import Session

from Log import logger
from app.config.const import RAGFLOW
from app.models.base_model import Base


@@ -59,23 +58,23 @@
    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()
        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)

        # 提交事务
@@ -88,5 +87,4 @@
        db.rollback()  # 回滚事务


def get_token(db: Session, user_id: int) -> Type[TokenModel] | None:
    return db.query(TokenModel).filter_by(user_id=user_id).first()