gouki/vps-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(review): 修复全量审查发现的域名到期日链路、AI 资产挂靠迁移与 2FA 取码缺陷

Browse Source 
92 文件全量代码审查后的高危/中危修复，每条都带回归测试，并在未修复的 HEAD 上
反向确认过「测试确实会失败」。

- H1 域名同步整批失败：NormalizedDomain.expiry_date 契约声明为字符串，Cloudflare
  适配器把 Registrar 的 ISO 时间戳截成 "YYYY-MM-DD" 直接赋给 Date 列，写库抛
  TypeError；而 _sync_domains 的 commit 在循环外，一个域名就让整批全量回滚
  （Cloudflare 域名实际从未同步成功）。现统一契约：base.py 新增 parse_iso_date，
  适配器产出 date，_sync_domains 写库前再归一一次以兼容仍返回字符串的适配器
- M9 Vultr / DigitalOcean 的域名到期日此前完全没映射，这两家域名永远进不了续费提醒
- M1 同步失败变 500：能力分支写库失败后未回滚，Session 停在 pending-rollback，
  随后更新 last_synced_at 的 commit 抛 PendingRollbackError，真实错误被 500 掩盖
- H2 AI 资产挂靠迁移静默零命中：迁移 SQL 用枚举值 'ai_agent' 匹配 Enum 列，而
  SQLAlchemy Enum 存的是成员名 'AI_AGENT'，该分支从未生效（6 个 AI 资产的
  account_id 全为 NULL）。改用成员名并加日志留痕；账号命名沿用步骤 1 的
  「{slug}-默认」约定，避免建出与既有账号语义重复、且没有凭证的空账号
  （dry-run 实测：deepseek 资产复用已有且有凭证的 deepseek-默认）
- H3 2FA 动态码不刷新：otpState 全局常驻、收起只清 timer 不删条目，「不存在才拉」
  的守卫导致再次展开显示上一次的旧码且倒计时冻结在 0。改为每次展开都重取；
  stopOtp 同时清空旧码（避免重展开瞬间闪过已失效的码），并加请求代号，防止
  「请求在途时收起」后返回的响应又重启轮询
- M10 账号弹窗关闭不清理：外层用 v-if 关闭并不卸载组件，onUnmounted 不触发，
  OTP 轮询定时器与明文密码驻留后台。关闭时显式调 closePwd
- M2 提醒发不出去：Telegram 用 parse_mode=HTML，资产名含 < > & 时整条消息被 400
  拒收；纯文本提醒不需要任何标签，去掉 parse_mode，并把拒收正文带回便于排查
- M3 负数天文案：cbb32ea 已修好提醒消息正文，本次补齐仍在直出负数的两处
  （check_renewals.py 输出、设置页预览），后端抽出 renewal_service.day_text 统一，
  前端走 Fmt.countdownText

