mirror of
https://github.com/PaiGramTeam/luoxu-api-pub.git
synced 2024-11-24 17:19:32 +00:00
109 lines
2.9 KiB
Python
109 lines
2.9 KiB
Python
from typing import Union
|
|
|
|
from fastapi import FastAPI
|
|
from httpx import AsyncClient
|
|
from starlette.middleware.cors import CORSMiddleware
|
|
from starlette.requests import Request
|
|
from starlette.responses import StreamingResponse, RedirectResponse
|
|
from starlette.staticfiles import StaticFiles
|
|
|
|
app = FastAPI()
|
|
local_api = "http://127.0.0.1:9008/luoxu/"
|
|
local_stats_api = "http://127.0.0.1:9010/api/"
|
|
ghost_url = "https://search-pub.xtaolabs.com/luoxu/avatar/ghost.jpg"
|
|
allowed_origins = [
|
|
"search-pub.xtaolabs.com",
|
|
"127.0.0.1:9009",
|
|
]
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=allowed_origins,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
groups_id = [1366383997, 1797471403, 1936247070]
|
|
groups_stats_id = [-1001797471403]
|
|
groups = {
|
|
"groups": [
|
|
{
|
|
"group_id": "1366383997",
|
|
"name": "原神 · 提瓦特大陆",
|
|
"pub_id": "GenshinImpact"
|
|
},
|
|
{
|
|
"group_id": "1797471403",
|
|
"name": "崩坏：星穹铁道 · 仙舟：罗浮",
|
|
"pub_id": "HSRCN_Group"
|
|
},
|
|
{
|
|
"group_id": "1936247070",
|
|
"name": "绝区零 · 新艾利都",
|
|
"pub_id": "ZZZGroups"
|
|
}
|
|
]
|
|
}
|
|
|
|
|
|
@app.get("/luoxu/groups")
|
|
async def get_groups():
|
|
return groups
|
|
|
|
|
|
@app.get("/luoxu/search")
|
|
async def get_search(request: Request):
|
|
try:
|
|
if int(request.query_params.get("g", 0)) in groups_id:
|
|
async with AsyncClient(timeout=60) as client:
|
|
r = await client.get(f"{local_api}search", params=request.query_params)
|
|
return r.json()
|
|
except:
|
|
pass
|
|
return {}
|
|
|
|
|
|
@app.get("/luoxu/names")
|
|
async def get_names(request: Request):
|
|
try:
|
|
if int(request.query_params.get("g", 0)) in groups_id:
|
|
async with AsyncClient(timeout=60) as client:
|
|
r = await client.get(f"{local_api}names", params=request.query_params)
|
|
return r.json()
|
|
except:
|
|
pass
|
|
return {}
|
|
|
|
|
|
@app.get("/luoxu/avatar/{avatar}")
|
|
async def read_avatar(avatar: str):
|
|
async with AsyncClient(timeout=60) as client:
|
|
r = await client.get(f"{local_api}avatar/{avatar}")
|
|
if r.status_code == 307:
|
|
return RedirectResponse(url=ghost_url)
|
|
return StreamingResponse(r.aiter_bytes(), media_type="image/png")
|
|
|
|
|
|
@app.get("/luoxu/stats/{gid}")
|
|
async def read_avatar(gid: str):
|
|
try:
|
|
if int(gid) not in groups_stats_id:
|
|
return []
|
|
except Exception:
|
|
return []
|
|
async with AsyncClient(timeout=60) as client:
|
|
r = await client.get(f"{local_stats_api}group_analytics?cid={gid}")
|
|
return r.json()
|
|
|
|
|
|
@app.get("/")
|
|
async def redirect_to_html():
|
|
return RedirectResponse(url="/index.html")
|
|
|
|
|
|
app.mount("/", StaticFiles(directory="html"), name="html")
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
uvicorn.run(app, port=9009)
|