refactor(database): migrate to sqlmodel

This commit is contained in:
MingxuanGame
2025-07-24 20:49:07 +08:00
parent 1655bb9f53
commit c43ca883a5
11 changed files with 582 additions and 743 deletions

View File

@@ -1,14 +1,16 @@
from fastapi import Depends, HTTPException
from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
from sqlalchemy.orm import Session
from __future__ import annotations
from app.auth import get_token_by_access_token
from .database import get_db
from app.database import (
User as DBUser,
)
from .database import get_db
from fastapi import Depends, HTTPException
from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
from sqlmodel import Session, select
security = HTTPBearer()
@@ -29,5 +31,5 @@ async def get_current_user_by_token(token: str, db: Session) -> DBUser | None:
token_record = get_token_by_access_token(db, token)
if not token_record:
return None
user = db.query(DBUser).filter(DBUser.id == token_record.user_id).first()
user = db.exec(select(DBUser).where(DBUser.id == token_record.user_id)).first()
return user