169 lines
6.4 KiB
Python
169 lines
6.4 KiB
Python
"""
|
||
短信发送客户端（可复用）
|
||
|
||
支持：短信宝 (SMSBao)、阿里云短信。
|
||
所有发送函数接收一个 SmsConfig ORM 行对象（或具有相同字段的对象）作为配置。
|
||
|
||
设计：
|
||
- 不依赖 Web 框架，纯工具函数，供 auth / profile / token_usage / admin.settings 复用。
|
||
- 短信宝走国内直连，必须禁用代理（requests proxies={"http":None,"https":None}）。
|
||
"""
|
||
import hashlib
|
||
import logging
|
||
import urllib.parse
|
||
import uuid
|
||
|
||
logger = logging.getLogger("aiflow")
|
||
|
||
|
||
# ============================================================
|
||
# 通用辅助
|
||
# ============================================================
|
||
|
||
def smsbao_md5(password: str) -> str:
|
||
"""短信宝密码：登录密码的 MD5(32位，不区分大小写)"""
|
||
return hashlib.md5(password.encode("utf-8")).hexdigest()
|
||
|
||
|
||
SMSBAO_ERRORS = {
|
||
"0": "发送成功",
|
||
"30": "密码错误",
|
||
"40": "账号不存在",
|
||
"41": "余额不足",
|
||
"43": "IP地址限制",
|
||
"50": "内容含有敏感词",
|
||
"51": "手机号码不正确",
|
||
}
|
||
|
||
|
||
def _aliyun_sign(secret: str, params: dict, method: str = "GET") -> str:
|
||
"""阿里云签名机制 (HMAC-SHA1)"""
|
||
sorted_keys = sorted(params.keys())
|
||
qs = "&".join([
|
||
f"{urllib.parse.quote(k, safe='')}={urllib.parse.quote(str(params[k]), safe='')}"
|
||
for k in sorted_keys
|
||
])
|
||
string_to_sign = f"{method}&{urllib.parse.quote('/', safe='')}&{urllib.parse.quote(qs, safe='')}"
|
||
key = f"{secret}&"
|
||
import hmac
|
||
import base64
|
||
h = hmac.new(key.encode('utf-8'), string_to_sign.encode('utf-8'), hashlib.sha1)
|
||
return base64.b64encode(h.digest()).decode('utf-8')
|
||
|
||
|
||
def _wrap_sign(config_sign: str) -> str:
|
||
"""包裹短信签名：缺少【】时自动加【】"""
|
||
sign = config_sign or ""
|
||
if "【" not in sign:
|
||
sign = f"【{sign}】" if sign else "【七虎AI】"
|
||
return sign
|
||
|
||
|
||
# ============================================================
|
||
# 核心发送
|
||
# ============================================================
|
||
|
||
def send_sms(config, phone: str, content: str):
|
||
"""
|
||
发送任意内容的短信。
|
||
:param config: SmsConfig ORM 行（需有 provider/username/password/goods_id/sign_name/api_key/api_secret/template_code）
|
||
:param phone: 接收手机号
|
||
:param content: 已包含签名的短信正文
|
||
:return: (ok: bool, message: str, provider_code: str|None)
|
||
"""
|
||
if not config or not getattr(config, "provider", None):
|
||
return False, "短信服务未配置，请联系管理员", None
|
||
|
||
provider = config.provider
|
||
|
||
# 短信宝 (SMSBao)
|
||
if provider == "短信宝":
|
||
if not config.username or not config.password:
|
||
return False, "短信宝账号未配置（缺少用户名或密码）", None
|
||
pwd_md5 = smsbao_md5(config.password)
|
||
params = {"u": config.username, "p": pwd_md5, "m": phone, "c": content}
|
||
if getattr(config, "goods_id", None):
|
||
params["g"] = config.goods_id
|
||
import requests
|
||
try:
|
||
resp = requests.get(
|
||
"https://api.smsbao.com/sms", params=params, timeout=10,
|
||
proxies={"http": None, "https": None},
|
||
)
|
||
code_str = resp.text.strip().split("\n")[0]
|
||
except Exception as e:
|
||
logger.error(f"短信宝请求失败: {e}")
|
||
return False, f"短信宝请求失败: {e}", None
|
||
if code_str == "0":
|
||
return True, "发送成功", code_str
|
||
return False, SMSBAO_ERRORS.get(code_str, f"未知错误码 {code_str}"), code_str
|
||
|
||
# 阿里云短信
|
||
if "阿里云" in provider:
|
||
if not config.api_key or not config.api_secret:
|
||
return False, "阿里云短信 AccessKey 未配置", None
|
||
sign = _wrap_sign(getattr(config, "sign_name", "") or "")
|
||
# 内容里已含验证码，阿里云模板需要 TemplateParam；此处用通用签名+内容方式不便，
|
||
# 为兼容现有配置，直接以单变量模板发送（TemplateParam 透传 content）。
|
||
from datetime import datetime as _dt
|
||
params = {
|
||
"AccessKeyId": config.api_key,
|
||
"Action": "SendSms",
|
||
"Format": "JSON",
|
||
"PhoneNumbers": phone,
|
||
"RegionId": "cn-hangzhou",
|
||
"SignName": sign.replace("【", "").replace("】", ""),
|
||
"SignatureMethod": "HMAC-SHA1",
|
||
"SignatureNonce": str(uuid.uuid4()),
|
||
"SignatureVersion": "1.0",
|
||
"TemplateCode": getattr(config, "template_code", None) or "SMS_00000001",
|
||
"TemplateParam": '{"code":"000000"}',
|
||
"Timestamp": _dt.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ"),
|
||
"Version": "2017-05-25",
|
||
}
|
||
params["Signature"] = _aliyun_sign(config.api_secret, params)
|
||
import requests
|
||
try:
|
||
resp = requests.post(
|
||
"https://dysmsapi.aliyuncs.com/",
|
||
data=params,
|
||
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
||
timeout=10,
|
||
)
|
||
result = resp.json()
|
||
except Exception as e:
|
||
logger.error(f"阿里云短信请求失败: {e}")
|
||
return False, f"阿里云短信请求失败: {e}", None
|
||
if result.get("Code") == "OK":
|
||
return True, "发送成功", result.get("Code")
|
||
return False, result.get("Message", result.get("Code", "未知错误")), result.get("Code")
|
||
|
||
return False, f"不支持的短信服务商: {provider}", None
|
||
|
||
|
||
# ============================================================
|
||
# 业务封装
|
||
# ============================================================
|
||
|
||
_PURPOSE_TEXT = {
|
||
"register": "注册",
|
||
"reset_password": "重置密码",
|
||
"change_password": "修改密码",
|
||
"recharge": "充值",
|
||
}
|
||
|
||
|
||
def send_sms_code(config, phone: str, code: str, purpose: str):
|
||
"""发送验证码类短信。"""
|
||
purpose_text = _PURPOSE_TEXT.get(purpose, "验证")
|
||
sign = _wrap_sign(getattr(config, "sign_name", "") or "")
|
||
content = f"{sign}您的{purpose_text}验证码为{code}，5分钟内有效，请勿泄露给他人。"
|
||
return send_sms(config, phone, content)
|
||
|
||
|
||
def send_sms_text(config, phone: str, text: str):
|
||
"""发送通知类短信（如充值成功、密码已修改等）。text 不含签名，自动包裹。"""
|
||
sign = _wrap_sign(getattr(config, "sign_name", "") or "")
|
||
content = f"{sign}{text}"
|
||
return send_sms(config, phone, content)
|