| 1234567891011121314151617 |
- from fastapi import Depends, HTTPException, status
- from fastapi.security import OAuth2PasswordBearer
- from .oauth_token import verify_token
- from .schemas import TokenData
- oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
- def get_current_user(token: str = Depends(oauth2_scheme)):
- payload = verify_token(token)
- username: str = payload.get("sub")
- if username is None:
- raise HTTPException(
- status_code=status.HTTP_401_UNAUTHORIZED,
- detail="Invalid authentication credentials",
- )
- return TokenData(username=username)
|
