This commit is contained in:
2026-03-04 01:15:11 +08:00
parent 42a4a59277
commit 074b935a0a
2 changed files with 5 additions and 6 deletions
+3 -3
View File
@@ -10,7 +10,7 @@ from sqlalchemy.orm import selectinload
from config.settings import settings
from database.database import get_db_session
from models.database import User
from models.database import User, UserRole
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/auth/login", auto_error=False)
@@ -56,7 +56,7 @@ async def get_current_user(
raise credentials_exception
result = await db_session.execute(
select(User).where(User.username == username)
select(User).options(selectinload(User.user_roles).selectinload(UserRole.role)).where(User.username == username)
)
user = result.scalar_one_or_none()
@@ -93,7 +93,7 @@ async def get_current_admin_user(
async def authenticate_user(db_session: AsyncSession, username: str, password: str) -> Optional[User]:
result = await db_session.execute(
select(User).where(User.username == username)
select(User).options(selectinload(User.user_roles).selectinload(UserRole.role)).where(User.username == username)
)
user = result.scalar_one_or_none()