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/ 指纹未变；忽略注释的逐字等价性校验通过。
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""FastAPI 依赖：会话解析、当前用户、登录校验。
|
|
|
|
与 auth_system 的语义对齐（共用同一 Cookie、同一站点库）：
|
|
- 会话 Cookie 经 core.session 解码（Flask 兼容签名），密钥同源
|
|
- 封禁账号（status 0/2）：按未登录处理
|
|
- 顶号校验：会话内的 `site_uid_seq` 必须与库中 `login_seq` 一致，否则视为失效
|
|
- 剩余次数/账本预扣属付费流程，由 web.billing 承担（此处只做身份解析）
|
|
"""
|
|
from dataclasses import dataclass
|
|
from typing import Annotated, Any
|
|
|
|
from fastapi import Cookie, Depends
|
|
|
|
from core import repo
|
|
from core import session as sess
|
|
from web.errors import AppError
|
|
|
|
_SESSION_COOKIE = 'session'
|
|
|
|
|
|
def secret_key() -> str:
|
|
"""会话签名密钥（密钥来源：data/secret.key）。"""
|
|
import auth_system
|
|
return auth_system.get_secret_key()
|
|
|
|
|
|
@dataclass
|
|
class SiteUser:
|
|
id: int
|
|
username: str
|
|
role: str
|
|
credits: int
|
|
|
|
@property
|
|
def is_admin(self) -> bool:
|
|
return self.role == 'admin'
|
|
|
|
@property
|
|
def unlimited(self) -> bool:
|
|
return self.is_admin or self.credits == -1
|
|
|
|
|
|
def _parse_cookie(session_cookie: str | None) -> dict[str, Any] | None:
|
|
return sess.loads(secret_key(), session_cookie or '')
|
|
|
|
|
|
async def current_user(
|
|
session_cookie: Annotated[str | None, Cookie(alias=_SESSION_COOKIE)] = None,
|
|
) -> SiteUser | None:
|
|
"""当前登录用户（未登录/失效返回 None）。"""
|
|
data = _parse_cookie(session_cookie)
|
|
uid = sess.get_uid(data)
|
|
if not uid:
|
|
return None
|
|
row = await repo.get_user_by_id(uid)
|
|
if row is None:
|
|
return None
|
|
status = row.get('status', 1)
|
|
if status in (0, 2): # 禁用/封禁：按未登录处理
|
|
return None
|
|
# 顶号校验：库有 login_seq 时会话版本必须严格一致
|
|
db_seq = int(row.get('login_seq') or 0)
|
|
sess_seq = int((data or {}).get(sess.SESSION_KEY + '_seq') or 0)
|
|
if db_seq and sess_seq != db_seq:
|
|
return None
|
|
return SiteUser(
|
|
id=int(row['id']),
|
|
username=row['username'],
|
|
role=row['role'],
|
|
credits=int(row.get('duration_seconds') or 0),
|
|
)
|
|
|
|
|
|
async def require_login(
|
|
user: Annotated[SiteUser | None, Depends(current_user)],
|
|
) -> SiteUser:
|
|
"""登录闸：未登录抛 401，已登录返回用户。"""
|
|
if user is None:
|
|
raise AppError('请先登录后再使用解析功能', 401, error_code='unauthorized')
|
|
return user
|
|
|
|
|
|
async def require_admin(
|
|
user: Annotated[SiteUser | None, Depends(current_user)],
|
|
) -> SiteUser:
|
|
"""管理员闸：非管理员抛 403。"""
|
|
if user is None or not user.is_admin:
|
|
raise AppError('需要管理员权限', 403, error_code='forbidden')
|
|
return user
|