You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.6 KiB
54 lines
1.6 KiB
4 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding:utf-8 -*-
|
||
|
|
# project: 3月
|
||
|
|
# author: liuyu
|
||
|
|
# date: 2020/3/22
|
||
|
|
|
||
|
|
import requests
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
class JiGuangMsgSender(object):
|
||
|
|
BASE_URL = 'https://api.sms.jpush.cn/v1/'
|
||
|
|
|
||
|
|
def __init__(self, app_key, master_secret, sign_id, template_code):
|
||
|
|
self.template_code = template_code
|
||
|
|
self.sign_id = sign_id
|
||
|
|
self.session = requests.Session()
|
||
|
|
self.session.auth = (app_key, master_secret)
|
||
|
|
|
||
|
|
def _post(self, end_point, body):
|
||
|
|
return self._request('POST', end_point, body)
|
||
|
|
|
||
|
|
def _request(self, method, end_point, body=None):
|
||
|
|
uri = self.BASE_URL + end_point
|
||
|
|
if body is not None:
|
||
|
|
body = json.dumps(body)
|
||
|
|
r = self.session.request(method, uri, data=body)
|
||
|
|
if 0 == len(r.content):
|
||
|
|
return r.status_code
|
||
|
|
else:
|
||
|
|
return r.json()
|
||
|
|
|
||
|
|
def send_msg(self, template_code, phone, code):
|
||
|
|
body = {
|
||
|
|
'mobile': phone,
|
||
|
|
'temp_id': template_code,
|
||
|
|
'temp_para': {'code': code},
|
||
|
|
'sign_id': self.sign_id
|
||
|
|
}
|
||
|
|
response = self._post('messages', body)
|
||
|
|
if response.get('msg_id', None):
|
||
|
|
return True, response.get('msg_id')
|
||
|
|
else:
|
||
|
|
return False, response.get('error')
|
||
|
|
|
||
|
|
def send_register_msg(self, phone, code):
|
||
|
|
return self.send_msg(self.template_code.get('register'), phone, code)
|
||
|
|
|
||
|
|
def send_change_msg(self, phone, code):
|
||
|
|
return self.send_msg(self.template_code.get('change'), phone, code)
|
||
|
|
|
||
|
|
def send_login_msg(self, phone, code):
|
||
|
|
return self.send_msg(self.template_code.get('login'), phone, code)
|