70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
|
# 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)
|
||
|
|
|