验证：pytest 94 passed（新增 tests/test_domain_expiry.py、tests/test_notification.py）；
本地起服务对 H2 迁移做真实 dry-run（6 个 AI 资产挂到 4 个账号、无重复账号、幂等、
非 AI 资产零影响，迁移前已备份）；浏览器端以 otpState 状态断言验证 H3/M10，
设置页续费预览确认已过期项显示「已过期 N 天」。
This commit is contained in:
gouki
2026-09-22 22:26:42 +00:00
parent 7045843db8
commit 61372f7ceb
16 changed files with 561 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/adapters/base.py
+30 -3
View File
@@ -7,7 +7,30 @@
from abc import ABC, abstractmethod
from dataclasses import asdict, dataclass, field
from typing import Optional
from datetime import date, datetime
from typing import Any, Optional
def parse_iso_date(value: Any) -> Optional[date]:
"""把到期日归一化为 date 对象（接受 date / datetime / ISO 字符串，解析失败返回 None）
各厂商返回形态不一（'2027-03-15'、'2027-03-15T00:00:00Z'、'2027-03-15T00:00:00+00:00'）。
必须统一成 date 再交给上层：expiry_date 是 SQLAlchemy Date 列，
直接塞字符串会在写库时抛 TypeError，导致整批同步在循环末尾的 commit 上全量失败。
"""
if value is None:
return None
if isinstance(value, datetime):
return value.date()
if isinstance(value, date):
return value
text = str(value).strip()
if not text:
return None
try:
return date.fromisoformat(text[:10])
except ValueError:
return None
@dataclass
@@ -38,12 +61,16 @@ class NormalizedDomain:
external_id: str
domain_name: str
registrar: Optional[str] = None
expiry_date: Optional[str] = None # ISO 日期字符串 YYYY-MM-DD
expiry_date: Optional[date] = None # 到期日 date；厂商原始值请过 parse_iso_date
status: str = "active"
raw: dict = field(default_factory=dict)
def to_dict(self) -> dict:
return asdict(self)
d = asdict(self)
# date 不可直接 JSON 序列化，转回 ISO 字符串（对外形态与旧契约一致）
if d.get("expiry_date") is not None:
d["expiry_date"] = d["expiry_date"].isoformat()
return d
@dataclass
app/adapters/cloudflare.py
+8 -5
View File
@@ -8,7 +8,7 @@ Cloudflare 无传统 VPS，主要同步托管域名（zones）；Workers/R2/Tunn
import httpx
from app.adapters.base import BaseAdapter, NormalizedDomain
from app.adapters.base import BaseAdapter, NormalizedDomain, parse_iso_date
from app.adapters.registry import register
@@ -43,7 +43,7 @@ class CloudflareAdapter(BaseAdapter):
return {"ok": False, "message": str(e)}
def list_domains(self) -> list:
# 先拉取 Registrar 域名到期日映射（domain_name -> expires_at ISO 日期）
# 先拉取 Registrar 域名到期日映射（domain_name -> date）
expiry_map = self._registrar_expiry_map()
result = []
@@ -69,7 +69,7 @@ class CloudflareAdapter(BaseAdapter):
return result
def _registrar_expiry_map(self) -> dict:
"""拉取 Cloudflare Registrar 域名到期日映射 {domain_name: YYYY-MM-DD}
"""拉取 Cloudflare Registrar 域名到期日映射 {domain_name: date}
流程：/accounts → 对每个 account 调 /accounts/{id}/registrar/domains。
若 Token 无 Registrar 权限或账号无 Registrar 域名，静默返回空映射。
@@ -92,6 +92,9 @@ class CloudflareAdapter(BaseAdapter):
name = d.get("id") or d.get("domain_name") or ""
expires_at = d.get("expires_at")
if name and expires_at:
# expires_at 为 ISO8601 时间戳，截取日期部分
expiry[name] = expires_at[:10]
# expires_at 为 ISO8601 时间戳，必须解析成 date 再交给上层：
# 直接塞截取的字符串会在写库时触发 SQLite Date 类型错误，整批同步失败
parsed = parse_iso_date(expires_at)
if parsed:
expiry[name] = parsed
return expiry
app/adapters/digitalocean.py
+3
View File
@@ -11,6 +11,7 @@ from app.adapters.base import (
BaseAdapter,
NormalizedDomain,
NormalizedVPS,
parse_iso_date,
)
from app.adapters.registry import register
@@ -92,6 +93,8 @@ class DigitalOceanAdapter(BaseAdapter):
external_id=d.get("name"),
domain_name=d.get("name"),
registrar="digitalocean",
# /domains 用 expiry_date，部分账号下会返回 expires_at，两者都兼容
expiry_date=parse_iso_date(d.get("expiry_date") or d.get("expires_at")),
raw=d,
)
)
app/adapters/vultr.py
+3
View File
@@ -11,6 +11,7 @@ from app.adapters.base import (
BaseAdapter,
NormalizedDomain,
NormalizedVPS,
parse_iso_date,
)
from app.adapters.registry import register
@@ -93,6 +94,8 @@ class VultrAdapter(BaseAdapter):
external_id=d.get("domain"),
domain_name=d.get("domain"),
registrar="vultr",
# 非 Vultr 注册的域名 expiry_date 为空，解析失败统一归为 None
expiry_date=parse_iso_date(d.get("expiry_date")),
raw=d,
)
)
app/database.py
+17 -4
View File
@@ -4,11 +4,14 @@
- metrics.db：MetricPoint / ServerInfo / SecurityCheck / EventLog
"""
import logging
from pathlib import Path
from typing import Generator
from sqlmodel import Session, SQLModel, create_engine
logger = logging.getLogger("vps-manager.db")
BASE_DIR = Path(__file__).resolve().parent.parent
DATA_DIR = BASE_DIR / "data"
DATA_DIR.mkdir(parents=True, exist_ok=True)
@@ -367,21 +370,28 @@ def _backfill_account_credentials(conn) -> None:
)
# 2. 无账号的 AI 资产按 provider 挂靠（account_id 外键 + account 字符串兼容）
# 注意：SQLAlchemy Enum 列存的是成员名（'AI_AGENT'）而不是枚举值（'ai_agent'），
# 写值会让这个分支静默零命中（同文件 _migrate_account_credentials_to_vault 用的 'PASSWORD' 即成员名）
rows = conn.execute(
sa.text(
"SELECT a.id, COALESCE(ai.provider, '') FROM assets a "
"JOIN ai_accounts ai ON ai.asset_id = a.id "
"WHERE a.asset_type = 'ai_agent' AND a.account_id IS NULL AND (a.account IS NULL OR a.account = '')"
"WHERE a.asset_type = 'AI_AGENT' AND a.account_id IS NULL AND (a.account IS NULL OR a.account = '')"
)
).fetchall()
attached = 0
for asset_id, provider in rows:
if not provider:
continue
acc_id = _ensure_account(provider, provider)
# 沿用步骤 1 的「{slug}-默认」命名：同平台已有账号时直接复用。
# 用裸 provider 名会建出「deepseek」与「deepseek-默认」两个语义重复的账号，
# 而新建那个没凭证，在 UI 上点同步/测试会直接报「缺少凭证配置」
acc_id = _ensure_account(f"{provider}-默认", provider)
conn.execute(
sa.text("UPDATE assets SET account = :acc, account_id = :aid WHERE id = :id"),
{"acc": provider, "aid": acc_id, "id": asset_id},
)
attached += 1
# 账号尚无 API 配置时，把该资产的 api_key 写入账号配置（解密后重组 JSON 再加密）
key_row = conn.execute(
sa.text("SELECT api_key_encrypted FROM ai_accounts WHERE asset_id = :id"),
@@ -396,11 +406,14 @@ def _backfill_account_credentials(conn) -> None:
conn.execute(
sa.text(
"UPDATE accounts SET api_config_encrypted = :cfg "
"WHERE name = :name AND COALESCE(platform,'') = :platform "
"WHERE id = :aid "
"AND (api_config_encrypted IS NULL OR api_config_encrypted = '')"
),
{"cfg": cfg_enc, "name": provider, "platform": provider},
{"cfg": cfg_enc, "aid": acc_id},
)
if attached:
# 这段迁移每次启动都会重跑（幂等），有实际改动时必须留痕迹便于事后核对
logger.info("凭证下沉迁移：本次将 %s 个 AI 资产挂靠到账号", attached)
def get_session() -> Generator[Session, None, None]:
app/services/notification_service.py
+11 -1
View File
@@ -22,13 +22,23 @@ def send_telegram(message: str) -> dict:
json={
"chat_id": settings.TELEGRAM_CHAT_ID,
"text": message,
"parse_mode": "HTML",
# 不传 parse_mode：消息里嵌的是资产名/平台名等用户数据，
# 含 < > & 时 HTML 解析会让整条消息 400 失败（提醒根本发不出去），
# 而纯文本提醒本来也不需要任何标签
"disable_web_page_preview": True,
},
timeout=15,
)
resp.raise_for_status()
return {"ok": True, "channel": "telegram", "message": "发送成功"}
except httpx.HTTPStatusError as e:
# Telegram 把拒收原因放在响应体里（bot 被拉黑、chat_id 错等），
# 只留状态码无法排查，把正文一并带回
return {
"ok": False,
"channel": "telegram",
"message": f"HTTP {e.response.status_code}：{e.response.text[:200]}",
}
except Exception as e: # noqa: BLE001
return {"ok": False, "channel": "telegram", "message": str(e)}
app/services/renewal_service.py
+14 -7
View File
@@ -35,6 +35,19 @@ def get_expiring_assets(session: Session, threshold_days: int) -> list:
return expiring
def day_text(days: int) -> str:
"""单项到期文案（提醒消息与 check_renewals 脚本共用）
days 可以为负（已过期），直接拼成「{days} 天后到期」会得到「-3 天后到期」；
前端对应的统一文案是 Fmt.countdownText，两边保持一致。
"""
if days < 0:
return f"已过期 {-days} 天"
if days == 0:
return "今天到期"
return f"{days} 天后"
def build_renewal_message(expiring: list, threshold_days: int) -> str:
"""生成续费提醒消息（纯文本，兼容 Telegram/邮件）"""
if not expiring:
@@ -46,13 +59,7 @@ def build_renewal_message(expiring: list, threshold_days: int) -> str:
head = f"⏰ 资产续费提醒（{threshold_days} 天内到期，共 {len(expiring)} 项）"
lines = [head, ""]
for asset, days in expiring:
if days < 0:
tag = f"（已过期 {-days} 天）"
elif days == 0:
tag = "（今天到期）"
else:
tag = f"（{days} 天后）"
lines.append(f"• {asset.name} [{asset.provider}] {tag} 到期 {asset.expiry_date}")
lines.append(f"• {asset.name} [{asset.provider}] （{day_text(days)}） 到期 {asset.expiry_date}")
return "\n".join(lines)
app/services/sync_service.py
+11 -4
View File
@@ -19,7 +19,7 @@ from sqlmodel import Session, select
logger = logging.getLogger("vps-manager.sync")
from app.adapters import registry
from app.adapters.base import BaseAdapter
from app.adapters.base import BaseAdapter, parse_iso_date
from app.core import crypto
from app.core.timeutils import utcnow
from app.models.asset import (
@@ -227,11 +227,14 @@ def _sync_vps(session: Session, provider: Provider, adapter: BaseAdapter, accoun
def _sync_domains(session: Session, provider: Provider, adapter: BaseAdapter, account_id: Optional[int] = None) -> dict:
created = updated = 0
for dom in adapter.list_domains():
# 适配器契约上 expiry_date 已是 date；这里再归一一次，
# 防止旧适配器/自定义适配器返回字符串时整个 commit 抛 TypeError
expiry = parse_iso_date(dom.expiry_date)
existing = _find_asset(session, provider.id, dom.external_id, AssetType.DOMAIN)
if existing:
existing.status = _norm_status(dom.status)
if dom.expiry_date:
existing.expiry_date = dom.expiry_date
if expiry:
existing.expiry_date = expiry
if account_id:
existing.account_id = account_id
session.add(existing)
@@ -252,7 +255,7 @@ def _sync_domains(session: Session, provider: Provider, adapter: BaseAdapter, ac
external_id=dom.external_id,
account_id=account_id,
status=_norm_status(dom.status),
expiry_date=dom.expiry_date,
expiry_date=expiry,
)
session.add(asset)
session.flush() # 获取 asset.id，统一在循环外提交
@@ -297,11 +300,15 @@ def sync_account(session: Session, account_id: int) -> dict:
try:
result["vps"] = _sync_vps(session, provider, adapter, account.id)
except Exception as e: # noqa: BLE001
# 写库失败会让 Session 进入 pending-rollback 状态，必须先回滚才能继续，
# 否则下面更新 last_synced_at 的 commit 会抛 PendingRollbackError → 接口 500
session.rollback()
result["vps_error"] = str(e)
if caps["list_domains"]:
try:
result["domains"] = _sync_domains(session, provider, adapter, account.id)
except Exception as e: # noqa: BLE001
session.rollback()
result["domains_error"] = str(e)
if caps["get_account"]:
try: