Initial YakPanel commit

mod/base/msg/sms_msg.py

@@ -0,0 +1,123 @@
+# 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()