增加批量接口,完善日期解析

This commit is contained in:
sansen
2026-08-14 17:06:10 +08:00
parent 86e7ef2971
commit 2f3830bd5d
3 changed files with 203 additions and 87 deletions
+71 -4
View File
@@ -1,5 +1,6 @@
import json
from fastapi import FastAPI, HTTPException, Header, Form, Request
from time import perf_counter
from fastapi import FastAPI, HTTPException, Header, Request
from fastapi.responses import JSONResponse
from fastapi.middleware.cors import CORSMiddleware
from loguru import logger
@@ -16,21 +17,36 @@ app.add_middleware(
allow_headers=["*"],
)
# batch 接口单次请求的条目上限(防滥用)
BATCH_MAX_ITEMS = 1000
def _check_auth(Authorization: str):
if Authorization != "3FFA6A4B073CF065969630692331A873":
raise HTTPException(status_code=403, detail="未授权的操作")
def _parse_urlencoded_content(body: bytes) -> str:
"""手动解析 x-www-form-urlencoded 的 content 字段:绕过 starlette 1MB 表单限制"""
from urllib.parse import parse_qs
params = parse_qs(body.decode("utf-8"), keep_blank_values=True)
return params.get("content", [""])[0]
@app.post("/convert/start")
async def convert_str(
request: Request,
content: str = Form(...), # ✅ 改为接收 x-www-form-urlencoded 参数
Authorization: str = Header(None)
):
"""
带调试信息的转换接口 (支持 x-www-form-urlencoded)
手动解析表单体:绕过 starlette 1MB 表单限制(族谱全文可达数 MB)
"""
if Authorization != "3FFA6A4B073CF065969630692331A873":
raise HTTPException(status_code=403, detail="未授权的操作")
_check_auth(Authorization)
try:
content = _parse_urlencoded_content(await request.body())
res = convert_text(content)
logger.info(f"转换结果: {json.dumps(res, ensure_ascii=False, indent=2)}")
return JSONResponse(content={"code": 0, "result": res})
@@ -42,6 +58,57 @@ async def convert_str(
)
@app.post("/convert/batch")
async def convert_batch(
request: Request,
Authorization: str = Header(None)
):
"""
批量转换接口:短文本多任务场景,一次请求处理 N 条 content。
请求: {"items": [{"id": "row1", "content": "..."}, ...]}application/json
响应: {"code": 0, "result": [{"id": "row1", "result": [...]}, ...]}
- id 透传用于前端回填,缺省按数组顺序编号
- 单条解析失败不影响其他条目(该条返回 error 字段)
- 每条 content 独立解析,位置语义与单接口一致(从 0 起)
"""
_check_auth(Authorization)
# 请求体解析与入参校验失败 → 400(客户端错误)
try:
payload = json.loads((await request.body()).decode("utf-8"))
except Exception as e:
logger.warning(f"batch 请求体解析失败: {str(e)}")
raise HTTPException(status_code=400, detail="请求体必须是合法 JSON")
items = payload.get("items") if isinstance(payload, dict) else None
if not isinstance(items, list) or not items:
raise HTTPException(status_code=400, detail="items 必须是非空数组")
if len(items) > BATCH_MAX_ITEMS:
raise HTTPException(status_code=400, detail=f"items 超过上限 {BATCH_MAX_ITEMS}")
try:
t0 = perf_counter()
results = []
for i, item in enumerate(items):
item_id = item.get("id") if isinstance(item, dict) else None
content = item.get("content", "") if isinstance(item, dict) else ""
if item_id is None:
item_id = str(i)
try:
results.append({"id": item_id, "result": convert_text(content)})
except Exception as e:
logger.error(f"batch 条目 {item_id} 解析异常: {str(e)}")
results.append({"id": item_id, "error": str(e)})
dt = perf_counter() - t0
logger.info(f"批量转换: {len(results)} 条, 耗时 {dt * 1000:.0f}ms")
return JSONResponse(content={"code": 0, "result": results})
except Exception as e:
logger.error(f"批量转换异常: {str(e)}")
return JSONResponse(
content={"code": 500, "result": f"批量转换服务异常: {str(e)}"},
status_code=500
)
if __name__ == "__main__":
import uvicorn