feat(ext-data): 出站请求默认携带 tsp 标识头

扩展数据的三个出站点 (定时/手动拉取与历史回补、内置预设拉取、URL 探测
测试) 统一经 outbound_headers() 注入 User-Agent: tsp/<版本> 与
X-TSP-Client: tick-stock-panel, 服务端 (如 tickflow-hub) 可据此识别
本项目的请求来源。

用户在拉取配置里显式设置的同名请求头优先 (大小写不敏感匹配, 不重复
发送), 需要特定 UA 的数据源不受影响。

验证: pytest 22/22 (含 3 个新用例: 默认标识头/用户头优先/实际请求携带);
逐文件 ruff 与 HEAD 对比零新增。
This commit is contained in:
shy3130
2026-09-06 23:16:27 +08:00
parent d949d9618c
commit ddb265b0f8
5 changed files with 71 additions and 5 deletions
+21 -1
View File
@@ -21,6 +21,26 @@ from app.services.ext_data import (
logger = logging.getLogger(__name__)
def outbound_headers(user_headers: dict[str, str] | None = None) -> dict[str, str]:
"""扩展数据出站请求的默认标识头。
默认携带 User-Agent: tsp/<版本> 与 X-TSP-Client: tick-stock-panel,
供服务端 (如 tickflow-hub) 识别本项目的请求。用户在拉取配置里显式
设置的同名头优先 (大小写不敏感), 不被标识头覆盖。
"""
from app import __version__
defaults = {
"User-Agent": f"tsp/{__version__}",
"X-TSP-Client": "tick-stock-panel",
}
override = {k.lower() for k in (user_headers or {})}
return {
**{k: v for k, v in defaults.items() if k.lower() not in override},
**(user_headers or {}),
}
def _in_time_window(start: str | None, end: str | None) -> bool:
"""检查当前本地时间是否在每日时间窗口内。
@@ -144,7 +164,7 @@ async def fetch_rows_for_date(config: ExtConfig, target_date: date) -> list[dict
url = _with_date_param(pull.url, pull.date_param, target_date)
async with httpx.AsyncClient(timeout=30) as client:
headers = pull.headers or {}
headers = outbound_headers(pull.headers)
kwargs: dict[str, Any] = {"headers": headers}
if pull.method.upper() == "POST" and pull.body: