fix image proxy

jxxghp · Jan 8, 2024 · 8aa1027 · 8aa1027
1 parent b4cb9c3
commit 8aa1027
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/app/api/endpoints/system.py b/app/api/endpoints/system.py
@@ -24,14 +24,17 @@
 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