From 8aa1027aae67d935830d5421e03a804ddd5098a9 Mon Sep 17 00:00:00 2001 From: jxxghp Date: Mon, 8 Jan 2024 12:24:00 +0800 Subject: [PATCH] fix image proxy --- app/api/endpoints/system.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/app/api/endpoints/system.py b/app/api/endpoints/system.py index 76c65db5..8cc18f1b 100644 --- a/app/api/endpoints/system.py +++ b/app/api/endpoints/system.py @@ -24,14 +24,17 @@ from version import APP_VERSION router = APIRouter() -@router.get("/img/{imgurl:path}", summary="图片代理") -def get_img(imgurl: str) -> Any: +@router.get("/img/{imgurl:path}/{proxy}", summary="图片代理") +def get_img(imgurl: str, proxy: bool = False) -> Any: """ 通过图片代理（使用代理服务器） """ if not imgurl: return None - response = RequestUtils(ua=settings.USER_AGENT, proxies=settings.PROXY).get_res(url=imgurl) + if proxy: + response = RequestUtils(ua=settings.USER_AGENT, proxies=settings.PROXY).get_res(url=imgurl) + else: + response = RequestUtils(ua=settings.USER_AGENT).get_res(url=imgurl) if response: return Response(content=response.content, media_type="image/jpeg") return None