290 lines
10 KiB
Python
290 lines
10 KiB
Python
#coding: utf-8
|
||
# +-------------------------------------------------------------------
|
||
# | YakPanel
|
||
# +-------------------------------------------------------------------
|
||
# | Copyright (c) 2015-2020 YakPanel(www.yakpanel.com) All rights reserved.
|
||
# +-------------------------------------------------------------------
|
||
# | Author: 沐落 <cjx@yakpanel.com>
|
||
# | Author: lx
|
||
# | 消息通道邮箱模块
|
||
# +-------------------------------------------------------------------
|
||
|
||
import os, sys
|
||
import time,base64
|
||
|
||
panelPath = "/www/server/panel"
|
||
os.chdir(panelPath)
|
||
sys.path.insert(0,panelPath + "/class/")
|
||
import public, json, requests
|
||
from requests.packages import urllib3
|
||
# 关闭警告
|
||
urllib3.disable_warnings()
|
||
import socket
|
||
import requests.packages.urllib3.util.connection as urllib3_cn
|
||
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_default_channel(self):
|
||
"""
|
||
@获取默认消息通道
|
||
"""
|
||
try:
|
||
if public.readFile(self.__default_pl) == self.__module_name:
|
||
return True
|
||
except:pass
|
||
return False
|
||
|
||
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, public.lang("The setup was successful"))
|
||
|
||
def get_web_info(self,get):
|
||
if public.is_self_hosted():
|
||
return public.returnMsg(False, public.lang("WeChat official account cloud features are not available in self-hosted mode."))
|
||
if self.user_info is None: return public.returnMsg(False, public.lang("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": 'B' * 32,
|
||
"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":"The link to the cloud failed, please check the network,请检查网络"}))
|
||
return public.returnMsg(False, public.lang("The link to the cloud failed, please check the network"))
|
||
|
||
def get_web_info2(self):
|
||
if public.is_self_hosted():
|
||
return public.returnMsg(False, public.lang("WeChat official account cloud features are not available in self-hosted mode."))
|
||
if self.user_info is None: return public.returnMsg(False, public.lang("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": 'B' * 32,
|
||
"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":"The link to the cloud failed, please check the network"}))
|
||
return public.returnMsg(False, public.lang("The link to the cloud failed, please check the network"))
|
||
|
||
def get_auth_url(self,get):
|
||
if public.is_self_hosted():
|
||
return public.returnMsg(False, public.lang("WeChat official account cloud features are not available in self-hosted mode."))
|
||
if self.user_info is None: return public.returnMsg(False, public.lang("The user binding information was not obtained"))
|
||
url = "https://wafapi2.yakpanel.com/api/v2/user/wx_web/get_auth_url"
|
||
data = {
|
||
"uid": self.user_info["uid"],
|
||
"access_key": 'B' * 32,
|
||
"serverid":self.user_info["server_id"]
|
||
}
|
||
try:
|
||
datas = json.loads(public.httpPost(url,data))
|
||
if datas["success"]:
|
||
return public.returnMsg(True, datas)
|
||
else:
|
||
return public.returnMsg(False, datas)
|
||
except:
|
||
return public.returnMsg(False, public.lang("The link to the cloud failed, please check the network"))
|
||
|
||
|
||
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 = '\n'.join(s_list)
|
||
|
||
|
||
s_list = []
|
||
for msg_info in msg.split('\n'):
|
||
reg = '<font.+>(.+)</font>'
|
||
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 = '\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, public.lang("No user information was obtained"))
|
||
if public.is_self_hosted():
|
||
return public.returnMsg(False, public.lang("WeChat official account cloud features are not available in self-hosted mode."))
|
||
|
||
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))
|
||
# public.print_log(json.dumps(x))
|
||
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,'Send complete.Send success :{}, send failure :{}',success,error)
|
||
result['success'] = success
|
||
result['error'] = error
|
||
return result
|
||
except:
|
||
print(public.get_error_info())
|
||
return public.returnMsg(False, public.lang("WeChat message delivery failed. --> {}", public.get_error_info()))
|
||
|
||
def push_data(self,data):
|
||
return self.send_msg(data['msg'])
|
||
|
||
def uninstall(self):
|
||
if os.path.exists(self.conf_path):
|
||
os.remove(self.conf_path) |