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/ 指纹未变；忽略注释的逐字等价性校验通过。
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""站点数据异步读写仓储。
|
|
|
|
直接读写 auth_system 建立的同一份 SQLite（同一 DB_PATH / 同一 schema），
|
|
但走 core.db 的 aiosqlite 异步层——读并发、写串行、WAL 读写同时进行。
|
|
|
|
设计约束：本模块**不改动** auth_system 既有 schema，只做读写；字段口径与
|
|
auth_system 保持一致（键名、默认值），读写同一份库文件。
|
|
"""
|
|
from typing import Any
|
|
|
|
from core import db
|
|
|
|
# ---------- settings ----------
|
|
|
|
async def get_setting(key: str, default: str = '') -> str:
|
|
"""读单个站点设置（对应 auth_system.get_setting）。"""
|
|
row = await db.fetch_one('SELECT value FROM settings WHERE key = ?', (key,))
|
|
return row['value'] if row is not None else default
|
|
|
|
|
|
async def set_setting(key: str, value: str) -> None:
|
|
"""写站点设置（upsert）。"""
|
|
await db.execute(
|
|
'INSERT INTO settings (key, value) VALUES (?, ?) '
|
|
'ON CONFLICT(key) DO UPDATE SET value = excluded.value',
|
|
(key, value),
|
|
)
|
|
|
|
|
|
async def all_settings() -> dict[str, str]:
|
|
rows = await db.fetch_all('SELECT key, value FROM settings')
|
|
return {r['key']: r['value'] for r in rows}
|
|
|
|
|
|
# ---------- announcements ----------
|
|
|
|
async def list_announcements(limit: int = 20) -> list[dict[str, Any]]:
|
|
rows = await db.fetch_all(
|
|
'SELECT * FROM announcements ORDER BY rowid DESC LIMIT ?', (limit,))
|
|
return [dict(r) for r in rows]
|
|
|
|
|
|
# ---------- users（只读统计；写操作请走 auth_system 的事务口径） ----------
|
|
|
|
async def count_users(role: str | None = None) -> int:
|
|
if role:
|
|
return int(await db.fetch_val('SELECT COUNT(*) FROM users WHERE role = ?', (role,), 0))
|
|
return int(await db.fetch_val('SELECT COUNT(*) FROM users', (), 0))
|
|
|
|
|
|
async def get_user_by_name(username: str) -> dict[str, Any] | None:
|
|
row = await db.fetch_one('SELECT * FROM users WHERE username = ?', (username,))
|
|
return dict(row) if row is not None else None
|
|
|
|
|
|
async def get_user_by_id(uid: int) -> dict[str, Any] | None:
|
|
row = await db.fetch_one('SELECT * FROM users WHERE id = ?', (uid,))
|
|
return dict(row) if row is not None else None
|
|
|
|
|
|
# ---------- 公告 / SMTP 配置（读） ----------
|
|
|
|
async def get_announcement() -> dict[str, Any]:
|
|
row = await db.fetch_one('SELECT * FROM announcements WHERE id = 1')
|
|
return dict(row) if row else {'title': '', 'content': '', 'popup_mode': 'off'}
|
|
|
|
|
|
async def get_smtp_config() -> dict[str, Any]:
|
|
row = await db.fetch_one('SELECT * FROM smtp_config WHERE id = 1')
|
|
return dict(row) if row else {}
|
|
|
|
|
|
async def smtp_configured() -> bool:
|
|
cfg = await get_smtp_config()
|
|
return bool(cfg.get('host') and cfg.get('sender') and cfg.get('password'))
|
|
|
|
|
|
# ---------- billing ledger（只读对账） ----------
|
|
|
|
async def ledger_open_count() -> int:
|
|
return int(await db.fetch_val(
|
|
"SELECT COUNT(*) FROM billing_ledger WHERE status = 'open'", (), 0))
|