sync data

zhaoqingang

2024-12-16 88360b4ac6f051f62a91e93d602fd393935071ab

 app/models/token_model.py

@@ -4,10 +4,8 @@
from sqlalchemy import Column, Integer, DateTime, Text
from sqlalchemy.orm import Session

from Log import logger
from app.config.const import RAGFLOW
# from Log import logger
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)

        # 提交事务
@@ -84,14 +82,9 @@
        db.refresh(db_token)

    except Exception as e:
        logger.error(e)
        # logger.error(e)
        # 异常处理
        db.rollback()  # 回滚事务


async def get_token(db: Session, user_id: int):
    res = {i.app_type.replace("app", "token"): i.access_token for i in await UserAppDao(db).get_user_datas(user_id)}
    if not res:
        token = db.query(TokenModel).filter_by(user_id=user_id).first()
        res = {"ragflow_token": token.ragflow_token, "bisheng_token": token.bisheng_token}
    return res