PaiGramTeam/luoxu-api-pub
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/luoxu-api-pub.git synced 2024-11-24 17:19:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ghost avatar

Browse Source
This commit is contained in:
xtaodada 2023-09-14 23:19:44 +08:00
parent 0b8505080f
commit 46c4292c3e
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -1,3 +1,5 @@
from typing import Union
from fastapi import FastAPI from fastapi import FastAPI
from httpx import AsyncClient from httpx import AsyncClient
from starlette.middleware.cors import CORSMiddleware from starlette.middleware.cors import CORSMiddleware
@ -8,6 +10,7 @@ from starlette.staticfiles import StaticFiles
app = FastAPI() app = FastAPI()
local_api = "http://127.0.0.1:9008/luoxu/" local_api = "http://127.0.0.1:9008/luoxu/"
local_stats_api = "http://127.0.0.1:9010/api/" local_stats_api = "http://127.0.0.1:9010/api/"
ghost_url = "https://search-pub.xtaolabs.com/luoxu/avatar/ghost.jpg"
allowed_origins = [ allowed_origins = [
"search-pub.xtaolabs.com", "search-pub.xtaolabs.com",
"127.0.0.1:9009", "127.0.0.1:9009",
@ -72,9 +75,11 @@ async def get_names(request: Request):
@app.get("/luoxu/avatar/{avatar}") @app.get("/luoxu/avatar/{avatar}")
async def read_avatar(avatar: str) -> StreamingResponse: async def read_avatar(avatar: str):
async with AsyncClient(timeout=60) as client: async with AsyncClient(timeout=60) as client:
r = await client.get(f"{local_api}avatar/{avatar}") r = await client.get(f"{local_api}avatar/{avatar}")
if r.status_code:
return RedirectResponse(url=ghost_url)
return StreamingResponse(r.aiter_bytes(), media_type="image/png") return StreamingResponse(r.aiter_bytes(), media_type="image/png")