import time import traceback from mod.base.push_mod import SenderConfig from .weixin_msg import WeiXinMsg from .mail_msg import MailMsg from .tg_msg import TgMsg 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 import json from mod.base import json_response from .util import write_file, read_file import sys,os sys.path.insert(0, "/www/server/panel/class/") import public # 短信会自动添加到 sender 库中的第一个 且通过官方接口更新 # 微信公众号信息通过官网接口更新， 不写入数据库，需要时由文件中读取并序列化 # 其他告警通道本质都类似于web hook 在确认完数据信息无误后，都可以自行添加或启用 class SenderManager: def __init__(self): self.custom_parameter_filename = "/www/server/panel/data/mod_push_data/custom_parameter.pl" self.init_default_sender() def set_sender_conf(self, get): args = json.loads(get.sender_data.strip()) try: sender_id = None try: if hasattr(get, "sender_id"): sender_id = get.sender_id.strip() if not sender_id: sender_id = None sender_type = get.sender_type.strip() args = json.loads(get.sender_data.strip()) except (json.JSONDecoder, AttributeError, TypeError): return json_response(status=False, msg=public.lang('The parameter is incorrect')) sender_config = SenderConfig() if sender_id is not None: tmp = sender_config.get_by_id(sender_id) if tmp is None: sender_id = None if sender_type == "weixin": data = WeiXinMsg.check_args(args) if isinstance(data, str): return json_response(status=False, data=data, msg=public.lang('Test send failed')) elif sender_type == "mail": _, data = MailMsg.check_args(args) if isinstance(data, str): return json_response(status=False, data=data, msg=public.lang('Test send failed')) elif sender_type == "tg": _, data = TgMsg.check_args(args) if isinstance(data, str): return json_response(status=False, data=data, msg=data) elif sender_type == "webhook": custom_parameter = args.get("custom_parameter", {}) if custom_parameter: try: public.writeFile(self.custom_parameter_filename, json.dumps(custom_parameter)) except: pass data = WebHookMsg.check_args(args) if isinstance(data, str): return json_response(status=False, data=data, msg=public.lang('Test send failed')) # 从文件读取并删除文件 try: if os.path.exists(self.custom_parameter_filename): custom_parameter = json.loads(public.readFile(self.custom_parameter_filename)) data['custom_parameter'] = custom_parameter os.remove(self.custom_parameter_filename) except: pass elif sender_type == "feishu": data = FeiShuMsg.check_args(args) if isinstance(data, str): return json_response(status=False, data=data, msg=public.lang('Test send failed')) elif sender_type == "dingding": data = DingDingMsg.check_args(args) if isinstance(data, str): return json_response(status=False, data=data, msg=public.lang('Test send failed')) else: return json_response(status=False, msg=public.lang('A type that is not supported by the current interface')) # Check if the sender configuration already exists existing_sender = any( conf for conf in sender_config.config if conf['sender_type'] == sender_type and 'title' in conf['data'] and conf['data']['title'] == data['title'] and conf['id'] != sender_id ) # for conf in sender_config.config: # if conf['sender_type'] == sender_type and 'title' in conf['data'] and conf['data']['title'] == data[ # 'title'] and conf['id'] != sender_id: # public.print_log('000 -{}'.format(conf['sender_type'])) # public.print_log('000 -{}'.format(sender_type)) # # public.print_log('111 conf -{}'.format(conf['sender_type'])) # public.print_log('111 -{}'.format(sender_type)) # # public.print_log('222 conf -{}'.format(conf['data']['title'])) # public.print_log('222 data -{}'.format(data['title'])) # # public.print_log('333 conf -{}'.format(conf['id'])) # public.print_log('333 -{}'.format(sender_id)) if existing_sender: return json_response(status=False, msg=public.lang('The same send configuration already exists and cannot be added repeatedly')) now_sender_id = None if not sender_id: now_sender_id = sender_config.nwe_id() sender_config.config.append( { "id": now_sender_id, "sender_type": sender_type, "data": data, "used": True, }) else: now_sender_id = sender_id tmp = sender_config.get_by_id(sender_id) tmp["data"].update(data) # type_senders = [conf for conf in sender_config.config if conf['sender_type'] == sender_type] # if len(type_senders) == 1: # for conf in sender_config.config: # conf["original"] = (conf['id'] == now_sender_id) sender_config.save_config() if sender_type == "webhook": self.set_default_for_compatible(sender_config.get_by_id(now_sender_id)) return json_response(status=True, msg=public.lang('Saved successfully')) except: public.print_log('Error:{}'.format(str(public.get_error_info()))) @staticmethod def change_sendr_used(get): try: sender_id = get.sender_id.strip() except (AttributeError, TypeError): return json_response(status=False, msg=public.lang('The parameter is incorrect')) sender_config = SenderConfig() tmp = sender_config.get_by_id(sender_id) if tmp is None: return json_response(status=False, msg=public.lang('Corresponding sender not found')) tmp["used"] = not tmp["used"] sender_config.save_config() return json_response(status=True, msg=public.lang('Saved successfully')) @staticmethod def remove_sender(get): try: sender_id = get.sender_id.strip() except (AttributeError, TypeError): return json_response(status=False, msg=public.lang('The parameter is incorrect')) sender_config = SenderConfig() tmp = sender_config.get_by_id(sender_id) if tmp is None: return json_response(status=False, msg=public.lang('Corresponding sender not found')) sender_config.config.remove(tmp) sender_config.save_config() return json_response(status=True, msg=public.lang('Successfully delete')) @staticmethod def get_sender_list(get): # 微信， 飞书， 钉钉， web-hook， 邮箱 refresh = False try: if hasattr(get, 'refresh'): refresh = get.refresh.strip() if refresh in ("1", "true"): refresh = True except (AttributeError, TypeError): return json_response(status=False, msg=public.lang('The parameter is incorrect')) res = [] # WeChatAccountMsg.refresh_config(force=refresh) simple = ("weixin", "mail", "webhook", "feishu", "dingding", "tg") for conf in SenderConfig().config: if conf["sender_type"] in simple or conf["sender_type"] == "wx_account": res.append(conf) # 去掉短信设置 # elif conf["sender_type"] == "sms": # conf["data"] = SMSMsg(conf).refresh_config(force=refresh) # res.append(conf) res.sort(key=lambda x: x["sender_type"]) return json_response(status=True, data=res) @staticmethod def test_send_msg(get): try: sender_id = get.sender_id.strip() except (json.JSONDecoder, AttributeError, TypeError): return json_response(status=False, msg=public.lang('The parameter is incorrect')) sender_config = SenderConfig() tmp = sender_config.get_by_id(sender_id) if tmp is None: return json_response(status=False, msg=public.lang('Corresponding sender not found')) sender_type = tmp["sender_type"] if sender_type == "weixin": sender_obj = WeiXinMsg(tmp) elif sender_type == "mail": sender_obj = MailMsg(tmp) elif sender_type == "webhook": sender_obj = WebHookMsg(tmp) elif sender_type == "feishu": sender_obj = FeiShuMsg(tmp) elif sender_type == "dingding": sender_obj = DingDingMsg(tmp) elif sender_type == "tg": sender_obj = TgMsg(tmp) # elif sender_type == "wx_account": # sender_obj = WeChatAccountMsg(tmp) else: return json_response(status=False, msg=public.lang('A type that is not supported by the current interface')) res = sender_obj.test_send_msg() if isinstance(res, str): return json_response(status=False, data=res, msg=public.lang('Test send failed')) return json_response(status=True, msg=public.lang('The sending was successful')) @staticmethod def set_default_for_compatible(sender_data: dict): if sender_data["sender_type"] in ("sms", "wx_account"): return panel_data = "/www/server/panel/data" if sender_data["sender_type"] == "weixin": weixin_file = "{}/weixin.json".format(panel_data) write_file(weixin_file, json.dumps({ "state": 1, "weixin_url": sender_data["data"]["url"], "title": sender_data["data"]["title"], "list": { "default": { "data": sender_data["data"]["url"], "title": sender_data["data"]["title"], "status": 1, "addtime": int(time.time()) } } })) elif sender_data["sender_type"] == "mail": stmp_mail_file = "{}/stmp_mail.json".format(panel_data) mail_list_file = "{}/mail_list.json".format(panel_data) write_file(stmp_mail_file, json.dumps(sender_data["data"]["send"])) write_file(mail_list_file, json.dumps(sender_data["data"]["receive"])) elif sender_data["sender_type"] == "feishu": feishu_file = "{}/feishu.json".format(panel_data) write_file(feishu_file, json.dumps({ "feishu_url": sender_data["data"]["url"], "title": sender_data["data"]["title"], "isAtAll": True, "user": [] })) elif sender_data["sender_type"] == "dingding": dingding_file = "{}/dingding.json".format(panel_data) write_file(dingding_file, json.dumps({ "dingding_url": sender_data["data"]["url"], "title": sender_data["data"]["title"], "isAtAll": True, "user": [] })) elif sender_data["sender_type"] == "tg": tg_file = "{}/tg_bot.json".format(panel_data) write_file(tg_file, json.dumps({ "my_id": sender_data["data"]["my_id"], "bot_token": sender_data["data"]["bot_token"], "title": sender_data["data"]["title"] })) elif sender_data["sender_type"] == "webhook": webhook_file = "{}/hooks_msg.json".format(panel_data) try: webhook_data = json.loads(read_file(webhook_file)) except: webhook_data =[] target_idx = -1 for idx, i in enumerate(webhook_data): if i["name"] == sender_data["data"]["title"]: target_idx = idx break else: sender_data["data"]["name"] = sender_data["data"]["title"] webhook_data.append(sender_data["data"]) if target_idx != -1: sender_data["data"]["name"] = sender_data["data"]["title"] webhook_data[target_idx] = sender_data["data"] write_file(webhook_file, json.dumps(webhook_data)) def init_default_sender(self): import os,sys sys.path.insert(0, "/www/server/panel/mod/project/push") import msgconfMod sender_config = SenderConfig() sender_types = set(conf['sender_type'] for conf in sender_config.config) all_types = {"feishu", "dingding", "weixin", "mail", "webhook"} # 所有可能的类型 for sender_type in sender_types: type_senders = [conf for conf in sender_config.config if conf['sender_type'] == sender_type] # 检查是否已有默认通道 has_default = any(conf.get('original', False) for conf in type_senders) if has_default: continue if len(type_senders) == 1: # 只有一个通道，设置为默认通道 for conf in type_senders: get = public.dict_obj() get['sender_id'] = conf['id'] get['sender_type'] = conf['sender_type'] self.set_default_sender(get) else: # 有多个通道，根据添加时间设置默认通道 sorted_senders = sorted(type_senders, key=lambda x: x['data'].get('create_time', '')) if sorted_senders: get = public.dict_obj() get['sender_id'] = sorted_senders[0]['id'] get['sender_type'] = sorted_senders[0]['sender_type'] self.set_default_sender(get) # 检查没有通道的类型，并删除对应文件 missing_types = all_types - sender_types for missing_type in missing_types: file_path = f"/www/server/panel/data/{missing_type}.json" if os.path.exists(file_path): os.remove(file_path)