# coding: utf-8 # +------------------------------------------------------------------- # | yakpanel # +------------------------------------------------------------------- # | Copyright (c) 2015-2020 yakpanel(http://www.yakpanel.com) All rights reserved. # +------------------------------------------------------------------- # | Author: baozi # | 消息通道微信公众号模块 # +------------------------------------------------------------------- import os, sys import time, base64 import re import json import requests import traceback import socket import public import requests.packages.urllib3.util.connection as urllib3_cn from requests.packages import urllib3 from typing import Optional, Union, List, Dict, Any from .util import write_push_log, get_test_msg, read_file, public_http_post from mod.base.push_mod import WxAccountMsg, SenderConfig from mod.base import json_response # 关闭警告 urllib3.disable_warnings() class WeChatAccountMsg: USER_PATH = '/www/server/panel/data/userInfo.json' need_refresh_file = '/www/server/panel/data/mod_push_data/refresh_wechat_account.tip' refresh_time = '/www/server/panel/data/mod_push_data/refresh_wechat_account_time.pl' def __init__(self, *config_data): if len(config_data) == 0: self.config = None elif len(config_data) == 1: self.config = config_data[0]["data"] else: self.config = config_data[0]["data"] self.config["users"] = [i["data"]['id'] for i in config_data] self.config["users_nickname"] = [i["data"]['nickname'] for i in config_data] try: self.user_info = json.loads(read_file(self.USER_PATH)) except: self.user_info = None @classmethod def get_user_info(cls) -> Optional[dict]: try: return json.loads(read_file(cls.USER_PATH)) except: return None @classmethod def last_refresh(cls): tmp = read_file(cls.refresh_time) if not tmp: last_refresh_time = 0 else: try: last_refresh_time = int(tmp) except: last_refresh_time = 0 return last_refresh_time @staticmethod def get_local_ip() -> str: """获取内网IP""" import socket s = None try: s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) s.connect(('8.8.8.8', 80)) ip = s.getsockname()[0] return ip except: pass finally: if s is not None: s.close() return '127.0.0.1' def send_msg(self, msg: WxAccountMsg) -> Optional[str]: if self.user_info is None: return public.lang('No user information was obtained') if public.is_self_hosted(): return public.lang('WeChat official account cloud features are not available in self-hosted mode.') msg.set_ip_address(self.user_info["address"], self.get_local_ip()) template_id, msg_data = msg.to_send_data() url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/send_template_msg_v3" wx_account_ids = self.config["users"] if "users" in self.config else [self.config["id"], ] data = { "uid": self.user_info["uid"], "access_key": 'B' * 32, "data": base64.b64encode(json.dumps(msg_data).encode('utf-8')).decode('utf-8'), "wx_account_ids": base64.b64encode(json.dumps(wx_account_ids).encode('utf-8')).decode('utf-8'), } if template_id != "": data["template_id"] = template_id status = False error = None user_name = self.config["users_nickname"] if "users_nickname" in self.config else [self.config["nickname"], ] try: resp = public_http_post(url, data) x = json.loads(resp) if x["success"]: status = True else: status = False error = x["res"] except: error = traceback.format_exc() write_push_log("wx_account", status, msg.thing_type, user_name) return error @classmethod def refresh_config(cls, force: bool = False): if os.path.exists(cls.need_refresh_file): force = True os.remove(cls.need_refresh_file) if force or cls.last_refresh() + 60 * 10 < time.time(): cls._get_by_web() @classmethod def _get_by_web(cls) -> Optional[List]: user_info = cls.get_user_info() if user_info is None: return None if public.is_self_hosted(): return None url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/bound_wx_accounts" data = { "uid": user_info["uid"], "access_key": 'B' * 32, "serverid": user_info["server_id"] } try: data = json.loads(public_http_post(url, data)) if not data["success"]: return None except: return None cls._save_user_info(data["res"]) return data["res"] @staticmethod def _save_user_info(user_config_list: List[Dict[str, Any]]): print(user_config_list) user_config_dict = {i["hex"]: i for i in user_config_list} remove_list = [] sc = SenderConfig() for i in sc.config: if i['sender_type'] != "wx_account": continue if i['data'].get("hex", None) in user_config_dict: i['data'].update(user_config_dict[i['data']["hex"]]) user_config_dict.pop(i['data']["hex"]) else: remove_list.append(i) for r in remove_list: sc.config.remove(r) if user_config_dict: # 还有多的 for v in user_config_dict.values(): v["title"] = v["nickname"] sc.config.append({ "id": sc.nwe_id(), "used": True, "sender_type": "wx_account", "data": v }) sc.save_config() @classmethod def unbind(cls, wx_account_uid: str): user_info = cls.get_user_info() if user_info is None: return json_response(status=True, msg=public.lang('The user binding information was not obtained')) url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/unbind_wx_accounts" data = { "uid": user_info["uid"], "access_key": 'B' * 32, "serverid": user_info["server_id"], "ids": str(wx_account_uid) } try: datas = json.loads(public_http_post(url, data)) if datas["success"]: return json_response(status=True, data=datas, msg=public.lang('The unbinding is successful')) else: return json_response(status=False, data=datas, msg=datas["res"]) except: return json_response(status=True, msg=public.lang('Failed to link to the cloud')) @classmethod def get_auth_url(cls): user_info = cls.get_user_info() if user_info is None: return json_response(status=True, msg=public.lang('The user binding information was not obtained')) if public.is_self_hosted(): return json_response(status=False, msg=public.lang('WeChat official account cloud features are not available in self-hosted mode.')) url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/get_auth_url" data = { "uid": user_info["uid"], "access_key": 'B' * 32, "serverid": user_info["server_id"], } try: datas = json.loads(public_http_post(url, data)) if datas["success"]: return json_response(status=True, data=datas) else: return json_response(status=False, data=datas, msg=datas["res"]) except: return json_response(status=True, msg=public.lang('Failed to link to the cloud')) def test_send_msg(self) -> Optional[str]: test_msg = { "msg_list": ['>configuration state: Success \n\n'] } test_task = get_test_msg("Message channel configuration reminders") res = self.send_msg( test_task.to_wx_account_msg(test_msg, test_task.the_push_public_data()), ) if res is None: return None return res # class wx_account_msg: # __module_name = None # __default_pl = "{}/data/default_msg_channel.pl".format(panelPath) # conf_path = '{}/data/wx_account_msg.json'.format(panelPath) # user_info = None # # def __init__(self): # try: # self.user_info = json.loads(public.ReadFile("{}/data/userInfo.json".format(public.get_panel_path()))) # except: # self.user_info = None # self.__module_name = self.__class__.__name__.replace('_msg', '') # # def get_version_info(self, get): # """ # 获取版本信息 # """ # data = {} # data['ps'] = 'YakPanel 微信公众号，用于接收面板消息推送' # data['version'] = '1.0' # data['date'] = '2022-08-15' # data['author'] = 'YakPanel' # data['title'] = '微信公众号' # data['help'] = 'http://www.yakpanel.com' # return data # # def get_local_ip(self): # '''获取内网IP''' # import socket # try: # s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # s.connect(('8.8.8.8', 80)) # ip = s.getsockname()[0] # return ip # finally: # s.close() # return '127.0.0.1' # # def get_config(self, get): # """ # 微信公众号配置 # """ # if os.path.exists(self.conf_path): # # 60S内不重复加载 # start_time = int(time.time()) # if os.path.exists("data/wx_account_msg.lock"): # lock_time = 0 # try: # lock_time = int(public.ReadFile("data/wx_account_msg.lock")) # except: # pass # # 大于60S重新加载 # if start_time - lock_time > 60: # public.run_thread(self.get_web_info2) # public.WriteFile("data/wx_account_msg.lock", str(start_time)) # else: # public.WriteFile("data/wx_account_msg.lock", str(start_time)) # public.run_thread(self.get_web_info2) # data = json.loads(public.ReadFile(self.conf_path)) # # if not 'list' in data: data['list'] = {} # # title = '默认' # if 'res' in data and 'nickname' in data['res']: title = data['res']['nickname'] # # data['list']['default'] = {'title': title, 'data': ''} # # data['default'] = self.__get_default_channel() # return data # else: # public.run_thread(self.get_web_info2) # return {"success": False, "res": "未获取到配置信息"} # # def set_config(self, get): # """ # @设置默认值 # """ # if 'default' in get and get['default']: # public.writeFile(self.__default_pl, self.__module_name) # # return public.returnMsg(True, '设置成功') # # def get_web_info(self, get): # if self.user_info is None: return public.returnMsg(False, 'The user binding information was not obtained') # url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/info" # data = { # "uid": self.user_info["uid"], # "access_key": self.user_info["access_key"], # "serverid": self.user_info["server_id"] # } # try: # # datas = json.loads(public.httpPost(url, data)) # # if datas["success"]: # public.WriteFile(self.conf_path, json.dumps(datas)) # return public.returnMsg(True, datas) # else: # public.WriteFile(self.conf_path, json.dumps(datas)) # return public.returnMsg(False, datas) # except: # public.WriteFile(self.conf_path, json.dumps({"success": False, "res": "链接云端失败,请检查网络"})) # return public.returnMsg(False, "链接云端失败,请检查网络") # # def unbind(self): # if self.user_info is None: # return public.returnMsg(False, 'The user binding information was not obtained') # url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/unbind" # data = { # "uid": self.user_info["uid"], # "access_key": self.user_info["access_key"], # "serverid": self.user_info["server_id"] # } # try: # # datas = json.loads(public.httpPost(url, data)) # # if os.path.exists(self.conf_path): # os.remove(self.conf_path) # # if datas["success"]: # return public.returnMsg(True, datas) # else: # return public.returnMsg(False, datas) # except: # public.WriteFile(self.conf_path, json.dumps({"success": False, "res": "链接云端失败,请检查网络"})) # return public.returnMsg(False, "链接云端失败,请检查网络") # # def get_web_info2(self): # if self.user_info is None: # return public.returnMsg(False, 'The user binding information was not obtained') # url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/info" # data = { # "uid": self.user_info["uid"], # "access_key": self.user_info["access_key"], # "serverid": self.user_info["server_id"] # } # try: # datas = json.loads(public.httpPost(url, data)) # if datas["success"]: # public.WriteFile(self.conf_path, json.dumps(datas)) # return public.returnMsg(True, datas) # else: # public.WriteFile(self.conf_path, json.dumps(datas)) # return public.returnMsg(False, datas) # except: # public.WriteFile(self.conf_path, json.dumps({"success": False, "res": "链接云端失败"})) # return public.returnMsg(False, "链接云端失败") # # def get_send_msg(self, msg): # """ # @name 处理md格式 # """ # try: # import re # title = 'YakPanel 告警通知' # if msg.find("####") >= 0: # try: # title = re.search(r"####(.+)", msg).groups()[0] # except: # pass # # msg = msg.replace("####", ">").replace("\n\n", "\n").strip() # s_list = msg.split('\n') # # if len(s_list) > 3: # s_title = s_list[0].replace(" ", "") # s_list = s_list[3:] # s_list.insert(0, s_title) # msg=public.lang('\n').join(s_list) # # s_list = [] # for msg_info in msg.split('\n'): # reg = '(.+)' # tmp = re.search(reg, msg_info) # if tmp: # tmp = tmp.groups()[0] # msg_info = re.sub(reg, tmp, msg_info) # s_list.append(msg_info) # msg=public.lang('\n').join(s_list) # except: # pass # return msg, title # # def send_msg(self, msg): # """ # 微信发送信息 # @msg 消息正文 # """ # # if self.user_info is None: # return public.returnMsg(False, '未获取到用户信息') # # if not isinstance(msg, str): # return self.send_msg_v2(msg) # # msg, title = self.get_send_msg(msg) # url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/send_template_msg_v2" # datassss = { # "first": { # "value": "堡塔主机告警", # }, # "keyword1": { # "value": "内网IP " + self.get_local_ip() + "\n外网IP " + self.user_info[ # "address"] + " \n服务器别名 " + public.GetConfigValue("title"), # }, # "keyword2": { # "value": "堡塔主机告警", # }, # "keyword3": { # "value": msg, # }, # "remark": { # "value": "如有疑问，请联系YakPanel 支持", # }, # } # data = { # "uid": self.user_info["uid"], # "access_key": self.user_info["access_key"], # "data": base64.b64encode(json.dumps(datassss).encode('utf-8')).decode('utf-8') # } # # try: # res = {} # error, success = 0, 0 # # x = json.loads(public.httpPost(url, data)) # conf = self.get_config(None)['list'] # # # 立即刷新剩余次数 # public.run_thread(self.get_web_info2) # # res[conf['default']['title']] = 0 # if x['success']: # res[conf['default']['title']] = 1 # success += 1 # else: # error += 1 # # try: # public.write_push_log(self.__module_name, title, res) # except: # pass # # result = public.returnMsg(True, '发送完成,发送成功{},发送失败{}.'.format(success, error)) # result['success'] = success # result['error'] = error # return result # # except: # print(public.get_error_info()) # return public.returnMsg(False, '微信消息发送失败。 --> {}'.format(public.get_error_info())) # # def push_data(self, data): # if isinstance(data, dict): # return self.send_msg(data['msg']) # else: # return self.send_msg_v2(data) # # def uninstall(self): # if os.path.exists(self.conf_path): # os.remove(self.conf_path) # # def send_msg_v2(self, msg): # from push.base_push import WxAccountMsgBase, WxAccountMsg # if self.user_info is None: # return public.returnMsg(False, '未获取到用户信息') # # if isinstance(msg, public.dict_obj): # msg = getattr(msg, "msg", "测试信息") # if len(msg) >= 20: # return self.send_msg(msg) # # if isinstance(msg, str): # the_msg = WxAccountMsg.new_msg() # the_msg.thing_type = msg # the_msg.msg = msg # msg = the_msg # # if not isinstance(msg, WxAccountMsgBase): # return public.returnMsg(False, '消息类型错误') # # msg.set_ip_address(self.user_info["address"], self.get_local_ip()) # # template_id, msg_data = msg.to_send_data() # url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/send_template_msg_v2" # data = { # "uid": self.user_info["uid"], # "access_key": self.user_info["access_key"], # "data": base64.b64encode(json.dumps(msg_data).encode('utf-8')).decode('utf-8'), # } # if template_id != "": # data["template_id"] = template_id # # try: # error, success = 0, 0 # resp = public.httpPost(url, data) # x = json.loads(resp) # conf = self.get_config(None)['list'] # # # 立即刷新剩余次数 # public.run_thread(self.get_web_info2) # # res = { # conf['default']['title']: 0 # } # if x['success']: # res[conf['default']['title']] = 1 # success += 1 # else: # error += 1 # # try: # public.write_push_log(self.__module_name, msg.thing_type, res) # except: # pass # result = public.returnMsg(True, '发送完成,发送成功{},发送失败{}.'.format(success, error)) # result['success'] = success # result['error'] = error # return result # # except: # return public.returnMsg(False, '微信消息发送失败。 --> {}'.format(public.get_error_info()))