124 lines
4.2 KiB
Python
124 lines
4.2 KiB
Python
# coding: utf-8
|
||
# +-------------------------------------------------------------------
|
||
# | yakpanel
|
||
# +-------------------------------------------------------------------
|
||
# | Copyright (c) 2015-2020 yakpanel(http://www.yakpanel.com) All rights reserved.
|
||
# +-------------------------------------------------------------------
|
||
# | Author: baozi
|
||
# | 消息通道 短信模块(新)
|
||
# +-------------------------------------------------------------------
|
||
import json
|
||
import os
|
||
import time
|
||
import traceback
|
||
from typing import Union, Optional
|
||
from mod.base.push_mod import SenderConfig
|
||
from .util import write_push_log, PANEL_PATH, write_file, read_file, public_http_post
|
||
|
||
|
||
class SMSMsg:
|
||
API_URL = 'https://www.yakpanel.com/api/wmsg'
|
||
USER_PATH = '{}/data/userInfo.json'.format(PANEL_PATH)
|
||
|
||
# 构造方法
|
||
def __init__(self, msm_data: dict):
|
||
self.id = msm_data["id"]
|
||
self.data = msm_data["data"]
|
||
self.user_info = None
|
||
try:
|
||
self.user_info = json.loads(read_file(self.USER_PATH))
|
||
except:
|
||
self.user_info = None
|
||
|
||
self._PDATA = {
|
||
"access_key": "" if self.user_info is None else 'B' * 32,
|
||
"data": {}
|
||
}
|
||
|
||
def refresh_config(self, force=False):
|
||
if "last_refresh_time" not in self.data:
|
||
self.data["last_refresh_time"] = 0
|
||
if self.data.get("last_refresh_time") + 60 * 60 * 24 < time.time() or force: # 一天最多更新一次
|
||
result = self._request('get_user_sms')
|
||
if not isinstance(result, dict) or ("status" in result and not result["status"]):
|
||
return {
|
||
"count": 0,
|
||
"total": 0
|
||
}
|
||
sc = SenderConfig()
|
||
tmp = sc.get_by_id(self.id)
|
||
if tmp is not None:
|
||
result["last_refresh_time"] = time.time()
|
||
tmp["data"] = result
|
||
sc.save_config()
|
||
else:
|
||
result = self.data
|
||
return result
|
||
|
||
def send_msg(self, sm_type: str, sm_args: dict):
|
||
"""
|
||
@发送短信
|
||
@sm_type 预警类型, ssl_end|YakPanel SSL到期提醒
|
||
@sm_args 预警参数
|
||
"""
|
||
if not self.user_info:
|
||
return "未成功绑定官网账号，无法发送信息，请尝试重新绑定"
|
||
tmp = sm_type.split('|')
|
||
if "|" in sm_type and len(tmp) >= 2:
|
||
s_type = tmp[0]
|
||
title = tmp[1]
|
||
else:
|
||
s_type = sm_type
|
||
title = 'YakPanel 告警提醒'
|
||
|
||
sm_args = self.canonical_data(sm_args)
|
||
self._PDATA['data']['sm_type'] = s_type
|
||
self._PDATA['data']['sm_args'] = sm_args
|
||
print(s_type)
|
||
print(sm_args)
|
||
result = self._request('send_msg')
|
||
u_key = '{}****{}'.format(self.user_info['username'][:3], self.user_info['username'][-3:])
|
||
print(result)
|
||
if isinstance(result, str):
|
||
write_push_log("短信", False, title, [u_key])
|
||
return result
|
||
|
||
if result['status']:
|
||
write_push_log("短信", True, title, [u_key])
|
||
return None
|
||
else:
|
||
write_push_log("短信", False, title, [u_key])
|
||
return result.get("msg", "发送错误")
|
||
|
||
@staticmethod
|
||
def canonical_data(args):
|
||
"""规范数据内容"""
|
||
if not isinstance(args, dict):
|
||
return args
|
||
new_args = {}
|
||
for param, value in args.items():
|
||
if type(value) != str:
|
||
new_str = str(value)
|
||
else:
|
||
new_str = value.replace(".", "_").replace("+", "＋")
|
||
new_args[param] = new_str
|
||
return new_args
|
||
|
||
def push_data(self, data):
|
||
return self.send_msg(data['sm_type'], data['sm_args'])
|
||
|
||
# 发送请求
|
||
def _request(self, d_name: str) -> Union[dict, str]:
|
||
pdata = {
|
||
'access_key': self._PDATA['access_key'],
|
||
'data': json.dumps(self._PDATA['data'])
|
||
}
|
||
try:
|
||
import public
|
||
api_root = public.GetConfigValue('home').rstrip('/') + '/api/wmsg'
|
||
result = public_http_post(api_root + '/' + d_name, pdata)
|
||
result = json.loads(result)
|
||
return result
|
||
except Exception:
|
||
return traceback.format_exc()
|