74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
独立的短信发送脚本（绕过 eventlet DNS 问题）
|
||
|
||
使用方式：
|
||
python sms_sender.py <phone> <code> <template_id> <secret_id> <secret_key> <app_id> <sign_name>
|
||
|
||
返回值：
|
||
成功返回 0，失败返回 1
|
||
"""
|
||
|
||
import sys
|
||
import json
|
||
|
||
|
||
def send_sms(phone, code, template_id, secret_id, secret_key, app_id, sign_name):
|
||
"""发送短信验证码"""
|
||
try:
|
||
from tencentcloud.common import credential
|
||
from tencentcloud.common.profile.client_profile import ClientProfile
|
||
from tencentcloud.common.profile.http_profile import HttpProfile
|
||
from tencentcloud.sms.v20210111 import sms_client, models
|
||
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
|
||
|
||
cred = credential.Credential(secret_id, secret_key)
|
||
httpProfile = HttpProfile()
|
||
httpProfile.endpoint = "sms.tencentcloudapi.com"
|
||
|
||
clientProfile = ClientProfile()
|
||
clientProfile.httpProfile = httpProfile
|
||
client = sms_client.SmsClient(cred, "ap-beijing", clientProfile)
|
||
|
||
req = models.SendSmsRequest()
|
||
# 登录模板需要两个参数：验证码和有效期
|
||
template_params = [code, "5"] if "login" in template_id.lower() or template_id == "2365498" else [code]
|
||
|
||
params = {
|
||
"PhoneNumberSet": [phone],
|
||
"SmsSdkAppId": app_id,
|
||
"TemplateId": template_id,
|
||
"SignName": sign_name,
|
||
"TemplateParamSet": template_params
|
||
}
|
||
req.from_json_string(json.dumps(params))
|
||
|
||
resp = client.SendSms(req)
|
||
print(f"SMS sent successfully: {resp.to_json_string()}")
|
||
return True
|
||
|
||
except TencentCloudSDKException as err:
|
||
print(f"SMS Error: {err}")
|
||
return False
|
||
except Exception as e:
|
||
print(f"Unexpected error: {e}")
|
||
return False
|
||
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv) != 8:
|
||
print("Usage: python sms_sender.py <phone> <code> <template_id> <secret_id> <secret_key> <app_id> <sign_name>")
|
||
sys.exit(1)
|
||
|
||
phone = sys.argv[1]
|
||
code = sys.argv[2]
|
||
template_id = sys.argv[3]
|
||
secret_id = sys.argv[4]
|
||
secret_key = sys.argv[5]
|
||
app_id = sys.argv[6]
|
||
sign_name = sys.argv[7]
|
||
|
||
success = send_sms(phone, code, template_id, secret_id, secret_key, app_id, sign_name)
|
||
sys.exit(0 if success else 1)
|