105 lines
4.3 KiB
Python
105 lines
4.3 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""短信生态配置写入脚本（迁移自 PineSoundServer .env）。
|
|||
|
|
|
|||
|
|
把阿里云凭据/签名/区域与 PineSound 短信模板（仅本项目需要的 T1 登录）写入
|
|||
|
|
system_configs 表：
|
|||
|
|
|
|||
|
|
- sms.provider -> aliyun
|
|||
|
|
- sms.aliyun -> 凭据/签名/区域（与 PineSoundServer .env 完全一致：
|
|||
|
|
ACCESS_KEY_ID / ACCESS_KEY_SECRET / REGION_ID=cn-qingdao /
|
|||
|
|
SIGN_NAME=派音人工智能）
|
|||
|
|
- sms.template.login -> SMS_510980133（PineSound T1 登录验证码，变量 code）
|
|||
|
|
- sms.template.register -> SMS_510980133（本项目注册=手机号登录一体，复用 T1）
|
|||
|
|
|
|||
|
|
PineSound 其余模板（修改密码/绑定手机/找回密码/注销/充值/内测）本项目当前业务
|
|||
|
|
未使用，不迁移；notification / order_status / appointment 场景 PineSound 无匹配
|
|||
|
|
模板，保持留空（stub 打印，不阻断业务）。
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
python scripts/db/set_sms_config.py # 实际执行
|
|||
|
|
python scripts/db/set_sms_config.py --dry-run # 只预览不写库
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import asyncio
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
|
|||
|
|
load_dotenv(Path(__file__).resolve().parent.parent.parent / ".env")
|
|||
|
|
|
|||
|
|
from sqlalchemy import text # noqa: E402
|
|||
|
|
|
|||
|
|
from app.infrastructure.db import AsyncSessionLocal # noqa: E402
|
|||
|
|
from app.infrastructure.repositories import utcnow_iso # noqa: E402
|
|||
|
|
|
|||
|
|
# 与 PineSoundServer .env 完全一致的阿里云短信配置
|
|||
|
|
ALIYUN_CONFIG = {
|
|||
|
|
"access_key_id": "LTAI5t6ZU2Nf8qbGpEpn7bzN",
|
|||
|
|
"access_key_secret": "lnNkxiXvNVsCPdc7hVZSXz6goOz8YJ",
|
|||
|
|
"sign_name": "派音人工智能",
|
|||
|
|
"region_id": "cn-qingdao",
|
|||
|
|
"endpoint": "dysmsapi.aliyuncs.com",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# key -> (value, description)；幂等覆盖：已存在的键也更新为 PineSound 配置。
|
|||
|
|
SMS_CONFIGS: list[tuple[str, str, str]] = [
|
|||
|
|
("sms.provider", "aliyun", "短信 provider（stub|aliyun）"),
|
|||
|
|
("sms.aliyun",
|
|||
|
|
__import__("json").dumps(ALIYUN_CONFIG, ensure_ascii=False),
|
|||
|
|
"阿里云短信总配置（与 PineSoundServer .env 一致）"),
|
|||
|
|
("sms.template.login",
|
|||
|
|
'{"template_id":"SMS_510980133","variables":["code"]}',
|
|||
|
|
"登录验证码模板（PineSound T1：验证码 code，5分钟有效，请勿泄露）"),
|
|||
|
|
("sms.template.register",
|
|||
|
|
'{"template_id":"SMS_510980133","variables":["code"]}',
|
|||
|
|
"注册验证码模板（本项目注册=手机号登录一体，复用 T1）"),
|
|||
|
|
("sms.template.notification",
|
|||
|
|
'{"template_id":"","variables":[]}',
|
|||
|
|
"通知提醒模板（PineSound 无匹配模板，暂留空）"),
|
|||
|
|
("sms.template.order_status",
|
|||
|
|
'{"template_id":"","variables":["order_no","status"]}',
|
|||
|
|
"订单状态变动模板（PineSound 无匹配模板，暂留空）"),
|
|||
|
|
("sms.template.appointment",
|
|||
|
|
'{"template_id":"","variables":["service_name","schedule_at","contact"]}',
|
|||
|
|
"服务预约模板（PineSound 无匹配模板，暂留空）"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def main(dry_run: bool) -> int:
|
|||
|
|
now = utcnow_iso()
|
|||
|
|
async with AsyncSessionLocal() as session:
|
|||
|
|
for key, value, description in SMS_CONFIGS:
|
|||
|
|
exists = (
|
|||
|
|
await session.execute(
|
|||
|
|
text("SELECT `key` FROM system_configs WHERE `key` = :k"),
|
|||
|
|
{"k": key},
|
|||
|
|
)
|
|||
|
|
).scalar_one_or_none()
|
|||
|
|
if exists:
|
|||
|
|
sql = text(
|
|||
|
|
"UPDATE system_configs SET `value` = :v, description = :d, "
|
|||
|
|
"updated_at = :now WHERE `key` = :k"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
sql = text(
|
|||
|
|
"INSERT INTO system_configs (`key`, `value`, `description`, "
|
|||
|
|
"updated_at) VALUES (:k, :v, :d, :now)"
|
|||
|
|
)
|
|||
|
|
if dry_run:
|
|||
|
|
print(f"[dry-run] {'UPDATE' if exists else 'INSERT'} {key} = {value}")
|
|||
|
|
continue
|
|||
|
|
await session.execute(sql, {"k": key, "v": value, "d": description, "now": now})
|
|||
|
|
if not dry_run:
|
|||
|
|
await session.commit()
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
parser = argparse.ArgumentParser(description="写入短信生态配置（PineSound 迁移）")
|
|||
|
|
parser.add_argument("--dry-run", action="store_true", help="只预览不写库")
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
raise SystemExit(asyncio.run(main(args.dry_run)))
|