176 lines
5.6 KiB
Python
176 lines
5.6 KiB
Python
#!/usr/bin/python3
|
||
import _load
|
||
import psutil
|
||
import smtplib
|
||
import argparse
|
||
import yaml
|
||
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
from email.mime.text import MIMEText
|
||
from email.header import Header
|
||
from email.mime.multipart import MIMEMultipart
|
||
|
||
# 定义公共的部分
|
||
path = Path(__file__)
|
||
report_dir = path.parent.parent / "report"
|
||
# 初始化目录
|
||
if not report_dir.exists():
|
||
# shutil.rmtree(data_dir)
|
||
report_dir.mkdir()
|
||
#报告文件
|
||
week=str(datetime.now().isocalendar()[1])
|
||
file_name = "ComputerInfos-week-" + week + ".txt"
|
||
file_path = report_dir/file_name
|
||
|
||
#发邮件
|
||
def send_mail():
|
||
#发件人
|
||
sender = 'mh@unamail.com'
|
||
# 接收邮件，可以发给多人
|
||
receivers = ['mh@unamail.com']
|
||
# 邮件主体
|
||
msg = MIMEMultipart()
|
||
#正文
|
||
message = "第{0}周，{1}运行信息检测已完成，详情请查看附件！".format(week,get_IP())
|
||
msg.attach(MIMEText(message, 'plain', _charset="utf-8"))
|
||
# 发送者
|
||
msg['From'] = Header(sender, 'utf-8')
|
||
# 接收者
|
||
msg['To'] = Header(receivers[0], 'utf-8')
|
||
# 主题
|
||
subject = '【长期任务】第{0}周{1}系统运行信息'.format(week,get_IP())
|
||
msg['Subject'] = Header(subject, 'utf-8')
|
||
#附件信息
|
||
att = MIMEText(open(str(file_path), 'rb').read(), 'base64', 'utf-8')
|
||
att["Content-Type"] = 'application/octet-stream'
|
||
att["Content-Disposition"] = 'attachment; filename="{}"'.format(file_name)
|
||
msg.attach(att)
|
||
|
||
try:
|
||
smtpObj = smtplib.SMTP('10.10.110.102')
|
||
smtpObj.sendmail(sender, receivers, msg.as_string())
|
||
print ("邮件发送成功")
|
||
except smtplib.SMTPException:
|
||
print ("Error: 无法发送邮件")
|
||
|
||
|
||
#写入文件
|
||
def save_txt(datas):
|
||
with open(str(file_path),'a+',encoding='utf-8',newline='') as file_txt:
|
||
[file_txt.write(str(item)+'\n') for item in datas]
|
||
|
||
|
||
# 获取本机磁盘使用率和剩余空间G信息
|
||
def get_disk_info():
|
||
# 循环磁盘分区
|
||
content = []
|
||
for disk in psutil.disk_partitions():
|
||
# 读写方式 光盘 or 有效磁盘类型
|
||
if 'cdrom' in disk.opts or disk.fstype == '':
|
||
continue
|
||
disk_name_arr = disk.device.split(':')
|
||
disk_name = disk_name_arr[0]
|
||
disk_info = psutil.disk_usage(disk.device)
|
||
# 磁盘剩余空间，单位G
|
||
free_disk_size = disk_info.free//1024//1024//1024
|
||
# 磁盘总空间，单位G
|
||
total_disk_size = disk_info.total//1024//1024//1024
|
||
# 当前磁盘使用率、剩余空间G和磁盘总空间信息
|
||
info = "\t{0}盘使用率：{1}%， 剩余空间：{2}G， 总大小：{3}G".format(disk_name, str(disk_info.percent),free_disk_size,total_disk_size)
|
||
# print(info)
|
||
# 拼接多个磁盘的信息
|
||
content.append(info)
|
||
print(content)
|
||
save_txt(content)
|
||
|
||
# 获取某个目录的大小
|
||
def get_dir_size(path):
|
||
list1 = []
|
||
for item in path.iterdir():
|
||
if item.is_file():
|
||
file_size = file_path.stat().st_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}")
|
||
save_txt(['\t--------------'])
|
||
get_dir_size(item)
|
||
|
||
str_dir_tex = '\t{0} 的大小为: {1:.4f} MB'.format(path, (sum(list1)/1024/1024))
|
||
print(str_dir_tex)
|
||
save_txt([str_dir_tex])
|
||
|
||
|
||
# cpu信息
|
||
def get_cpu_info():
|
||
cpu_percent = psutil.cpu_percent(interval=1)
|
||
cpu_info = ["","CPU使用率：{0}%".format(cpu_percent),""]
|
||
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),""]
|
||
print(memory_info)
|
||
# return memory_info
|
||
save_txt(memory_info)
|
||
|
||
|
||
def get_IP():
|
||
"""获取ipv4地址"""
|
||
dic = psutil.net_if_addrs()
|
||
ipv4_list = []
|
||
for adapter in dic:
|
||
snicList = dic[adapter]
|
||
for snic in snicList:
|
||
# if snic.family.name in {'AF_LINK', 'AF_PACKET'}:
|
||
# mac = snic.address
|
||
if snic.family.name == 'AF_INET':
|
||
ipv4 = snic.address
|
||
if ipv4 != '127.0.0.1':
|
||
ipv4_list.append(ipv4)
|
||
# elif snic.family.name == 'AF_INET6':
|
||
# ipv6 = snic.address
|
||
print(ipv4_list)
|
||
return(ipv4_list[0])
|
||
|
||
def main():
|
||
commoninfo = []
|
||
if not file_path.exists():
|
||
t1 = "本周是第{0}周".format(week)
|
||
commoninfo.append(t1)
|
||
commoninfo.append("")
|
||
commoninfo.append("--------------------START--------------------------")
|
||
commoninfo.append("检测时间： {0}".format(datetime.now()))
|
||
commoninfo.append("检测IP： {0}".format(get_IP()))
|
||
save_txt(commoninfo)
|
||
save_txt(["","磁盘使用情况:",""])
|
||
get_disk_info()
|
||
get_cpu_info()
|
||
get_memory_info()
|
||
path = "D:\\agent\\ubackup\\uagent\\logs"
|
||
directory_path = Path(path)
|
||
save_txt(["{0}目录大小:".format(path)])
|
||
get_dir_size(directory_path)
|
||
|
||
save_txt(['----------------------END----------------------------'])
|
||
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|
||
#默认不发邮件，通过参数控制
|
||
parser = argparse.ArgumentParser(description='send email')
|
||
parser.add_argument("--send-mail", type=bool, default=False)
|
||
args = parser.parse_args()
|
||
if args.send_mail:
|
||
send_mail()
|