Archived
snowy-yang
6288a98ecc
为什么：单文件 1100~1900 行的巨石模块让「改一处要通读全文」；同时注释里残留 大量已删台账的编号，以及「Flask/FastAPI 双栈迁移期」这类已失实的架构描述， 会误导后来者按不复存在的架构去改代码。 拆分（对外导入路径不变，行为零变更）： - music_downloader.py -> music_downloader/（models/downloader/cover/_http/_log/_concurrency） - auth_system.py -> auth_system/（db/security/users/credits/ledger/mail/cards/backup...） - music_api.py -> music_api/（api/client/crypto/qr/cookies/compat/constants/errors） - services/runtime.py -> services/（api_service/_shared/_taskcore/downloads/batch/podcast/ streaming/tasks）；services/runtime.py 保留为兼容门面 兼容性做法：各包用 __init__.py 以 __all__ 原样再导出全部符号，既有调用点 （auth_system.xxx / services.xxx / from music_api import Y）零改动；路径常量改为经包属性 动态读取（auth_system.DB_PATH），否则测试的 monkeypatch 会静默失效并写到生产库。 修复 9 处代码损伤（上一轮注释清理把 `()` 当「空括号残留」删掉所致；用「与拆分前逐字 等价」校验逐一定位，静态检查只会报出其中 4 处）： - core/config.py ).strip() or '*' - core/state.py threading.Lock() x2 - qr_login.py datetime.now() / .strip() - music_downloader/downloader.py .lower() - services/podcast.py set() / time.time() x2 清理与删除： - 删除 部署文档.md / 功能介绍.md / README_私有包警告.txt - 清理指向已删台账与文档的过时编号注释；把「迁移期 / 两栈 / 双框架 / 已迁移」等 失实描述改写为现状描述（保留 `<-> Flask /path` 映射表等仍有价值的契约来源说明） - _mkpkg.py：公开包剔除清单去掉已不存在的文档名；私有包警告改为自包含 - pyproject.toml：补 pythonpath = ["."]，使裸跑 pytest 不再 import core 失败 验证：ruff 全通过；pyright 0 errors / 0 warnings（102 文件）；pytest 144 passed； 真实 uvicorn 冒烟 6 条路由符合预期且生产 data/ 指纹未变；忽略注释的逐字等价性校验通过。
69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""公网限流——按 IP 滑动窗口。
|
|
|
|
限流桶集中在 core.state.RATE_BUCKETS/_RATE_LOCK，使所有端点共享同一批桶
|
|
（同一 IP 一律计入同一窗口），避免出现绕过限流的口子。
|
|
"""
|
|
import collections
|
|
import time
|
|
|
|
from fastapi import Request
|
|
|
|
from core.config import RESULT_RETENTION_SECONDS # noqa: F401 - 再导出，与 core.config 口径同源
|
|
from core.state import RATE_BUCKETS, RATE_LOCK
|
|
from web.errors import AppError
|
|
|
|
# 免于限流的端点（试听流/进度轮询）
|
|
_EXEMPT_PREFIXES = ('/audio_stream',)
|
|
_EXEMPT_PATHS = {'/api/queue', '/api/batch_progress', '/api/task/heartbeat'}
|
|
|
|
|
|
def client_ip(request: Request) -> str:
|
|
"""客户端 IP：仅当直连方是回环（可信反代）时采信 XFF/Real-IP。
|
|
|
|
规则见 core.net（与原实现同源）：XFF 最右段优先，
|
|
最右段为回环时回落 X-Real-IP，全部无效回落直连地址。
|
|
"""
|
|
from core.net import is_loopback_ip, trusted_proxy_ip
|
|
remote = request.client.host if request.client else '-'
|
|
if remote and remote != '-' and is_loopback_ip(remote):
|
|
xff = request.headers.get('x-forwarded-for')
|
|
if xff:
|
|
ip = trusted_proxy_ip(xff)
|
|
if ip and not is_loopback_ip(ip):
|
|
return ip
|
|
xreal = request.headers.get('x-real-ip')
|
|
if xreal:
|
|
ip = trusted_proxy_ip(xreal)
|
|
if ip:
|
|
return ip
|
|
return remote or '-'
|
|
|
|
|
|
def rate_ok(key: str, limit: int, window: int) -> bool:
|
|
"""滑动窗口限流：True=放行。"""
|
|
now = time.time()
|
|
with RATE_LOCK:
|
|
dq = RATE_BUCKETS.get(key)
|
|
if dq is None:
|
|
if len(RATE_BUCKETS) > 8000:
|
|
for k in [k for k, v in RATE_BUCKETS.items() if not v or now - v[-1] > window]:
|
|
RATE_BUCKETS.pop(k, None)
|
|
dq = collections.deque()
|
|
RATE_BUCKETS[key] = dq
|
|
while dq and now - dq[0] > window:
|
|
dq.popleft()
|
|
if len(dq) >= limit:
|
|
return False
|
|
dq.append(now)
|
|
return True
|
|
|
|
|
|
async def guard(request: Request, *, bucket: str, limit: int, window: int) -> None:
|
|
"""限流闸：超限抛 429。豁免端点直接放行。"""
|
|
path = request.url.path
|
|
if path in _EXEMPT_PATHS or path.startswith(_EXEMPT_PREFIXES):
|
|
return
|
|
if not rate_ok(f'{bucket}:' + client_ip(request), limit, window):
|
|
raise AppError('请求过于频繁，请稍后再试', 429)
|