RRRosmontis
371d3a3624
- source/dufe/ginkgostu_sign_in/ginkgo_sign_in.py: 签到码查询服务 （Flask + waitress 16 线程，监听 127.0.0.1:9005，端点 GET|POST /dufe/ginkgostu_sign_in） - systemd/ginkgo_sign_in.service: 常驻服务单元，Restart=always，日志 /var/log/ginkgo_sign_in.log - source/main.py: 新增 DUFE_BACKEND 与 handle_dufe_sign_in， 并新增 proxy_forward()（保留请求方法与请求体，支持 POST）； 原有 proxy()（固定 GET，供 pixiv/bilibili 用）未做任何改动
98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
from aiohttp import web, ClientSession, ClientTimeout
|
||
|
||
|
||
PIXIV_BACKEND = "http://127.0.0.1:9007"
|
||
BILIBILI_BACKEND = "http://127.0.0.1:9006"
|
||
DUFE_BACKEND = "http://127.0.0.1:9005"
|
||
|
||
|
||
async def handle_pixiv(request):
|
||
pid = request.query.get("id", "")
|
||
if not pid:
|
||
return web.json_response({"error": "missing id parameter"}, status=400)
|
||
target = f"{PIXIV_BACKEND}/{pid}"
|
||
return await proxy(request, target)
|
||
|
||
|
||
async def handle_bilibili(request):
|
||
qs = request.query_string
|
||
target = f"{BILIBILI_BACKEND}/bilibili/get_video"
|
||
if qs:
|
||
target += "?" + qs
|
||
return await proxy(request, target)
|
||
|
||
|
||
async def handle_dufe_sign_in(request):
|
||
# ginkgo_sign_in.service（127.0.0.1:9005），端点 /dufe/ginkgostu_sign_in，支持 GET|POST
|
||
target = f"{DUFE_BACKEND}/dufe/ginkgostu_sign_in"
|
||
if request.query_string:
|
||
target += "?" + request.query_string
|
||
return await proxy_forward(request, target)
|
||
|
||
|
||
async def proxy(request, target):
|
||
timeout = ClientTimeout(total=300)
|
||
async with ClientSession(timeout=timeout) as session:
|
||
try:
|
||
async with session.get(target) as resp:
|
||
body = await resp.read()
|
||
headers = {}
|
||
for key in ("Content-Type", "Content-Disposition", "Cache-Control"):
|
||
if key in resp.headers:
|
||
headers[key] = resp.headers[key]
|
||
return web.Response(
|
||
body=body,
|
||
status=resp.status,
|
||
headers=headers,
|
||
)
|
||
except Exception as e:
|
||
return web.json_response(
|
||
{"error": "backend unreachable", "detail": str(e)}, status=502
|
||
)
|
||
|
||
|
||
async def proxy_forward(request, target):
|
||
# 保留请求方法与请求体的转发，供需要 POST 的后端使用（dufe 签到码查询）
|
||
timeout = ClientTimeout(total=300)
|
||
data = None
|
||
fwd_headers = {}
|
||
if request.method not in ("GET", "HEAD"):
|
||
data = await request.read()
|
||
ctype = request.headers.get("Content-Type")
|
||
if ctype:
|
||
fwd_headers["Content-Type"] = ctype
|
||
async with ClientSession(timeout=timeout) as session:
|
||
try:
|
||
async with session.request(
|
||
request.method, target, data=data, headers=fwd_headers
|
||
) as resp:
|
||
body = await resp.read()
|
||
headers = {}
|
||
for key in ("Content-Type", "Content-Disposition", "Cache-Control"):
|
||
if key in resp.headers:
|
||
headers[key] = resp.headers[key]
|
||
return web.Response(
|
||
body=body,
|
||
status=resp.status,
|
||
headers=headers,
|
||
)
|
||
except Exception as e:
|
||
return web.json_response(
|
||
{"error": "backend unreachable", "detail": str(e)}, status=502
|
||
)
|
||
|
||
|
||
async def handle_root(request):
|
||
raise web.HTTPFound("https://rrrosmontis.icu/articles/index.html?id=00003")
|
||
|
||
|
||
app = web.Application()
|
||
app.router.add_get("/", handle_root)
|
||
app.router.add_get("/pixiv/get_image", handle_pixiv)
|
||
app.router.add_get("/bilibili/get_video", handle_bilibili)
|
||
app.router.add_get("/dufe/ginkgostu_sign_in", handle_dufe_sign_in)
|
||
app.router.add_post("/dufe/ginkgostu_sign_in", handle_dufe_sign_in)
|
||
|
||
if __name__ == "__main__":
|
||
web.run_app(app, host="127.0.0.1", port=9008)
|