Files
vps-manager/app/services/notification_service.py
T
gouki 61372f7ceb fix(review): 修复全量审查发现的域名到期日链路、AI 资产挂靠迁移与 2FA 取码缺陷
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 天」。
2026-09-22 22:26:42 +00:00

83 lines
3.2 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
"""通知服务:Telegram / 邮件多渠道发送
根据 .env 配置的渠道发送通知。未配置的渠道自动跳过。
"""
import smtplib
from email.mime.text import MIMEText
import httpx
from app.core.config import settings
def send_telegram(message: str) -> dict:
"""通过 Telegram Bot 发送消息"""
if not settings.TELEGRAM_BOT_TOKEN or not settings.TELEGRAM_CHAT_ID:
return {"ok": False, "channel": "telegram", "message": "未配置 Telegram"}
url = f"https://api.telegram.org/bot{settings.TELEGRAM_BOT_TOKEN}/sendMessage"
try:
resp = httpx.post(
url,
json={
"chat_id": settings.TELEGRAM_CHAT_ID,
"text": message,
# 不传 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)}
def send_email(message: str, subject: str = "资产续费提醒") -> dict:
"""通过 SMTP 发送邮件"""
if not settings.SMTP_HOST or not settings.SMTP_USER:
return {"ok": False, "channel": "email", "message": "未配置邮件"}
try:
msg = MIMEText(message, "plain", "utf-8")
msg["Subject"] = subject
msg["From"] = settings.SMTP_FROM or settings.SMTP_USER
msg["To"] = settings.SMTP_TO or settings.SMTP_USER
with smtplib.SMTP_SSL(settings.SMTP_HOST, settings.SMTP_PORT, timeout=15) as server:
server.login(settings.SMTP_USER, settings.SMTP_PASSWORD)
server.send_message(msg)
return {"ok": True, "channel": "email", "message": "发送成功"}
except Exception as e: # noqa: BLE001
return {"ok": False, "channel": "email", "message": str(e)}
def notify(message: str, subject: str = "资产续费提醒") -> list:
"""发送到所有已配置的渠道,返回各渠道结果"""
results = []
if settings.TELEGRAM_BOT_TOKEN:
results.append(send_telegram(message))
if settings.SMTP_HOST:
results.append(send_email(message, subject))
if not results:
results.append({"ok": False, "channel": "none", "message": "未配置任何通知渠道"})
return results
def configured_channels() -> list:
"""返回已配置的通知渠道列表"""
channels = []
if settings.TELEGRAM_BOT_TOKEN and settings.TELEGRAM_CHAT_ID:
channels.append("telegram")
if settings.SMTP_HOST and settings.SMTP_USER:
channels.append("email")
return channels