pythonasyncioapiasyncfastapiframeworkjsonjson-schemaopenapiopenapi3pydanticpython-typespython3redocreststarletteswaggerswagger-uiuvicornweb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
810 B
27 lines
810 B
from fastapi import Depends, FastAPI, HTTPException
|
|
from fastapi.security.api_key import APIKeyHeader
|
|
|
|
# Replace with your actual API key, ideally injected as an Environment Variable
|
|
# or stored in a secure DB.
|
|
API_KEY = "mysecretapikey"
|
|
|
|
API_KEY_NAME = "X-API-Key"
|
|
api_key_header = APIKeyHeader(
|
|
name=API_KEY_NAME,
|
|
description="API Key required to access secure endpoints.",
|
|
)
|
|
|
|
|
|
def verify_api_key(api_key: str = Depends(api_key_header)):
|
|
if api_key != API_KEY:
|
|
raise HTTPException(status_code=401, detail="Invalid API Key")
|
|
|
|
|
|
# Apply it for all endpoints
|
|
app = FastAPI(dependencies=[Depends(verify_api_key)])
|
|
|
|
|
|
# Apply it for specific endpoints
|
|
@app.get("/secure-data")
|
|
def secure_endpoint(api_key: str = Depends(verify_api_key)):
|
|
return {"message": "You have access to secure data"}
|
|
|