297 lines
11 KiB
Python
297 lines
11 KiB
Python
|
import json
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
from typing import Union, Optional
|
|||
|
|
|
|||
|
|
from mod.base import json_response
|
|||
|
|
from .mods import TaskTemplateConfig, TaskConfig, SenderConfig, TaskRecordConfig
|
|||
|
|
from .system import PushSystem
|
|||
|
|
|
|||
|
|
sys.path.insert(0, "/www/server/panel/class/")
|
|||
|
|
import public
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PushManager:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.template_conf = TaskTemplateConfig()
|
|||
|
|
self.task_conf = TaskConfig()
|
|||
|
|
self.send_config = SenderConfig()
|
|||
|
|
self._send_conf_cache = {}
|
|||
|
|
|
|||
|
|
def _get_sender_conf(self, sender_id):
|
|||
|
|
if sender_id in self._send_conf_cache:
|
|||
|
|
return self._send_conf_cache[sender_id]
|
|||
|
|
tmp = self.send_config.get_by_id(sender_id)
|
|||
|
|
self._send_conf_cache[sender_id] = tmp
|
|||
|
|
return tmp
|
|||
|
|
|
|||
|
|
def normalize_task_config(self, task, template) -> Union[dict, str]:
|
|||
|
|
result = {}
|
|||
|
|
sender = task.get("sender", None)
|
|||
|
|
if sender is None:
|
|||
|
|
return "No alarm channel is configured"
|
|||
|
|
if not isinstance(sender, list):
|
|||
|
|
return "The alarm channel is incorrect"
|
|||
|
|
|
|||
|
|
new_sender = []
|
|||
|
|
for i in sender:
|
|||
|
|
sender_conf = self._get_sender_conf(i)
|
|||
|
|
if not sender_conf:
|
|||
|
|
continue
|
|||
|
|
else:
|
|||
|
|
new_sender.append(i)
|
|||
|
|
if sender_conf["sender_type"] not in template["send_type_list"]:
|
|||
|
|
if sender_conf["sender_type"] == "sms":
|
|||
|
|
return "SMS alerts are not supported"
|
|||
|
|
return "Unsupported alerting methods:{}".format(sender_conf['data']["title"])
|
|||
|
|
if not sender_conf["used"]:
|
|||
|
|
if sender_conf["sender_type"] == "sms":
|
|||
|
|
return "The SMS alert channel has been closed"
|
|||
|
|
return "Closed alert mode:{}".format(sender_conf['data']["title"])
|
|||
|
|
|
|||
|
|
result["sender"] = new_sender
|
|||
|
|
|
|||
|
|
if "default" in template and template["default"]:
|
|||
|
|
task_data = task.get("task_data", {})
|
|||
|
|
for k, v in template["default"].items():
|
|||
|
|
if k not in task_data:
|
|||
|
|
task_data[k] = v
|
|||
|
|
|
|||
|
|
result["task_data"] = task_data
|
|||
|
|
# 避免default为空时，无数据
|
|||
|
|
else:
|
|||
|
|
result["task_data"] = task.get("task_data", {})
|
|||
|
|
|
|||
|
|
if "task_data" not in result:
|
|||
|
|
result["task_data"] = {}
|
|||
|
|
|
|||
|
|
time_rule = task.get("time_rule", {})
|
|||
|
|
|
|||
|
|
if "send_interval" in time_rule:
|
|||
|
|
if not isinstance(time_rule["send_interval"], int):
|
|||
|
|
return "The minimum interval is set incorrectly"
|
|||
|
|
if time_rule["send_interval"] < 0:
|
|||
|
|
return "The minimum interval is set incorrectly"
|
|||
|
|
|
|||
|
|
if "time_range" in time_rule:
|
|||
|
|
if not isinstance(time_rule["time_range"], list):
|
|||
|
|
return "The time range is set incorrectly"
|
|||
|
|
if not len(time_rule["time_range"]) == 2:
|
|||
|
|
del time_rule["time_range"]
|
|||
|
|
else:
|
|||
|
|
time_range = time_rule["time_range"]
|
|||
|
|
if not (isinstance(time_range[0], int) and isinstance(time_range[1], int) and
|
|||
|
|
0 <= time_range[0] < time_range[1] <= 60 * 60 * 24):
|
|||
|
|
return "The time range is set incorrectly"
|
|||
|
|
|
|||
|
|
result["time_rule"] = time_rule
|
|||
|
|
|
|||
|
|
number_rule = task.get("number_rule", {})
|
|||
|
|
if "day_num" in number_rule:
|
|||
|
|
if not (isinstance(number_rule["day_num"], int) and number_rule["day_num"] >= 0):
|
|||
|
|
return "The minimum number of times per day is set incorrectly"
|
|||
|
|
|
|||
|
|
if "total" in number_rule:
|
|||
|
|
if not (isinstance(number_rule["total"], int) and number_rule["total"] >= 0):
|
|||
|
|
return "The maximum number of alarms is set incorrectly"
|
|||
|
|
|
|||
|
|
result["number_rule"] = number_rule
|
|||
|
|
|
|||
|
|
if "status" not in task:
|
|||
|
|
result["status"] = True
|
|||
|
|
if "status" in task:
|
|||
|
|
if isinstance(task["status"], bool):
|
|||
|
|
result["status"] = task["status"]
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
def set_task_conf_data(self, push_data: dict) -> Optional[str]:
|
|||
|
|
task_id = push_data.get("task_id", None)
|
|||
|
|
template_id = push_data.get("template_id")
|
|||
|
|
task = push_data.get("task_data")
|
|||
|
|
|
|||
|
|
target_task_conf = None
|
|||
|
|
if task_id is not None:
|
|||
|
|
tmp = self.task_conf.get_by_id(task_id)
|
|||
|
|
if tmp is None:
|
|||
|
|
target_task_conf = tmp
|
|||
|
|
|
|||
|
|
template = self.template_conf.get_by_id(template_id)
|
|||
|
|
|
|||
|
|
if not template:
|
|||
|
|
# 如果没有找到模板，则尝试加载默认的安全推送模板
|
|||
|
|
from .mods import load_task_template_by_file
|
|||
|
|
if not os.path.exists("/www/server/panel/mod/base/push_mod/safe_mod_push_template.json"):
|
|||
|
|
return "No alarm template was found"
|
|||
|
|
load_task_template_by_file("/www/server/panel/mod/base/push_mod/safe_mod_push_template.json")
|
|||
|
|
self.template_conf = TaskTemplateConfig()
|
|||
|
|
template = self.template_conf.get_by_id(template_id)
|
|||
|
|
if not template:
|
|||
|
|
return "No alarm template was found"
|
|||
|
|
|
|||
|
|
if template["unique"] and not target_task_conf:
|
|||
|
|
for i in self.task_conf.config:
|
|||
|
|
if i["template_id"] == template["id"]:
|
|||
|
|
target_task_conf = i
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
task_obj = PushSystem().get_task_object(template_id, template["load_cls"])
|
|||
|
|
if not task_obj:
|
|||
|
|
return "Loading task type error, you can try to fix the panel"
|
|||
|
|
|
|||
|
|
res = self.normalize_task_config(task, template)
|
|||
|
|
if isinstance(res, str):
|
|||
|
|
return res
|
|||
|
|
|
|||
|
|
task_data = task_obj.check_task_data(res["task_data"])
|
|||
|
|
if isinstance(task_data, str):
|
|||
|
|
return task_data
|
|||
|
|
|
|||
|
|
number_rule = task_obj.check_num_rule(res["number_rule"])
|
|||
|
|
if isinstance(number_rule, str):
|
|||
|
|
return number_rule
|
|||
|
|
|
|||
|
|
time_rule = task_obj.check_time_rule(res["time_rule"])
|
|||
|
|
if isinstance(time_rule, str):
|
|||
|
|
return time_rule
|
|||
|
|
|
|||
|
|
res["task_data"] = task_data
|
|||
|
|
res["number_rule"] = number_rule
|
|||
|
|
res["time_rule"] = time_rule
|
|||
|
|
|
|||
|
|
res["keyword"] = task_obj.get_keyword(task_data)
|
|||
|
|
res["source"] = task_obj.source_name
|
|||
|
|
res["title"] = task_obj.get_title(task_data)
|
|||
|
|
|
|||
|
|
if not target_task_conf:
|
|||
|
|
tmp = self.task_conf.get_by_keyword(res["source"], res["keyword"])
|
|||
|
|
if tmp:
|
|||
|
|
target_task_conf = tmp
|
|||
|
|
|
|||
|
|
if not target_task_conf:
|
|||
|
|
res["id"] = self.task_conf.nwe_id()
|
|||
|
|
res["template_id"] = template_id
|
|||
|
|
res["status"] = True
|
|||
|
|
res["pre_hook"] = {}
|
|||
|
|
res["after_hook"] = {}
|
|||
|
|
res["last_check"] = 0
|
|||
|
|
res["last_send"] = 0
|
|||
|
|
res["number_data"] = {}
|
|||
|
|
res["create_time"] = time.time()
|
|||
|
|
res["record_time"] = 0
|
|||
|
|
self.task_conf.config.append(res)
|
|||
|
|
task_obj.task_config_create_hook(res)
|
|||
|
|
else:
|
|||
|
|
target_task_conf.update(res)
|
|||
|
|
target_task_conf["last_check"] = 0
|
|||
|
|
target_task_conf["number_data"] = {} # 次数控制数据置空
|
|||
|
|
task_obj.task_config_update_hook(target_task_conf)
|
|||
|
|
|
|||
|
|
self.task_conf.save_config()
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def update_task_status(self, get):
|
|||
|
|
# 先调用 set_task_conf 修改任务配置
|
|||
|
|
set_conf_response = self.set_task_conf(get)
|
|||
|
|
|
|||
|
|
if set_conf_response['status'] != 0:
|
|||
|
|
return set_conf_response # 返回错误信息
|
|||
|
|
|
|||
|
|
# 读取任务数据
|
|||
|
|
file_path = '{}/data/mod_push_data/task.json'.format(public.get_panel_path())
|
|||
|
|
try:
|
|||
|
|
with open(file_path, 'r') as file:
|
|||
|
|
tasks = json.load(file)
|
|||
|
|
except (IOError, json.JSONDecodeError):
|
|||
|
|
return json_response(status=False, msg=public.lang("Unable to read task data."))
|
|||
|
|
# 查找对应的 task_id
|
|||
|
|
task_title = get.title.strip() # 假设 get 中有 title 参数
|
|||
|
|
task_id = None
|
|||
|
|
|
|||
|
|
for task in tasks:
|
|||
|
|
if task.get('title') == task_title:
|
|||
|
|
task_id = task.get('id')
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
if not task_id:
|
|||
|
|
return json_response(status=False, msg=public.lang("The task has not been found."))
|
|||
|
|
|
|||
|
|
# 调用 change_task_conf 修改任务状态
|
|||
|
|
get.task_id = task_id
|
|||
|
|
return self.change_task_conf(get)
|
|||
|
|
|
|||
|
|
def set_task_conf(self, get):
|
|||
|
|
task_id = None
|
|||
|
|
try:
|
|||
|
|
if hasattr(get, "task_id"):
|
|||
|
|
task_id = get.task_id.strip()
|
|||
|
|
if not task_id:
|
|||
|
|
task_id = None
|
|||
|
|
else:
|
|||
|
|
self.remove_task_conf(get)
|
|||
|
|
template_id = get.template_id.strip()
|
|||
|
|
task = json.loads(get.task_data.strip())
|
|||
|
|
except (AttributeError, json.JSONDecodeError, TypeError, ValueError):
|
|||
|
|
return json_response(status=False, msg="The parameter is incorrect")
|
|||
|
|
push_data = {
|
|||
|
|
"task_id": task_id,
|
|||
|
|
"template_id": template_id,
|
|||
|
|
"task_data": task,
|
|||
|
|
}
|
|||
|
|
res = self.set_task_conf_data(push_data)
|
|||
|
|
if res:
|
|||
|
|
return json_response(status=False, msg=res)
|
|||
|
|
return json_response(status=True, msg="The alarm task is saved successfully")
|
|||
|
|
|
|||
|
|
def change_task_conf(self, get):
|
|||
|
|
try:
|
|||
|
|
task_id = get.task_id.strip()
|
|||
|
|
status = int(get.status) # 获取status字段并转换为整数
|
|||
|
|
except (AttributeError, ValueError):
|
|||
|
|
return json_response(status=False, msg="Parameter error")
|
|||
|
|
|
|||
|
|
if status not in [0, 1]:
|
|||
|
|
return json_response(status=False, msg="Invalid status value")
|
|||
|
|
|
|||
|
|
tmp = self.task_conf.get_by_id(task_id)
|
|||
|
|
if tmp is None:
|
|||
|
|
return json_response(status=True, msg="No alarm task was queried")
|
|||
|
|
|
|||
|
|
tmp["status"] = bool(status) # 将status转换为布尔值并设置
|
|||
|
|
|
|||
|
|
self.task_conf.save_config()
|
|||
|
|
return json_response(status=True, msg="operate successfully")
|
|||
|
|
|
|||
|
|
def change_task(self, task_id, status):
|
|||
|
|
tmp = self.task_conf.get_by_id(task_id)
|
|||
|
|
tmp["status"] = bool(status) # 将status转换为布尔值并设置
|
|||
|
|
self.task_conf.save_config()
|
|||
|
|
|
|||
|
|
def remove_task_conf(self, get):
|
|||
|
|
try:
|
|||
|
|
task_id = get.task_id.strip()
|
|||
|
|
except AttributeError:
|
|||
|
|
return json_response(status=False, msg="The parameter is incorrect")
|
|||
|
|
|
|||
|
|
tmp = self.task_conf.get_by_id(task_id)
|
|||
|
|
if tmp is None:
|
|||
|
|
return json_response(status=True, msg="No alarm task was queried")
|
|||
|
|
|
|||
|
|
self.task_conf.config.remove(tmp)
|
|||
|
|
|
|||
|
|
self.task_conf.save_config()
|
|||
|
|
template = self.template_conf.get_by_id(tmp["template_id"])
|
|||
|
|
if template:
|
|||
|
|
task_obj = PushSystem().get_task_object(template["id"], template["load_cls"])
|
|||
|
|
if task_obj:
|
|||
|
|
task_obj.task_config_remove_hook(tmp)
|
|||
|
|
|
|||
|
|
return json_response(status=True, msg="operate successfully")
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def clear_task_record_by_task_id(task_id):
|
|||
|
|
tr_conf = TaskRecordConfig(task_id)
|
|||
|
|
if os.path.exists(tr_conf.config_file_path):
|
|||
|
|
os.remove(tr_conf.config_file_path)
|