PaiGramTeam/luoxu-api-pub
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/luoxu-api-pub.git synced 2024-11-21 14:48:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: timeout error

Browse Source
This commit is contained in:
omg-xtao 2024-03-30 14:02:27 +00:00 committed by GitHub
parent e40e2b3052
commit d7dc6b0884
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@ -1,7 +1,7 @@
from typing import Union
from fastapi import FastAPI
from httpx import AsyncClient
from httpx import AsyncClient, ReadTimeout
from starlette.middleware.cors import CORSMiddleware
from starlette.requests import Request
from starlette.responses import StreamingResponse, RedirectResponse
@ -77,7 +77,10 @@ async def get_names(request: Request):
@app.get("/luoxu/avatar/{avatar}")
async def read_avatar(avatar: str):
async with AsyncClient(timeout=60) as client:
try:
r = await client.get(f"{local_api}avatar/{avatar}")
except ReadTimeout:
return RedirectResponse(url=ghost_url)
if r.status_code == 307:
return RedirectResponse(url=ghost_url)
return StreamingResponse(r.aiter_bytes(), media_type="image/png")