mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-17 21:25:47 +02:00
Jack
a24f2e23c6
1. Added Mail Marketing Automation trigger tasks 2. Added Mail Marketing Groups Import, Export, Merge 3. Added Mail Marketing Subscribers to support paste import 4. Added Mail Marketing Template Import Export Duplicate 5. Added Mail server add timed automatic reply 6. Added HTTPS Protection function for Website (disables automatic HTTP to HTTPS redirection when enabled) 7. Optimize Website Interface Button Integration 8. Optimize the response speed of WP Toolkit interface
228 lines
7.5 KiB
Python
228 lines
7.5 KiB
Python
# coding: utf-8
|
||
# +-------------------------------------------------------------------
|
||
# | aapanel
|
||
# +-------------------------------------------------------------------
|
||
# | Copyright (c) 2015-2020 aapanel(http://www.aapanel.com) All rights reserved.
|
||
# +-------------------------------------------------------------------
|
||
# | Author: baozi <baozi@bt.cn>
|
||
# | 消息通道HOOK模块
|
||
# +-------------------------------------------------------------------
|
||
|
||
|
||
import requests
|
||
from typing import Optional, Union
|
||
from urllib3.util import parse_url
|
||
|
||
from .util import write_push_log, get_test_msg
|
||
import json
|
||
import public
|
||
# config = {
|
||
# "name": "default",
|
||
# "url": "https://www.bt.cn",
|
||
# "query": {
|
||
# "aaa": "111"
|
||
# },
|
||
# "header": {
|
||
# "AAA": "BBBB",
|
||
# },
|
||
# "body_type": ["json", "form_data", "null"],
|
||
# "custom_parameter": {
|
||
# "rrr": "qqqq"
|
||
# },
|
||
# "method": ["GET", "POST", "PUT", "PATCH"],
|
||
# "ssl_verify": [True, False]
|
||
# }
|
||
# #
|
||
# # 1.自动解析Query参数，拼接并展示给用户 # 可不做
|
||
# # 2.自定义Header头 # 必做
|
||
# # 3.Body中的内容是: type:str="首页磁盘告警", time:int=168955427, data:str="xxxxxx" # ？
|
||
# # 4.自定义参数: key=value 添加在Body中 # 可不做
|
||
# # 5.请求类型自定义 # 必做
|
||
# # 以上内容需要让用户可测试--!
|
||
|
||
|
||
class WebHookMsg(object):
|
||
DEFAULT_HEADERS = {
|
||
"User-Agent": "BT-Panel",
|
||
}
|
||
|
||
def __init__(self, hook_data: dict):
|
||
self.id = hook_data["id"]
|
||
self.config = hook_data["data"]
|
||
|
||
def _replace_and_parse(self, value, real_data):
|
||
"""替换占位符并递归解析JSON字符串"""
|
||
if isinstance(value, str):
|
||
value = value.replace("$1", json.dumps(real_data, ensure_ascii=False))
|
||
elif isinstance(value, dict):
|
||
for k, v in value.items():
|
||
value[k] = self._replace_and_parse(v, real_data)
|
||
return value
|
||
|
||
def send_msg(self, msg: str, title:str, push_type:str) -> Optional[str]:
|
||
the_url = parse_url(self.config['url'])
|
||
|
||
ssl_verify = self.config.get("ssl_verify", None)
|
||
if ssl_verify is None:
|
||
ssl_verify = the_url.scheme == "https"
|
||
else:
|
||
ssl_verify = bool(int(ssl_verify)) # 转换为布尔值
|
||
|
||
|
||
real_data = {
|
||
"title": title,
|
||
"msg": msg,
|
||
"type": push_type,
|
||
}
|
||
custom_parameter = self.config.get("custom_parameter", {})
|
||
if not isinstance(custom_parameter, dict):
|
||
custom_parameter = {} # 如果 custom_parameter 不是字典，则设置为空字典
|
||
# 处理custom_parameter，将$1替换为real_data内容并递归解析
|
||
custom_data = {}
|
||
for k, v in custom_parameter.items():
|
||
custom_data[k] = self._replace_and_parse(v, real_data)
|
||
|
||
if custom_data:
|
||
real_data = custom_data
|
||
|
||
|
||
data = None
|
||
json_data = None
|
||
headers = self.DEFAULT_HEADERS.copy()
|
||
if self.config["body_type"] == "json":
|
||
json_data = real_data
|
||
elif self.config["body_type"] == "form_data":
|
||
data = real_data
|
||
|
||
for k, v in self.config.get("headers", {}).items():
|
||
if not isinstance(v, str):
|
||
v = str(v)
|
||
headers[k] = v
|
||
|
||
status = False
|
||
error = None
|
||
timeout = 10
|
||
if data:
|
||
for k, v in data.items():
|
||
if isinstance(v, str):
|
||
continue
|
||
else:
|
||
data[k]=json.dumps(v)
|
||
|
||
for i in range(3):
|
||
try:
|
||
if json_data is not None:
|
||
res = requests.request(
|
||
method=self.config["method"],
|
||
url=str(the_url),
|
||
json=json_data,
|
||
headers=headers,
|
||
timeout=timeout,
|
||
verify=ssl_verify,
|
||
)
|
||
else:
|
||
res = requests.request(
|
||
method=self.config["method"],
|
||
url=str(the_url),
|
||
data=data,
|
||
headers=headers,
|
||
timeout=timeout,
|
||
verify=ssl_verify,
|
||
)
|
||
|
||
if res.status_code == 200:
|
||
status = True
|
||
break
|
||
else:
|
||
status = False
|
||
return res.text
|
||
except (requests.exceptions.Timeout, requests.exceptions.ConnectionError):
|
||
timeout += 5
|
||
continue
|
||
except requests.exceptions.RequestException as e:
|
||
error = str(e)
|
||
break
|
||
|
||
write_push_log("Web Hook", status, title)
|
||
return error
|
||
|
||
@classmethod
|
||
def check_args(cls, args) -> Union[str, dict]:
|
||
"""配置hook"""
|
||
try:
|
||
title = args['title']
|
||
url = args["url"]
|
||
query = args.get("query", {})
|
||
headers = args.get("headers", {})
|
||
body_type = args.get("body_type", "json")
|
||
custom_parameter = args.get("custom_parameter", {})
|
||
method = args.get("method", "POST")
|
||
ssl_verify = args.get("ssl_verify", None) # null Ture
|
||
except (ValueError, KeyError):
|
||
return public.lang('The parameter is incorrect')
|
||
|
||
the_url = parse_url(url)
|
||
if the_url.scheme is None or the_url.host is None:
|
||
return"URL parsing error, which may not be a legitimate URL"
|
||
|
||
for i in (query, headers, custom_parameter):
|
||
if not isinstance(i, dict):
|
||
return public.lang('Parameter format error')
|
||
|
||
if body_type not in ('json', 'form_data', 'null'):
|
||
return public.lang('The body type must be json,form data, or null')
|
||
|
||
if method not in ('GET', 'POST', 'PUT', 'PATCH'):
|
||
return public.lang('The sending method is incorrect')
|
||
|
||
if ssl_verify not in (True, False, None):
|
||
return public.lang('Verify if the SSL option is wrong')
|
||
|
||
title = title.strip()
|
||
if title == "":
|
||
return"The name cannot be empty"
|
||
|
||
data = {
|
||
"title": title,
|
||
"url": url,
|
||
"query": query,
|
||
"headers": headers,
|
||
"body_type": body_type,
|
||
"custom_parameter": custom_parameter,
|
||
"method": method,
|
||
"ssl_verify": ssl_verify,
|
||
"status": True
|
||
}
|
||
|
||
test_obj = cls({"data": data, "id": None})
|
||
test_msg = {
|
||
"msg_list": ['>configuration state: Success\n\n']
|
||
}
|
||
|
||
test_task = get_test_msg("Message channel configuration reminders")
|
||
|
||
res = test_obj.send_msg(
|
||
test_task.to_web_hook_msg(test_msg, test_task.the_push_public_data()),
|
||
"Message channel configuration reminders",
|
||
"Message channel configuration reminders"
|
||
)
|
||
if res is None:
|
||
return data
|
||
|
||
return res
|
||
|
||
def test_send_msg(self) -> Optional[str]:
|
||
test_msg = {
|
||
"msg_list": ['>configuration state: <font color=#20a53a> Success </font>\n\n']
|
||
}
|
||
test_task = get_test_msg("Message channel configuration reminders")
|
||
res = self.send_msg(
|
||
test_task.to_web_hook_msg(test_msg, test_task.the_push_public_data()),
|
||
"Message channel configuration reminders",
|
||
"Message channel configuration reminders"
|
||
)
|
||
if res is None:
|
||
return None
|
||
return res
|
||
|