135 lines
6.3 KiB
Python
135 lines
6.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
# @Time: 2025/6/16 15:43
|
||
|
||
# -*- coding: utf-8 -*-
|
||
# @Time: 2023/4/23 17:18
|
||
|
||
import datetime
|
||
|
||
#从Python SDK导入SMS配置管理模块以及安全认证模块
|
||
from baidubce.bce_client_configuration import BceClientConfiguration
|
||
from baidubce.auth.bce_credentials import BceCredentials
|
||
|
||
# 导入SMS相关模块
|
||
import baidubce.services.sms.sms_client as sms
|
||
import baidubce.exception as ex
|
||
|
||
from sqlor.dbpools import DBPools
|
||
from appPublic.uniqueID import getID as uuid
|
||
|
||
|
||
class BaiduSMS:
|
||
# 百度短信错误码文档（需替换为实际文档链接）
|
||
doc = "https://cloud.baidu.com/doc/SMS/s/zjwvxry6e"
|
||
|
||
def __init__(self):
|
||
# 替换为您的百度AK/SK
|
||
self.access_key = 'ALTAKPk92fX9cgGDax83yNL8mP'
|
||
self.access_key_secret = '9b16b8efd4dc463d8bbd5462db1db8a5'
|
||
self.host = 'sms.bj.baidubce.com'
|
||
self.sms_client = self.create_client(self.access_key, self.access_key_secret, self.host)
|
||
# 替换为您的百度短信签名名称
|
||
# self.signature_id = 'sms-sign-qQHYeC17077' # 开元云科技
|
||
self.signature_id = 'sms-sign-BqOhYB33019' # 开元云
|
||
# 短信模板类型映射（键为业务类型，值为对应模板ID）
|
||
self.sms_types = {
|
||
"注册登录验证": "sms-tpl-123", # 示例模板ID
|
||
}
|
||
|
||
async def send(self, stype, template_id, phone, params) -> dict:
|
||
"""
|
||
主发送逻辑（适配百度API）
|
||
"""
|
||
# 百度短信请求参数（根据实际API调整字段名）
|
||
try:
|
||
resp = self.sms_client.send_message(signature_id=self.signature_id,
|
||
template_id=template_id,
|
||
mobile=phone,
|
||
content_var_dict=params
|
||
) # 百度异步发送方法
|
||
return await self.__validation(stype=stype, template_id=template_id, params=params, phone=phone, resp=resp)
|
||
except ex.BceHttpClientError as e:
|
||
if isinstance(e.last_error, ex.BceServerError):
|
||
print('send request failed. Response %s, code: %s, request_id: %s'
|
||
% (e.last_error.status_code, e.last_error.code, e.last_error.request_id))
|
||
else:
|
||
print('send request failed. Unknown exception: %s' % e)
|
||
return {'status': False, 'msg': str(e)}
|
||
|
||
async def send_vcode(self, phone: str, stype: str, vcode) -> dict:
|
||
"""
|
||
发送验证码
|
||
"""
|
||
db = DBPools()
|
||
async with db.sqlorContext('kboss') as sor:
|
||
# 假设数据库表结构与原代码一致（sms_template表）
|
||
template_info = await sor.R('sms_template', {'name': stype, 'del_flg': '0'})
|
||
if template_info:
|
||
template_id = template_info[0]['code'] # 假设code字段存储百度模板ID
|
||
else:
|
||
log = {
|
||
'id': uuid(),
|
||
'send_type': stype,
|
||
'mobile': phone,
|
||
'message': str(vcode),
|
||
'send_time': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||
'send_status': 0,
|
||
'remark': f'未找到模板类型: {stype}'
|
||
}
|
||
await sor.C('sms_record', log)
|
||
return {'status': False, 'msg': '模板未配置，请检查sms_template表'}
|
||
|
||
return await self.send(stype, template_id, phone, vcode)
|
||
|
||
async def __validation(self, stype, template_id, params, phone, resp) -> dict:
|
||
"""
|
||
验证百度短信返回结果
|
||
"""
|
||
db = DBPools()
|
||
async with db.sqlorContext('kboss') as sor:
|
||
send_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
log = {
|
||
'id': uuid(),
|
||
'send_type': stype,
|
||
'mobile': phone,
|
||
'message': str(params),
|
||
'send_time': send_time,
|
||
'send_status': 1
|
||
}
|
||
|
||
# 百度通常用0表示成功（需根据实际API调整）
|
||
if resp.code == '1000':
|
||
msg = f'{send_time} {phone} 百度短信发送成功，code: {resp.code}'
|
||
log['send_status'] = 1
|
||
await sor.C('sms_record', log)
|
||
return {'status': True, 'msg': msg}
|
||
else:
|
||
msg = f'{send_time} {phone} 百度短信发送失败，code: {resp.code}，参考文档: {self.doc}'
|
||
log['send_status'] = 0
|
||
log['remark'] = msg
|
||
await sor.C('sms_record', log)
|
||
return {'status': False, 'msg': msg}
|
||
|
||
def create_client(self, access_key: str, access_key_secret: str, host: str):
|
||
"""
|
||
创建百度短信客户端
|
||
"""
|
||
baiDuSmsConfig = BceClientConfiguration(credentials=BceCredentials(access_key, access_key_secret), endpoint=host)
|
||
sms_client = sms.SmsClient(baiDuSmsConfig)
|
||
return sms_client
|
||
|
||
|
||
async def send_baidu_vcode(phone: str, stype: str, vcode: dict) -> dict:
|
||
"""
|
||
外部调用入口（与原代码结构一致）
|
||
"""
|
||
result = await BaiduSMS().send_vcode(phone, stype, vcode)
|
||
return result
|
||
|
||
# if __name__ == '__main__':
|
||
# loop = asyncio.get_event_loop()
|
||
# loop.run_until_complete(send_baidu_vcode('18053191417', '用户注册登录验证', {'SMSvCode': '989898'}))
|
||
# loop.run_until_complete(send_baidu_vcode('18053191417', '余额不足提醒', {'time': '2025-06-18 17:47:00'}))
|
||
# loop.run_until_complete(send_baidu_vcode('18053191417', '用户欠费通知', {'time': '2025-06-18 17:47:00', 'productname': 'BCC_云主机'}))
|
||
# loop.run_until_complete(send_baidu_vcode('18053191417', '百度kafka普通验证码', {'content': '验证码为：636814,感谢您使用百度云服务，请填写完整完成验证。'}))
|
||
# loop.run_until_complete(send_baidu_vcode('18053191417', '百度kafka普通通知', {'content': '尊敬的用户，您的百度智能云账户（登录名：-）的 弹性公网IP资源（区域：华北-北京 IP：120.48.5.167 ,回收站中 ）将于2025-06-18释放， 截至当前仅剩1天。 为保证正常使用，请及时续费或保留好数据，资源详情请查看邮件，感谢支持！ '})) |