parker/linklogger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove custom 404 handler

Browse Source
This commit is contained in:
Parker M. 2024-11-12 00:34:06 -06:00
parent 7736846d86
commit 8d3edb4ffe
Signed by: parker
GPG Key ID: 505ED36FC12B5D5E
1 changed files with 1 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
api/main.py
View File

@ -1,13 +1,11 @@
from fastapi import FastAPI, Depends, Request, Path from fastapi import FastAPI, Depends, Request, Path
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse, RedirectResponse from fastapi.responses import RedirectResponse
from api.routes.auth_routes import router as auth_router from api.routes.auth_routes import router as auth_router
from api.routes.links_routes import router as links_router from api.routes.links_routes import router as links_router
from api.routes.user_routes import router as user_router from api.routes.user_routes import router as user_router
from api.routes.log_routes import router as log_router from api.routes.log_routes import router as log_router
from typing import Annotated from typing import Annotated
from fastapi.exceptions import HTTPException
from starlette.status import HTTP_404_NOT_FOUND
from api.util.db_dependency import get_db from api.util.db_dependency import get_db
from api.util.log import log from api.util.log import log
@ -65,12 +63,3 @@ async def log_redirect(
log(link, ip, user_agent) log(link, ip, user_agent)
db.close() db.close()
return RedirectResponse(url=link_record.redirect_link) return RedirectResponse(url=link_record.redirect_link)
# Custom handler for 404 errors
@app.exception_handler(HTTP_404_NOT_FOUND)
async def custom_404_handler(request: Request, exc: HTTPException):
return JSONResponse(
status_code=404,
content={"message": "Resource not found"},
)