python中使用gevent协程实现系统监控报警
一、版本说明二、主要逻辑与功能三、整体代码四、未解决问题 一、版本说明 版本 更新日期 1.0.Alpha2022-02-26二、主要逻辑与功能
主要功能还是监控系统信息并报警
内存利用率大于80%。发送告警邮件CPU利用率大于80%。发送告警邮件每晚12点定时发送系统信息，包括但不限于网卡信息、主机名与IP地址 三、整体代码代码逻辑比较简单，很容易就能看懂
注意：请修改代码中的邮件发送参数，具体百度网易邮箱开启SMTP
代码如下：
import psutil, time, logging, gevent, smtplib, datetime, schedule, socketfrom gevent import monkeyfrom email.mime.text import MIMETextfrom logging import handlersfomatter = logging.Formatter(' %(levelname)s %(asctime)s - %(funcName)s - %(thread)s [%(message)s]')# 创建日志器对象logger = logging.getLogger()# 日志级别总开关logger.setLevel("INFO")# 设置日志文件logFile = './default_monitor.log'# 创建日志文件hander# fileHandler = logging.FileHandler(logFile, mode='a')# fileHandler.setLevel(logging.INFO)# 创建控制台handerconsole_hander = logging.StreamHandler()# 创建控制台hander日志等级console_hander.setLevel(level='INFO')# 设置日志格式console_hander.setFormatter(fomatter)th = handlers.TimedRotatingFileHandler(filename=logFile, when='D', backupCount=3, encoding='utf-8')th.setFormatter(fomatter)th.setLevel(level='INFO')# 添加日志handerlogger.addHandler(console_hander)logger.addHandler(th)# logger.addHandler(fileHandler)# 设置邮件发送参数,需要自己修改成自己的参数mail_host = 'smtp.163.com'mail_user = 'cc17854388872@163.com'mail_pass = 'VBYZZtttttRDQJAXN'sender = 'cc17854388872@163.com'# 接收方邮箱地址receivers = ['1849999179@qq.com']# 获取当前时间current_time = datetime.datetime.now()# 使用gevent中的monky不修改原来使用的python标准库函数的程序的情况下，将程序转换成可以使用gevent框架的异步程序。monkey.patch_all()def print_system_info(mem, cpu): while True: # 闲的没事干加点日志输出吧 logger.info(f'Current Memory utilization is {mem.percent}%') logger.info(f'Current CPU utilization is {cpu}%') time.sleep(10)# 系统信息定时发送函数def system_sender(): # 创建任务调用mail_sender并传递两个参数，每天晚上12点定时发送系统信息邮件 schedule.every().day.at("20:37").do(mail_sender, title='system info', content=get_system()) logger.info('scheduler starting ') while True: schedule.run_pending() time.sleep(1)# 获取系统信息函数def get_system(): date = "" # 服务器基本信息，主机名与IP地址 myname = socket.getfqdn(socket.gethostname()) myaddr = socket.gethostbyname(myname) # 用户信息 now_time = time.strftime('%Y-%m-%d-%H:%M:%S', time.localtime(time.time())) start_time = datetime.datetime.fromtimestamp(psutil.boot_time()).strftime("%Y-%m-%d %H: %M: %S") # CPU信息 cpu_num = psutil.cpu_count(logical=False) cpu = (str(psutil.cpu_percent(1))) + '%' # 内存信息 total = str(round(psutil.virtual_memory().total / (1024.0 * 1024.0 * 1024.0), 2)) + 'G' total_used = str(round(psutil.virtual_memory().used / (1024.0 * 1024.0 * 1024.0), 2)) + 'G' total_free = str(round(psutil.virtual_memory().free / (1024.0 * 1024.0 * 1024.0), 2)) + 'G' memory = str(psutil.virtual_memory().percent) + '%' # 网卡信息 net = psutil.net_io_counters() bytes_sent = '{0:.2f} Mb'.format(net.bytes_recv / 1024 / 1024) bytes_rcvd = '{0:.2f} Mb'.format(net.bytes_sent / 1024 / 1024) date += "
当前用户: " + str(psutil.users()[0][0]) + '
' date += "
当前主机:" + str(myname) + '
' date += "
IP 地址: " + myaddr + '
' date += "
系统当前时间: " + now_time + '
' date += "
系统开机时间: " + start_time + '
' date += "
CPU个数: " + str(cpu_num) + '
' date += "
CPU使用率: " + cpu + '
' date += "
内存: " + total + '
' date += "
内存已使用: " + total_used + '
' date += "
剩余内存: " + total_free + '
' date += "
内存使用率: " + memory + '
' date += "
网卡发送流量: " + bytes_sent + '
' date += "
网卡接收流量: " + bytes_rcvd + '
' return datedef mail_sender(title, content): message = MIMEText(content, 'plain', 'utf-8') message['Subject'] = title message['From'] = sender message['To'] = receivers[0] try: smtpObj = smtplib.SMTP() # 连接到服务器 smtpObj.connect(mail_host, 25) # 登录到服务器 smtpObj.login(mail_user, mail_pass) # 发送 smtpObj.sendmail( sender, receivers, message.as_string()) # 退出 smtpObj.quit() logger.info(f'Mail sent {title} successfully') except smtplib.SMTPException as e: logger.error('error', e) # 打印错误def check_memory_alarm(mem): logger.info('check_memory_alarm start') while True: # 内存利用率大于80%发送告警邮件 if int(mem.percent) > 80: content = f'
Current time：{current_time}
Memory utilization exceeds 80%' mail_sender('memory warning', content) break time.sleep(5)# cpu告警模块def check_cpu_alarm(cpu): logger.info('check_cpu_alarm start') while True: # cpu利用率大于80%发送告警邮件 if int(cpu) > 80: content = f'
Current time：{current_time}
CPU utilization exceeds 80%' mail_sender('cpu warning', content) break time.sleep(5)def main(): mem = psutil.virtual_memory() cpu = psutil.cpu_percent(interval=1) gevent.joinall( [ gevent.spawn(print_system_info, mem, cpu), gevent.spawn(check_memory_alarm, mem), gevent.spawn(check_cpu_alarm, cpu), gevent.spawn(system_sender) ] )if __name__ == '__main__': main()