Gecko [ www.jdcpatala.org ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	5.49 KB	-rw-r--r--
dingding_msg.py	4.57 KB	-rw-r--r--
feishu_msg.py	4.06 KB	-rw-r--r--
mail_msg.py	5.1 KB	-rw-r--r--
manager.py	14.91 KB	-rw-r--r--
sms_msg.py	4.11 KB	-rw-r--r--
test.json	2.48 KB	-rw-r--r--
tg_msg.py	9.08 KB	-rw-r--r--
util.py	3.63 KB	-rw-r--r--
web_hook_msg.py	7.46 KB	-rw-r--r--
weixin_msg.py	3.79 KB	-rw-r--r--
wx_account_msg.py	19.77 KB	-rw-r--r--

Code Editor : init.py

import json
import os.path

from .weixin_msg import WeiXinMsg
from .mail_msg import MailMsg
from .web_hook_msg import WebHookMsg
from .feishu_msg import FeiShuMsg
from .dingding_msg import DingDingMsg
from .sms_msg import SMSMsg
# from .wx_account_msg import WeChatAccountMsg
from .tg_msg import TgMsg
from .manager import SenderManager
from .util import read_file,write_file

from mod.base.push_mod import SenderConfig, PUSH_DATA_PATH

# 把旧地告警系统的信息通道更新
def update_mod_push_msg():

if os.path.exists(PUSH_DATA_PATH + "/update_sender.pl"):
        return
    # else:
    #     with open(PUSH_DATA_PATH + "/update_sender.pl", "w") as f:
    #             f.write("")

# WeChatAccountMsg.refresh_config(force=True)
    sms_status = False
    sc = SenderConfig()
    for conf in sc.config:
        if conf["sender_type"] == "sms":
            sms_status = True
            break

# sms 取消自动添加
    # if not sms_status:
    #     sc.config.append({
    #         "id": sc.nwe_id(),
    #         "used": True,
    #         "sender_type": "sms",
    #         "data": {},
    #         "original": True   # 标记这个通道是该类型 旧有的通道, 同时也是默认通道
    #     })

panel_data_path = "/www/server/panel/data"

# weixin
    if os.path.exists(panel_data_path + "/weixin.json"):
        try:
            weixin_data = json.loads(read_file(panel_data_path + "/weixin.json"))
        except:
            weixin_data = None

if isinstance(weixin_data, dict) and "weixin_url" in weixin_data:
            sc.config.append({
                "id": sc.nwe_id(),
                "used": True,
                "sender_type": "weixin",
                "data": {
                    "url": weixin_data["weixin_url"],
                    "title": "weixin" if "title" not in weixin_data else weixin_data["title"]
                },
                "original": True
            })

# mail
    stmp_file = panel_data_path + "/stmp_mail.json"
    mail_list_file = panel_data_path + "/mail_list.json"
    if os.path.exists(stmp_file) and os.path.exists(mail_list_file):
        stmp_data = None
        try:
            stmp_data = json.loads(read_file(stmp_file))
            mail_list_data = json.loads(read_file(mail_list_file))
        except:
            mail_list_data = None

if isinstance(stmp_data, dict):
            if 'qq_mail' in stmp_data or 'qq_stmp_pwd' in stmp_data or 'hosts' in stmp_data:
                sc.config.append({
                    "id": sc.nwe_id(),
                    "used": True,
                    "sender_type": "mail",
                    "data": {
                        "send": stmp_data,
                        "title": "mail",
                        "receive": [] if not mail_list_data else mail_list_data,
                    },
                    "original": True
                })

# webhook
    webhook_file = panel_data_path + "/hooks_msg.json"
    if os.path.exists(stmp_file) and os.path.exists(mail_list_file):
        try:
            webhook_data = json.loads(read_file(webhook_file))
        except:
            webhook_data = None

if isinstance(webhook_data, list):
            for i in webhook_data:
                i["title"] = i["name"]
                sc.config.append({
                    "id": sc.nwe_id(),
                    "used": True,
                    "sender_type": "webhook",
                    "data": i,
                })

# feishu
    if os.path.exists(panel_data_path + "/feishu.json"):
        try:
            feishu_data = json.loads(read_file(panel_data_path + "/feishu.json"))
        except:
            feishu_data = None

if isinstance(feishu_data, dict) and "feishu_url" in feishu_data:
            sc.config.append({
                "id": sc.nwe_id(),
                "used": True,
                "sender_type": "feishu",
                "data": {
                    "url": feishu_data["feishu_url"],
                    "title": "feishu" if "title" not in feishu_data else feishu_data["title"]
                },
                "original": True
            })

# dingding
    if os.path.exists(panel_data_path + "/dingding.json"):
        try:
            dingding_data = json.loads(read_file(panel_data_path + "/dingding.json"))
        except:
            dingding_data = None

if isinstance(dingding_data, dict) and "dingding_url" in dingding_data:
            sc.config.append({
                "id": sc.nwe_id(),
                "used": True,
                "sender_type": "dingding",
                "data": {
                    "url": dingding_data["dingding_url"],
                    "title": "dingding" if "title" not in dingding_data else dingding_data["title"]
                },
                "original": True
            })

# tg
    if os.path.exists(panel_data_path + "/tg_bot.json"):
        try:
            tg_data = json.loads(read_file(panel_data_path + "/tg_bot.json"))
        except:
            tg_data = None

if isinstance(tg_data, dict) and "bot_token" in tg_data:
            sc.config.append({
                "id": sc.nwe_id(),
                "used": True,
                "sender_type": "tg",
                "data": {
                    "my_id": tg_data["my_id"],
                    "bot_token": tg_data["bot_token"],
                    "title": "tg" if "title" not in tg_data else tg_data["title"],
                },
                "original": True
            })

sc.save_config()
    write_file(PUSH_DATA_PATH + "/update_sender.pl", "")

${this.title}

Code Editor : __init__.py

Code Editor : init.py