chat2api/utils/authorization.py
2024-04-12 18:33:17 +08:00

18 lines
535 B
Python

from fastapi import Depends, HTTPException
from fastapi.security import OAuth2PasswordBearer
from utils.config import authorization_list
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token", auto_error=False)
def verify_token(token: str = Depends(oauth2_scheme)):
if not authorization_list:
return token
elif token in authorization_list:
return token
elif token and token.startswith("eyJhbGciOi"):
return token
else:
raise HTTPException(status_code=401, detail="Not authenticated")