Initial YakPanel commit

mod/project/push/msgconfMod.py

@@ -0,0 +1,69 @@
+# coding: utf-8
+# -------------------------------------------------------------------
+# yakpanel
+# -------------------------------------------------------------------
+# Copyright (c) 2015-2017 yakpanel(https://www.yakpanel.com) All rights reserved.
+# -------------------------------------------------------------------
+# Author: baozi <baozi@yakpanel.com>
+# -------------------------------------------------------------------
+# 新告警通道管理模块
+# ------------------------------
+from mod.base.msg import SenderManager, update_mod_push_msg
+from mod.base.push_mod import SenderConfig
+from mod.base import json_response
+
+import public
+
+
+update_mod_push_msg()
+
+
+class main(SenderManager):
+
+    @staticmethod
+    def wx_account_auth(get=None):
+        return WeChatAccountMsg.get_auth_url()
+
+    @staticmethod
+    def unbind_wx_account(get):
+        try:
+            sender_id = get.sender_id.strip()
+        except AttributeError:
+            return json_response(status=False, msg="Parameter error")
+
+        conf = SenderConfig().get_by_id(sender_id)
+        if not conf:
+            return json_response(status=False, msg="No binding information was found")
+
+        res = WeChatAccountMsg.unbind(conf["data"]["id"])
+        public.WriteFile(WeChatAccountMsg.need_refresh_file, "")
+        return res
+
+    def set_default_sender(self, get):
+        try:
+
+            try:
+                sender_id = get.sender_id.strip()
+                sender_type = get.sender_type.strip()
+            except AttributeError:
+                return json_response(status=False, msg="Parameter error")
+
+            sc = SenderConfig()
+            change = False
+            print("SenderConfig",sc.config)
+            for conf in sc.config:
+                if conf["sender_type"] == sender_type:
+                    is_original = conf.get("original", False)
+                    if conf["id"] == sender_id:
+                        change = True
+                        conf["original"] = True
+                    else:
+                        conf["original"] = False
+
+            sc.save_config()
+            if change:
+                self.set_default_for_compatible(sc.get_by_id(sender_id))
+            return json_response(status=True, msg="Successfully set")
+        except Exception as e:
+            return json_response(status=False, msg=e)
+