File size: 1,981 Bytes
0aee47a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
from typing import Any, Dict
from fastapi import FastAPI, Request, Response
from .parser import Parser
def Result(code: int, data: Any) -> dict:
"""
请求结果
Args:
code (int): 状态码
data (Any): 返回数据
Result:
dict: 返回体
"""
return {
"code": code,
"data" if code == 0 else "error": data
}
async def bilibili_api_web(
request: Request,
response: Response,
path: str,
) -> dict:
"""
网络接口
Args:
request (request): 请求
response (Response): 响应
path (str): 请求路径 例如 `user.User(434334701).get_user_info()`
max_age (int, Optional): 请求头 `Cache-Control` 中的 `max-age`
params (str, Optional): 可选变量 如果 path 中多次用到某个值可保存
例如 `?uid=434334701&roomid=21452505`
Returns:
dict: 解析结果
"""
# 返回头设置
vars: Dict[str, str] = dict(request.query_params)
max_age = vars.pop("max_age", None)
if max_age is not None:
response.headers["Cache-Control"] = f"max-age={max_age}"
response.headers["Access-Control-Allow-Origin"] = "*"
# 先判断是否有效 再分析
try:
async with Parser(vars) as parser:
if not parser.valid:
return Result(1, "Credential 验证失败")
obj, err = await parser.parse(path) # 什么 golang 写法
if err is None:
return Result(0, obj)
return Result(2, f"解析语句 {err} 错误")
except Exception as e:
return Result(3, f"未知错误 {e}")
def get_fastapi(app_run_path: str = "/{path}") -> FastAPI:
app = FastAPI()
app_run_path = app_run_path if "{path}" in app_run_path else "/{path}"
app.add_api_route(app_run_path, bilibili_api_web, methods=["GET"])
return app |