diff --git a/main.py b/main.py index 8b2f19c..251975a 100644 --- a/main.py +++ b/main.py @@ -23,7 +23,7 @@ BASE_DIR = Path("/usr/src/app/taobin_project") # BASE_DIR = Path("/taobin_project") SERVICE_NAME = os.getenv("SERVICE_NAME") -PREFIX = os.getenv("PREFIX", "") +PREFIX = os.getenv("PREFIX", "/taobin-image") GIT_REPO_SERVER_URL = os.getenv("GIT_REPO_SERVER_URL") FRONTEND_NOTIFY_URL = os.getenv("FRONTEND_NOTIFY_URL") @@ -44,7 +44,7 @@ ALLOWED_EXTENSIONS = {".jpg", ".jpeg", ".png", ".gif", ".webp"} async def strip_prefix_middleware(request: Request, call_next): path = request.scope["path"] - if path.startswith(PREFIX): + if PREFIX and path.startswith(PREFIX): new_path = path[len(PREFIX):] if not new_path: