26 lines
872 B
Python
26 lines
872 B
Python
 
|
from fastapi import Depends, HTTPException, status
|
||
|
|
from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
|
||
|
|
from sqlalchemy import select
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from app.auth.security import decode_access_token
|
||
|
|
from app.db.models import User
|
||
|
|
from app.db.session import get_session
|
||
|
|
|
||
|
|
_bearer = HTTPBearer()
|
||
|
|
|
||
|
|
|
||
|
|
async def get_current_user(
|
||
|
|
credentials: HTTPAuthorizationCredentials = Depends(_bearer),
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> User:
|
||
|
|
try:
|
||
|
|
user_id = decode_access_token(credentials.credentials)
|
||
|
|
except Exception as exc:
|
||
|
|
raise HTTPException(status.HTTP_401_UNAUTHORIZED, "invalid token") from exc
|
||
|
|
|
||
|
|
user = await session.scalar(select(User).where(User.id == user_id))
|
||
|
|
if user is None:
|
||
|
|
raise HTTPException(status.HTTP_401_UNAUTHORIZED, "user not found")
|
||
|
|
return user
|