Initial YakPanel commit
This commit is contained in:
Niranjan
239
mod/base/push_mod/database_push.py
Normal file
239
mod/base/push_mod/database_push.py
Normal file
@@ -0,0 +1,239 @@
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import ipaddress
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Tuple, Union, Optional
|
||||
|
||||
from .send_tool import WxAccountMsg
|
||||
from .base_task import BaseTask
|
||||
from .util import read_file, DB, GET_CLASS
|
||||
|
||||
try:
|
||||
if "/www/server/panel/class" not in sys.path:
|
||||
sys.path.insert(0, "/www/server/panel/class")
|
||||
from panel_msg.collector import DatabasePushMsgCollect
|
||||
except ImportError:
|
||||
DatabasePushMsgCollect = None
|
||||
|
||||
|
||||
def is_ipaddress(ip_data: str) -> bool:
|
||||
try:
|
||||
ipaddress.ip_address(ip_data)
|
||||
except ValueError:
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
# class MysqlPwdEndTimeTask(BaseTask):
|
||||
#
|
||||
# def __init__(self):
|
||||
# super().__init__()
|
||||
# self.template_name = "MySQL数据库密码到期"
|
||||
# self.source_name = "mysql_pwd_end"
|
||||
#
|
||||
# self.push_db_user = ""
|
||||
#
|
||||
# def get_title(self, task_data: dict) -> str:
|
||||
# return "Msql:" + task_data["project"][1] + "用户密码到期提醒"
|
||||
#
|
||||
# def check_task_data(self, task_data: dict) -> Union[dict, str]:
|
||||
# task_data["interval"] = 600
|
||||
# if not (isinstance(task_data["project"], list) and len(task_data["project"]) == 3):
|
||||
# return "设置的用户格式错误"
|
||||
# project = task_data["project"]
|
||||
# if not (isinstance(project[0], int) and isinstance(project[1], str) and is_ipaddress(project[2])):
|
||||
# return "设置的检测用户格式错误"
|
||||
#
|
||||
# if not (isinstance(task_data["cycle"], int) and task_data["cycle"] >= 1):
|
||||
# return "到期时间参数错误，至少为 1 天"
|
||||
# return task_data
|
||||
#
|
||||
# def get_keyword(self, task_data: dict) -> str:
|
||||
# return "_".join([str(i) for i in task_data["project"]])
|
||||
#
|
||||
# def check_num_rule(self, num_rule: dict) -> Union[dict, str]:
|
||||
# num_rule["day_num"] = 1
|
||||
# return num_rule
|
||||
#
|
||||
# def get_push_data(self, task_id: str, task_data: dict) -> Optional[dict]:
|
||||
# sid = task_data["project"][0]
|
||||
# username = task_data["project"][1]
|
||||
# host = task_data["project"][2]
|
||||
#
|
||||
# if "/www/server/panel/class" not in sys.path:
|
||||
# sys.path.insert(0, "/www/server/panel/class")
|
||||
# try:
|
||||
# import panelMysql
|
||||
# import db_mysql
|
||||
# except ImportError:
|
||||
# return None
|
||||
#
|
||||
# if sid == 0:
|
||||
# try:
|
||||
# db_port = int(panelMysql.panelMysql().query("show global variables like 'port'")[0][1])
|
||||
# if db_port == 0:
|
||||
# db_port = 3306
|
||||
# except:
|
||||
# db_port = 3306
|
||||
# conn_config = {
|
||||
# "db_host": "localhost",
|
||||
# "db_port": db_port,
|
||||
# "db_user": "root",
|
||||
# "db_password": DB("config").where("id=?", (1,)).getField("mysql_root"),
|
||||
# "ps": "local server",
|
||||
# }
|
||||
# else:
|
||||
# conn_config = DB("database_servers").where("id=? AND LOWER(db_type)=LOWER('mysql')", (sid,)).find()
|
||||
# if not conn_config:
|
||||
# return None
|
||||
#
|
||||
# mysql_obj = db_mysql.panelMysql().set_host(conn_config["db_host"], conn_config["db_port"], None,
|
||||
# conn_config["db_user"], conn_config["db_password"])
|
||||
# if isinstance(mysql_obj, bool):
|
||||
# return None
|
||||
#
|
||||
# data_list = mysql_obj.query(
|
||||
# "SELECT password_last_changed FROM mysql.user WHERE user='{}' AND host='{}';".format(username, host))
|
||||
#
|
||||
# if not isinstance(data_list, list) or not data_list:
|
||||
# return None
|
||||
#
|
||||
# try:
|
||||
# # todo：检查这里的时间转化逻辑问题
|
||||
# last_time = data_list[0][0]
|
||||
# expire_time = last_time + timedelta(days=task_data["cycle"])
|
||||
# except:
|
||||
# return None
|
||||
#
|
||||
# if datetime.now() > expire_time:
|
||||
# self.title = self.get_title(task_data)
|
||||
# self.push_db_user = username
|
||||
# return {"msg_list": [
|
||||
# ">告警类型：MySQL密码即将到期",
|
||||
# ">Content of alarm: {} {}@{} 密码过期时间<font color=#ff0000>{} 天</font>".format(
|
||||
# conn_config["ps"], username, host, expire_time.strftime("%Y-%m-%d %H:%M:%S"))
|
||||
# ]}
|
||||
#
|
||||
# def filter_template(self, template: dict) -> Optional[dict]:
|
||||
# return template
|
||||
#
|
||||
# def to_sms_msg(self, push_data: dict, push_public_data: dict) -> Tuple[str, dict]:
|
||||
# return "", {}
|
||||
#
|
||||
# def to_wx_account_msg(self, push_data: dict, push_public_data: dict) -> WxAccountMsg:
|
||||
# msg = WxAccountMsg.new_msg()
|
||||
# msg.thing_type = "MySQL数据库密码到期"
|
||||
# msg.msg = "Mysql用户：{}的密码即将过期，请注意".format(self.push_db_user)
|
||||
# msg.next_msg = "Please log in to the panel to view the host status"
|
||||
# return msg
|
||||
|
||||
|
||||
class MysqlReplicateStatusTask(BaseTask):
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.template_name = "MySQL主从复制异常告警"
|
||||
self.source_name = "mysql_replicate_status"
|
||||
self.title = "MySQL主从复制异常告警"
|
||||
|
||||
self.slave_ip = ''
|
||||
|
||||
def check_task_data(self, task_data: dict) -> Union[dict, str]:
|
||||
if not (isinstance(task_data["project"], str) and task_data["project"]):
|
||||
return "请选择告警的从库！"
|
||||
|
||||
if not (isinstance(task_data["count"], int) and task_data["count"] in (1, 2)):
|
||||
return "是否自动修复选择错误！"
|
||||
|
||||
if not (isinstance(task_data["interval"], int) and task_data["interval"] >= 60):
|
||||
return "检查间隔时间错误，至少需要60s的间隔"
|
||||
return task_data
|
||||
|
||||
def get_keyword(self, task_data: dict) -> str:
|
||||
return task_data["project"]
|
||||
|
||||
def get_push_data(self, task_id: str, task_data: dict) -> Optional[dict]:
|
||||
import PluginLoader
|
||||
|
||||
args = GET_CLASS()
|
||||
args.slave_ip = task_data["project"]
|
||||
res = PluginLoader.plugin_run("mysql_replicate", "get_replicate_status", args)
|
||||
if res.get("status", False) is False:
|
||||
return None
|
||||
|
||||
self.slave_ip = task_data["project"]
|
||||
if len(res.get("data", [])) == 0:
|
||||
s_list = [">告警类型：MySQL主从复制异常告警",
|
||||
">Content of alarm: <font color=#ff0000>从库 {} 主从复制已停止，请尽快登录面板查看详情</font>".format(
|
||||
task_data["project"])]
|
||||
return {"msg_list": s_list}
|
||||
|
||||
sql_status = io_status = False
|
||||
for item in res.get("data", []):
|
||||
if item["name"] == "Slave_IO_Running" and item["value"] == "Yes":
|
||||
io_status = True
|
||||
if item["name"] == "Slave_SQL_Running" and item["value"] == "Yes":
|
||||
sql_status = True
|
||||
if io_status is True and sql_status is True:
|
||||
break
|
||||
|
||||
if io_status is False or sql_status is False:
|
||||
repair_txt = "请尽快登录面板查看详情"
|
||||
if task_data["count"] == 1: # 自动修复
|
||||
PluginLoader.plugin_run("mysql_replicate", "repair_replicate", args)
|
||||
repair_txt = "，正在尝试修复"
|
||||
|
||||
s_list = [">告警类型：MySQL主从复制异常告警",
|
||||
">Content of alarm: <font color=#ff0000>从库 {} 主从复制发生异常{}</font>".format(
|
||||
task_data["project"], repair_txt)]
|
||||
return {"msg_list": s_list}
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def _get_mysql_replicate():
|
||||
slave_list = []
|
||||
mysql_replicate_path = os.path.join("/www/server/panel/plugin", "mysql_replicate", "config.json")
|
||||
if os.path.isfile(mysql_replicate_path):
|
||||
conf = read_file(mysql_replicate_path)
|
||||
try:
|
||||
conf = json.loads(conf)
|
||||
slave_list = [{"title": slave_ip, "value": slave_ip} for slave_ip in conf["slave"].keys()]
|
||||
except:
|
||||
pass
|
||||
return slave_list
|
||||
|
||||
def filter_template(self, template: dict) -> Optional[dict]:
|
||||
template["field"][0]["items"] = self._get_mysql_replicate()
|
||||
if not template["field"][0]["items"]:
|
||||
return None
|
||||
return template
|
||||
|
||||
def to_sms_msg(self, push_data: dict, push_public_data: dict) -> Tuple[str, dict]:
|
||||
return '', {}
|
||||
|
||||
def to_wx_account_msg(self, push_data: dict, push_public_data: dict) -> WxAccountMsg:
|
||||
msg = WxAccountMsg.new_msg()
|
||||
msg.thing_type = "MySQL主从复制异常告警"
|
||||
msg.msg = "从库 {} 主从复制发生异常".format(self.slave_ip)
|
||||
msg.next_msg = "请登录面板，在[软件商店-MySQL主从复制(重构版)]中查看"
|
||||
return msg
|
||||
|
||||
|
||||
class ViewMsgFormat(object):
|
||||
_FORMAT = {
|
||||
"30": (
|
||||
lambda x: "<span>剩余时间小于{}天{}</span>".format(
|
||||
x["task_data"].get("cycle"),
|
||||
("(如未处理，次日会重新发送1次，持续%d天)" % x.get("number_rule", {}).get("day_num", 0)) if x.get(
|
||||
"number_rule", {}).get("day_num", 0) else ""
|
||||
|
||||
)
|
||||
),
|
||||
"31": (lambda x: "<span>MySQL主从复制异常告警</span>".format()),
|
||||
}
|
||||
|
||||
def get_msg(self, task: dict) -> Optional[str]:
|
||||
if task["template_id"] in self._FORMAT:
|
||||
return self._FORMAT[task["template_id"]](task)
|
||||
return None
|
||||
Reference in New Issue
Block a user