diff --git a/app/main.py b/app/main.py index b450ed0..b9306ae 100644 --- a/app/main.py +++ b/app/main.py @@ -7,7 +7,8 @@ from typing import Annotated, Optional, Union import geoip2.database from dotenv import load_dotenv -from fastapi import Body, FastAPI, Path, Request, Response, status +from fastapi import (Body, FastAPI, HTTPException, Path, Request, Response, + status) from fastapi.middleware.cors import CORSMiddleware from fastapi.responses import RedirectResponse from geoip2.errors import AddressNotFoundError @@ -126,4 +127,6 @@ def root_redirect(req: Request) -> RedirectResponse: ''' Redirect empty request using REMOTE_ADDR ''' + if not req.client: + raise HTTPException(status_code=404, detail="Item not found") return RedirectResponse(url=str(req.url) + str(req.client.host))