提交阈值
This commit is contained in:
parent
f0d57397af
commit
c2cbbd6567
|
|
@ -9,7 +9,7 @@
|
|||
"type": "python",
|
||||
"request": "launch",
|
||||
"program": "${file}",
|
||||
"args": ["--send-mail=true"],
|
||||
//"args": ["--send-mail=true"],
|
||||
"console": "integratedTerminal",
|
||||
"justMyCode": true
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
#!/usr/bin/python3file_today_path
|
||||
import _load
|
||||
import psutil
|
||||
import psutil
|
||||
import smtplib
|
||||
import argparse
|
||||
import yaml
|
||||
|
|
@ -20,64 +20,74 @@ config_file_patch = path.parent.parent / "etc/check_config.yml"
|
|||
if not report_dir.exists():
|
||||
# shutil.rmtree(data_dir)
|
||||
report_dir.mkdir()
|
||||
#报告文件
|
||||
week=str(datetime.now().isocalendar()[1])
|
||||
# 报告文件
|
||||
week = str(datetime.now().isocalendar()[1])
|
||||
file_name_week = "ComputerInfos-week-" + week + ".txt"
|
||||
file_name_today="ComputerInfos-today.txt"
|
||||
file_name_today = "ComputerInfos-today.txt"
|
||||
|
||||
file_today_path = report_dir/file_name_today
|
||||
file_week_path = report_dir/file_name_week
|
||||
|
||||
#获取配置文件
|
||||
with open(config_file_patch,"r",encoding='utf-8') as fy:
|
||||
config = yaml.safe_load(fy)
|
||||
#print(config)
|
||||
# 获取配置文件
|
||||
with open(config_file_patch, "r", encoding='utf-8') as fy:
|
||||
config = yaml.safe_load(fy)
|
||||
# print(config)
|
||||
|
||||
need_sendmail = False
|
||||
|
||||
# 发邮件
|
||||
|
||||
|
||||
#发邮件
|
||||
def send_mail():
|
||||
#发件人
|
||||
# 发件人
|
||||
sender = 'mh@unamail.com'
|
||||
# 接收邮件，可以发给多人
|
||||
receivers = config["receivers"]
|
||||
# 邮件主体
|
||||
msg = MIMEMultipart()
|
||||
#正文
|
||||
with open(str(file_today_path), "r", encoding='utf-8') as f: #打开文本
|
||||
msgdata = f.read()
|
||||
message = "第{0}周，{1}资源使用情况检测已完成，本次检测内容如下所示，本周的所有信息请查看附件！\n{2}".format(week,get_IP(),msgdata)
|
||||
# 正文
|
||||
with open(str(file_today_path), "r", encoding='utf-8') as f: # 打开文本
|
||||
msgdata = f.read()
|
||||
message = "第{0}周，{1}资源使用情况检测已完成，本次检测内容如下所示，本周的所有信息请查看附件！\n{2}".format(
|
||||
week, get_IP(), msgdata)
|
||||
msg.attach(MIMEText(message, 'plain', _charset="utf-8"))
|
||||
# 发送者
|
||||
msg['From'] = Header(sender, 'utf-8')
|
||||
# 接收者
|
||||
msg['To'] = Header(receivers[0], 'utf-8')
|
||||
msg['To'] = Header(receivers[0], 'utf-8')
|
||||
# 主题
|
||||
subject = '【长期任务】第{0}周{1}系统运行信息'.format(week,get_IP())
|
||||
subject = '【长期任务】第{0}周{1}系统运行信息'.format(week, get_IP())
|
||||
msg['Subject'] = Header(subject, 'utf-8')
|
||||
#附件信息
|
||||
# 附件信息
|
||||
att = MIMEText(open(str(file_week_path), 'rb').read(), 'base64', 'utf-8')
|
||||
att["Content-Type"] = 'application/octet-stream'
|
||||
att["Content-Disposition"] = 'attachment; filename="{}"'.format(file_name_week)
|
||||
att["Content-Disposition"] = 'attachment; filename="{}"'.format(
|
||||
file_name_week)
|
||||
msg.attach(att)
|
||||
|
||||
try:
|
||||
smtpObj = smtplib.SMTP('10.10.110.102')
|
||||
smtpObj.sendmail(sender, receivers, msg.as_string())
|
||||
print ("邮件发送成功")
|
||||
print("邮件发送成功")
|
||||
except smtplib.SMTPException:
|
||||
print ("Error: 无法发送邮件")
|
||||
print("Error: 无法发送邮件")
|
||||
|
||||
|
||||
#写入文件
|
||||
# 写入文件
|
||||
def save_txt(datas):
|
||||
with open(str(file_today_path),'a+',encoding='utf-8',newline='') as file_txt:
|
||||
[file_txt.write(str(item)+'\n') for item in datas]
|
||||
with open(str(file_today_path), 'a+', encoding='utf-8', newline='') as file_txt:
|
||||
[file_txt.write(str(item)+'\n') for item in datas]
|
||||
|
||||
# 合并文件
|
||||
|
||||
|
||||
#合并文件
|
||||
def save_all_tex():
|
||||
with open(file_today_path, 'rb') as f1, open(file_week_path, 'ab+') as f2:
|
||||
f2.write(f1.read())
|
||||
|
||||
# 获取本机磁盘使用率和剩余空间G信息
|
||||
|
||||
|
||||
def get_disk_info():
|
||||
# 循环磁盘分区
|
||||
content = []
|
||||
|
|
@ -91,28 +101,38 @@ def get_disk_info():
|
|||
free_disk_size = disk_info.free//1024//1024//1024
|
||||
# 磁盘总空间，单位G
|
||||
total_disk_size = disk_info.total//1024//1024//1024
|
||||
# 检测是否超过阈值
|
||||
error_msg = ""
|
||||
if disk_info.percent > config["Disk_MAX"]:
|
||||
error_msg = "\n\t警告： {0}使用率超过{1}，请及时处理！".format(
|
||||
disk_name, str(config["Disk_MAX"]))
|
||||
global need_sendmail
|
||||
need_sendmail = True
|
||||
|
||||
# 当前磁盘使用率、剩余空间G和磁盘总空间信息
|
||||
info = "\t{0}\t使用率：{1}%， 剩余空间：{2}G， 总大小：{3}G".format(disk_name, str(disk_info.percent),free_disk_size,total_disk_size)
|
||||
info = "\t{0}\t使用率：{1}%， 剩余空间：{2}G， 总大小：{3}G {4}".format(
|
||||
disk_name, str(disk_info.percent), free_disk_size, total_disk_size, error_msg)
|
||||
# print(info)
|
||||
# 拼接多个磁盘的信息
|
||||
content.append(info)
|
||||
print(content)
|
||||
save_txt(content)
|
||||
|
||||
|
||||
# 获取某个目录的大小
|
||||
def get_dir_size(path):
|
||||
def get_dir_size(path):
|
||||
list1 = []
|
||||
for item in path.iterdir():
|
||||
if item.is_file():
|
||||
file_size = file_today_path.stat().st_size
|
||||
list1.append(file_size)
|
||||
str_tex=f"\t{item}的大小是{file_size}字节"
|
||||
list1.append(file_size)
|
||||
str_tex = f"\t{item}的大小是{file_size}字节"
|
||||
print(str_tex)
|
||||
save_txt([str_tex])
|
||||
elif item.is_dir():
|
||||
#print(f"目录: {item}")
|
||||
# print(f"目录: {item}")
|
||||
save_txt(['\t--------------'])
|
||||
get_dir_size(item)
|
||||
get_dir_size(item)
|
||||
|
||||
str_dir_tex = '\t{0} 的大小为: {1:.4f} MB'.format(path, (sum(list1)/1024/1024))
|
||||
print(str_dir_tex)
|
||||
|
|
@ -122,18 +142,32 @@ def get_dir_size(path):
|
|||
# cpu信息
|
||||
def get_cpu_info():
|
||||
cpu_percent = psutil.cpu_percent(interval=1)
|
||||
cpu_info = ["","CPU使用率：{0}%".format(cpu_percent),""]
|
||||
# 检测是否超过阈值
|
||||
error_msg = ""
|
||||
if cpu_percent > config["CPU_MAX"]:
|
||||
error_msg = "\n\t警告： CPU使用率超过{0}，请及时处理！".format(str(config["CPU_MAX"]))
|
||||
global need_sendmail
|
||||
need_sendmail = True
|
||||
cpu_info = ["", "CPU使用率：{0}% {1}".format(cpu_percent, error_msg), ""]
|
||||
print(cpu_info)
|
||||
# return cpu_info
|
||||
save_txt(cpu_info)
|
||||
|
||||
|
||||
# 内存信息
|
||||
def get_memory_info():
|
||||
virtual_memory = psutil.virtual_memory()
|
||||
used_memory = virtual_memory.used/1024/1024/1024
|
||||
free_memory = virtual_memory.free/1024/1024/1024
|
||||
memory_percent = virtual_memory.percent
|
||||
memory_info = ["内存使用：{0:0.2f}G，使用率{1:0.1f}%，剩余内存：{2:0.2f}G".format(used_memory, memory_percent, free_memory),""]
|
||||
# 检测是否超过阈值
|
||||
error_msg = ""
|
||||
if memory_percent > config["Memory_MAX"]:
|
||||
error_msg = "\n\t警告： 内存使用率超过{0}，请及时处理！".format(str(config["Memory_MAX"]))
|
||||
global need_sendmail
|
||||
need_sendmail = True
|
||||
memory_info = ["内存使用：{0:0.2f}G，使用率{1:0.1f}%，剩余内存：{2:0.2f}G {3}".format(
|
||||
used_memory, memory_percent, free_memory,error_msg), ""]
|
||||
print(memory_info)
|
||||
# return memory_info
|
||||
save_txt(memory_info)
|
||||
|
|
@ -155,13 +189,14 @@ def get_IP():
|
|||
# elif snic.family.name == 'AF_INET6':
|
||||
# ipv6 = snic.address
|
||||
print(ipv4_list)
|
||||
return(ipv4_list[0])
|
||||
return (ipv4_list[0])
|
||||
|
||||
|
||||
def main():
|
||||
#删除之前的文件
|
||||
# 删除之前的文件
|
||||
if file_today_path.exists():
|
||||
file_today_path.unlink()
|
||||
#公共信息
|
||||
# 公共信息
|
||||
commoninfo = []
|
||||
if not file_week_path.exists():
|
||||
t1 = "本周是第{0}周".format(week)
|
||||
|
|
@ -171,7 +206,7 @@ def main():
|
|||
commoninfo.append("检测时间： {0}".format(datetime.now()))
|
||||
commoninfo.append("检测IP： {0}".format(get_IP()))
|
||||
save_txt(commoninfo)
|
||||
save_txt(["","磁盘使用情况:",""])
|
||||
save_txt(["", "磁盘使用情况:", ""])
|
||||
get_disk_info()
|
||||
get_cpu_info()
|
||||
get_memory_info()
|
||||
|
|
@ -179,17 +214,17 @@ def main():
|
|||
directory_path = Path(path)
|
||||
save_txt(["{0}目录大小:".format(path)])
|
||||
get_dir_size(directory_path)
|
||||
|
||||
save_txt(['----------------------END----------------------------'])
|
||||
|
||||
save_txt(['----------------------END----------------------------'])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
save_all_tex()
|
||||
#默认不发邮件，通过参数控制
|
||||
# 默认不发邮件，通过参数控制
|
||||
parser = argparse.ArgumentParser(description='send email')
|
||||
parser.add_argument("--send-mail", type=bool, default=False)
|
||||
args = parser.parse_args()
|
||||
if args.send_mail:
|
||||
# 超过阈值的和需要发邮件的
|
||||
if (args.send_mail or need_sendmail):
|
||||
send_mail()
|
||||
|
|
|
|||
|
|
@ -1,8 +1,17 @@
|
|||
#邮箱收件人,支持多个
|
||||
receivers:
|
||||
- "mh@unamail.com"
|
||||
- "ypy@unamail.com"
|
||||
# - "ypy@unamail.com"
|
||||
|
||||
#agent安装路径,windows路径要用双斜线，Linux不需要
|
||||
agent_dir : "D:\\agent\\ubackup\\uagent\\logs"
|
||||
|
||||
|
||||
#资源阈值,超过这个值的会发告警
|
||||
#CPU使用率最大值
|
||||
CPU_MAX: 80
|
||||
#内存使用率最大值
|
||||
Memory_MAX: 80
|
||||
#磁盘使用率最大值
|
||||
Disk_MAX: 90
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue