From 716295b7fdee17d33e1fa5b7d6bf373607b0c8b0 Mon Sep 17 00:00:00 2001 From: Michelle Date: Sat, 16 May 2026 21:10:21 +0200 Subject: [PATCH] oauth -> HTTPAuth --- main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 709c5c9..1a75fe4 100644 --- a/main.py +++ b/main.py @@ -3,7 +3,7 @@ import aiohttp from fastapi import FastAPI, HTTPException, Request, UploadFile from fastapi.responses import FileResponse from fastapi import Depends, FastAPI -from fastapi.security import OAuth2PasswordBearer +from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer from pyrate_limiter import Duration, Limiter, Rate from fastapi_limiter.depends import RateLimiter from fastapi import Depends @@ -33,7 +33,7 @@ logging.basicConfig( handlers=[file_handler, console_handler], ) -oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token") +security = HTTPBearer() SECRET = os.getenv("SECRET") @asynccontextmanager @@ -123,9 +123,9 @@ async def fetch_images(): await asyncio.sleep(86400) # Sleep for 24 hours @app.post("/upload", dependencies=[Depends(RateLimiter(limiter=limiter))]) -async def upload_image(file: UploadFile, request: Request, token: str = Depends(oauth2_scheme)): +async def upload_image(file: UploadFile, request: Request, credentials: HTTPAuthorizationCredentials = Depends(security)): # if SECRET isn't set, return Unauthrorized - if token != SECRET or token is None: + if credentials.credentials != SECRET or credentials.credentials is None: raise HTTPException(status_code=401, detail="Unauthorized") if not file: raise HTTPException(status_code=400, detail="No file uploaded")