gouki
61372f7ceb
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 天」。
98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
"""续费检查服务
|
||
|
||
检查即将到期的资产，生成提醒消息并通过通知渠道发送。
|
||
"""
|
||
|
||
from datetime import date, timedelta
|
||
|
||
from sqlmodel import Session, select
|
||
|
||
from app.core.config import settings
|
||
from app.models.asset import Asset, AssetStatus
|
||
from app.services import notification_service
|
||
|
||
# 已注销的资产不参与续费提醒
|
||
_SKIP_STATUS = {AssetStatus.CANCELLED}
|
||
|
||
|
||
def get_expiring_assets(session: Session, threshold_days: int) -> list:
|
||
"""返回 threshold_days 天内到期（含已过期未处理）的资产列表 [(asset, days), ...]，按剩余天数升序
|
||
|
||
到期日不设下界：过了续费日仍未注销/续费的资产，提醒不能就此消失。
|
||
性能优化：日期范围与状态过滤均下推到 SQL 层。
|
||
"""
|
||
today = date.today()
|
||
deadline = today + timedelta(days=threshold_days)
|
||
assets = session.exec(
|
||
select(Asset).where(
|
||
Asset.expiry_date.is_not(None),
|
||
Asset.expiry_date <= deadline,
|
||
Asset.status.notin_(_SKIP_STATUS),
|
||
)
|
||
).all()
|
||
expiring = [(asset, (asset.expiry_date - today).days) for asset in assets]
|
||
expiring.sort(key=lambda x: x[1])
|
||
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:
|
||
return ""
|
||
overdue = sum(1 for _, d in expiring if d < 0)
|
||
if overdue:
|
||
head = f"⏰ 资产续费提醒（已过期 {overdue} 项 + {threshold_days} 天内到期 {len(expiring) - overdue} 项）"
|
||
else:
|
||
head = f"⏰ 资产续费提醒（{threshold_days} 天内到期，共 {len(expiring)} 项）"
|
||
lines = [head, ""]
|
||
for asset, days in expiring:
|
||
lines.append(f"• {asset.name} [{asset.provider}] （{day_text(days)}） 到期 {asset.expiry_date}")
|
||
return "\n".join(lines)
|
||
|
||
|
||
def check_renewals(session: Session, threshold_days: int = None, send: bool = True) -> dict:
|
||
"""检查续费并（可选）发送提醒
|
||
|
||
threshold_days: 提醒阈值天数，默认取配置 RENEWAL_THRESHOLD_DAYS
|
||
send: 是否实际发送通知（False 时仅返回预览）
|
||
"""
|
||
if threshold_days is None:
|
||
threshold_days = settings.RENEWAL_THRESHOLD_DAYS
|
||
|
||
expiring = get_expiring_assets(session, threshold_days)
|
||
message = build_renewal_message(expiring, threshold_days)
|
||
|
||
result = {
|
||
"threshold_days": threshold_days,
|
||
"expiring_count": len(expiring),
|
||
"expiring": [
|
||
{
|
||
"name": asset.name,
|
||
"provider": asset.provider,
|
||
"days": days,
|
||
"expiry_date": str(asset.expiry_date),
|
||
}
|
||
for asset, days in expiring
|
||
],
|
||
"message": message,
|
||
}
|
||
|
||
if send and message:
|
||
result["notifications"] = notification_service.notify(message)
|
||
else:
|
||
result["notifications"] = []
|
||
return result
|