fix image proxy

This commit is contained in:
jxxghp 2024-01-08 12:24:00 +08:00
parent b4cb9c3fb3
commit 8aa1027aae

View File

@ -24,14 +24,17 @@ from version import APP_VERSION
router = APIRouter() router = APIRouter()
@router.get("/img/{imgurl:path}", summary="图片代理") @router.get("/img/{imgurl:path}/{proxy}", summary="图片代理")
def get_img(imgurl: str) -> Any: def get_img(imgurl: str, proxy: bool = False) -> Any:
""" """
通过图片代理使用代理服务器 通过图片代理使用代理服务器
""" """
if not imgurl: if not imgurl:
return None return None
if proxy:
response = RequestUtils(ua=settings.USER_AGENT, proxies=settings.PROXY).get_res(url=imgurl) 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: if response:
return Response(content=response.content, media_type="image/jpeg") return Response(content=response.content, media_type="image/jpeg")
return None return None