SiskelDev/WattpadDownloader
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(api): Error handlers accepting Request param

Browse Source
This commit is contained in:
TheOnlyWayUp
2024-12-08 13:14:42 +00:00
parent d924e1b6ce
commit 097c37f24e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/api/src/main.py
+3 -3
View File
@@ -7,7 +7,7 @@ from io import BytesIO
from enum import Enum from enum import Enum
from eliot import start_action from eliot import start_action
from aiohttp import ClientResponseError from aiohttp import ClientResponseError
from fastapi import FastAPI from fastapi import FastAPI, Request
from fastapi.responses import ( from fastapi.responses import (
FileResponse, FileResponse,
HTMLResponse, HTMLResponse,
@@ -92,7 +92,7 @@ def home():
@app.exception_handler(ClientResponseError) @app.exception_handler(ClientResponseError)
def download_error_handler(exception: ClientResponseError): def download_error_handler(request: Request, exception: ClientResponseError):
match exception.status: match exception.status:
case 400 | 404: case 400 | 404:
return HTMLResponse( return HTMLResponse(
@@ -114,7 +114,7 @@ def download_error_handler(exception: ClientResponseError):
@app.exception_handler(WattpadError) @app.exception_handler(WattpadError)
def download_wp_error_handler(exception: WattpadError): def download_wp_error_handler(request: Request, exception: WattpadError):
if isinstance(exception, StoryNotFoundError): if isinstance(exception, StoryNotFoundError):
return HTMLResponse( return HTMLResponse(
status_code=404, status_code=404,