mirror of https://github.com/tiangolo/fastapi.git
27 lines
869 B
Python
27 lines
869 B
Python
import secrets
|
|
|
|
from fastapi import Depends, FastAPI, HTTPException
|
|
from fastapi.security import HTTPBasic, HTTPBasicCredentials
|
|
from starlette.status import HTTP_401_UNAUTHORIZED
|
|
|
|
app = FastAPI()
|
|
|
|
security = HTTPBasic()
|
|
|
|
|
|
def get_current_username(credentials: HTTPBasicCredentials = Depends(security)):
|
|
correct_username = secrets.compare_digest(credentials.username, "stanleyjobson")
|
|
correct_password = secrets.compare_digest(credentials.password, "swordfish")
|
|
if not (correct_username and correct_password):
|
|
raise HTTPException(
|
|
status_code=HTTP_401_UNAUTHORIZED,
|
|
detail="Incorrect email or password",
|
|
headers={"WWW-Authenticate": "Basic"},
|
|
)
|
|
return credentials.username
|
|
|
|
|
|
@app.get("/users/me")
|
|
def read_current_user(username: str = Depends(get_current_username)):
|
|
return {"username": username}
|