Archived
snowy-yang
65adfc9089
core/aiohttp_compat 是「同步转异步」迁移期的 requests 同形异步垫片：它存在的 唯一理由是让既有 `requests.post/get + raise_for_status()` 风格的代码以最小 改动迁到异步。其两个调用方（music_api/api.py、music_api/client.py）**本就已是 async def**——垫片只剩「保留 requests 语义」这一层包装，属于迁移残留。 改动： - api.py：23 处 requests.post/get → await core.http_client.request()。 该文件每处都在请求后立刻 raise_for_status() 才取 .json()，从不把 4xx 当数据读， 故「_once 对非 2xx 直接抛 UpstreamHTTPError」与旧语义等价（原 raise_for_status 抛的 HTTPError 也一律落入 RequestException 分支）； except requests.RequestException → except UpstreamError。 - client.py：两个自管重试的方法改用新公开的 core.http_client.request_once() （单次、不重试），保留原有「5xx 退避重试一次 / 4xx 立即失败」的分流—— 把 status_code>=500 的判断换成 UpstreamHTTPError.status，行为逐条对齐 （4xx=风控不重试，网络/5xx 重试一次）。响应体上限交由客户端 max_bytes 统一把关。 post_request_full 返回类型改为 HTTPResult（qr.py 用到的 .text/.cookies/.headers 三者在 HTTPResult 上齐备）。 - core/http_client：_once 更名并公开为 request_once（自管重试的调用方需要它）。 注：这是为「删垫片」而补的唯一一个小口子，不是新增抽象。 注意一处**有意的行为变化**：api.py 原经垫片请求零重试，现在 request() 自带 「网络异常/5xx 退避重试一次、4xx 不重试」（core.http_client 既定口径）。 429/460 属 4xx 不重试，不加剧风控。 - 删除 core/aiohttp_compat.py；web/app.py 去掉其 close()；pyproject 去掉其 per-file-ignore；core/http_errors.py 注释更正为「仅供 core.sync_http」。 - 测试：test_aiohttp_compat.py 拆解为 test_async_bridge.py（只留桥的两条契约， 垫片自身的 4 条随垫片消亡）；test_search_music / test_playlist_detail 的替身 边界从 api.requests.post 换成 api._http.request，异常类型换成 UpstreamNetworkError。 门禁：ruff 全过 / pyright 0-0 / pytest 383 项全绿（0 失败；383=387-6+2，账面吻合）。
120 lines
4.5 KiB
Python
120 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""HTTP 客户端与响应体收窄工具。
|
|
|
|
（自 music_api.py 拆出，行为与原实现一致。）
|
|
"""
|
|
|
|
import asyncio
|
|
from typing import Any
|
|
|
|
from core.http_client import (
|
|
HTTPResult,
|
|
UpstreamError,
|
|
UpstreamHTTPError,
|
|
UpstreamTooLarge,
|
|
)
|
|
from core.http_client import client as _http
|
|
|
|
from .constants import (
|
|
APIConstants,
|
|
)
|
|
from .errors import (
|
|
APIException,
|
|
)
|
|
|
|
|
|
def _ensure_dict(result: Any, msg: str) -> dict[str, Any]:
|
|
"""上游 200 但返回 null/数组等非 dict 体（风控/抖动）时统一前置校验"""
|
|
if isinstance(result, dict):
|
|
return result
|
|
raise APIException(msg)
|
|
|
|
|
|
def _as_dict(v: Any) -> dict[str, Any]:
|
|
"""上游嵌套字段类型不稳定（dict / None / 列表都真实出现过）：非 dict 一律回落空 dict。
|
|
|
|
收窄成显式函数，避免满屏 `x if isinstance(x, dict) else {}` 三元式——
|
|
那种写法读起来累，且把字段来自 `user.get(...)`（类型 Any|None）时的收窄责任
|
|
散落在各处，容易被后来的改动改漏。
|
|
"""
|
|
return v if isinstance(v, dict) else {}
|
|
|
|
|
|
class HTTPClient:
|
|
"""HTTP客户端类"""
|
|
|
|
@staticmethod
|
|
async def post_request(url: str, params: str, cookies: dict[str, str]) -> str:
|
|
"""发送POST请求并返回文本响应。
|
|
网络异常/5xx 做一次 1 秒退避重试（全站 API 调用此前零重试，
|
|
上游抖动即整首失败）；4xx（429/460 风控）不重试——立即重试会加剧风控。
|
|
|
|
走 core.http_client.request_once（单次、不重试）：重试策略由本函数自管，
|
|
非 2xx 由 request_once 抛 UpstreamHTTPError，按 .status 分流——
|
|
与原先「垫片只在 raise_for_status() 时才抛 HTTPError」的分流等价。
|
|
"""
|
|
headers = {
|
|
'User-Agent': APIConstants.USER_AGENT,
|
|
'Referer': APIConstants.REFERER,
|
|
}
|
|
|
|
request_cookies = APIConstants.DEFAULT_COOKIES.copy()
|
|
request_cookies.update(cookies)
|
|
|
|
last_err = None
|
|
for attempt in range(2):
|
|
try:
|
|
response = await _http.request_once(
|
|
'POST', url, headers=headers, cookies=request_cookies,
|
|
data={"params": params}, timeout=30)
|
|
return response.text
|
|
except UpstreamHTTPError as e:
|
|
if e.status >= 500 and attempt == 0:
|
|
await asyncio.sleep(1.0)
|
|
continue
|
|
# 4xx（429/460 风控）不重试——原实现 HTTPError 落入
|
|
# 下方异常分支被重试一次，与意图相反
|
|
raise APIException(f"HTTP请求失败: {e}") from e
|
|
except UpstreamTooLarge as e:
|
|
# 响应体超限是确定性失败，重试无意义
|
|
raise APIException("上游响应体异常（超过大小上限）") from e
|
|
except UpstreamError as e:
|
|
last_err = e
|
|
if attempt == 0:
|
|
await asyncio.sleep(1.0)
|
|
continue
|
|
raise APIException(f"HTTP请求失败: {last_err}")
|
|
|
|
@staticmethod
|
|
async def post_request_full(url: str, params: str, cookies: dict[str, str]) -> HTTPResult:
|
|
"""发送POST请求并返回完整响应对象（同 post_request 的有界重试）"""
|
|
headers = {
|
|
'User-Agent': APIConstants.USER_AGENT,
|
|
'Referer': APIConstants.REFERER,
|
|
}
|
|
|
|
request_cookies = APIConstants.DEFAULT_COOKIES.copy()
|
|
request_cookies.update(cookies)
|
|
|
|
last_err = None
|
|
for attempt in range(2):
|
|
try:
|
|
response = await _http.request_once(
|
|
'POST', url, headers=headers, cookies=request_cookies,
|
|
data={"params": params}, timeout=30)
|
|
return response
|
|
except UpstreamHTTPError as e:
|
|
if e.status >= 500 and attempt == 0:
|
|
await asyncio.sleep(1.0)
|
|
continue
|
|
# 4xx 不重试（同上）
|
|
raise APIException(f"HTTP请求失败: {e}") from e
|
|
except UpstreamTooLarge as e:
|
|
raise APIException("上游响应体异常（超过大小上限）") from e
|
|
except UpstreamError as e:
|
|
last_err = e
|
|
if attempt == 0:
|
|
await asyncio.sleep(1.0)
|
|
continue
|
|
raise APIException(f"HTTP请求失败: {last_err}")
|