修改返回内容

zhangxiao

2024-10-23 db8c55d218c4952aa010cb1a4c5be623a1e4efe1

 app/models/token_model.py

@@ -1,6 +1,7 @@
from datetime import datetime
from typing import Type

from sqlalchemy import Column, Integer, String, DateTime, Text
from sqlalchemy import Column, Integer, DateTime, Text
from sqlalchemy.orm import Session

from app.models.base_model import Base
@@ -49,3 +50,7 @@
    except Exception as e:
        # 异常处理
        db.rollback()  # 回滚事务


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