Archived
snowy-yang
8446166120
## uv 环境管理 - pyproject.toml 声明 dependencies + [dependency-groups].dev，uv.lock 精确锁定 - 删除 requirements.txt / requirements-fastapi.txt / requirements-dev.txt - .python-version = 3.12；Dockerfile 改用 uv sync --frozen（可复现构建） - 新增 jinja2（首页渲染） ## 删除 Flask 技术债 - 从 main.py 用 ast 精确剥离 56 个 Flask 节点（路由/中间件/错误处理/启动）， 框架无关的服务层落为 services/runtime.py（MusicAPIService、下载/打包 worker、 任务监控、缓存清扫），零 Flask 依赖 - 删除 main.py（4527 行）与 auth_routes.py（1050 行） - auth_system.py 移除会话/请求上下文层（login_session/current_user/login_required/ admin_required/_client_ip 等），保留数据与业务层 - 新增 core/net.py（IP 解析）、web/auth_state.py（登录锁定/2FA 票据/发码限流） - FastAPI 侧改直接依赖 services.runtime，不再 import main - web/app.py 去掉 Flask fallback；入口默认根路径模式（API_PREFIX 可选） ## 删除 requests 技术债 - 新增 core/sync_http.py：requests 风格的同步门面（底层 aiohttp 连接池， 流式读取经有界队列背压），含 Session/headers/(connect,read) 元组超时兼容 - core/utils、auth_system、podcast_api、music_downloader、services 全部切换 - 实测真实下载 5.2MB 音频完整成功 ## 质量门禁归零 - ruff：0 错误（B904 全部修复，不再忽略） - pyright：0 错误（类型收窄修复） - pytest：140/140（含 golden 契约回归、root 模式 66 项路径覆盖） - 真实 uvicorn 启动：/health、/、/static、受保护端鉴权全部正确 新增 tests/golden/responses.json：删除 Flask 前捕获的 32 例响应快照， FastAPI 逐字段复现——删除巨石后行为回归仍有硬证据。
69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""公网限流（FastAPI 栈）——按 IP 滑动窗口，与 Flask 侧 _rate_ok 同一实现口径。
|
|
|
|
复用 core.state.RATE_BUCKETS/_RATE_LOCK，使两栈在迁移期共享同一批限流桶
|
|
（同一 IP 无论打到哪个栈都计入同一窗口），避免出现「绕到另一个框架即绕过限流」。
|
|
"""
|
|
import collections
|
|
import time
|
|
|
|
from fastapi import Request
|
|
|
|
from core.config import RESULT_RETENTION_SECONDS # noqa: F401 - 保持与 Flask 侧常量同源
|
|
from core.state import RATE_BUCKETS, RATE_LOCK
|
|
from web.errors import AppError
|
|
|
|
# 免于限流的端点（试听流/进度轮询，与 Flask _QUIET/_rate_limit_guard 同口径）
|
|
_EXEMPT_PREFIXES = ('/audio_stream',)
|
|
_EXEMPT_PATHS = {'/api/queue', '/api/batch_progress', '/api/task/heartbeat'}
|
|
|
|
|
|
def client_ip(request: Request) -> str:
|
|
"""客户端 IP：仅当直连方是回环（可信反代）时采信 XFF/Real-IP。
|
|
|
|
规则与 Flask 时代完全一致（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=放行。与 Flask main._rate_ok 同算法、同桶容器。"""
|
|
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（与 Flask 包体同形）。豁免端点直接放行。"""
|
|
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)
|