linux 系统资源报警

import os
import smtplib
from email.mime.text import MIMEText

def get_cpu_usage():
    # 使用top命令获取CPU使用率
    cpu_usage = os.popen("top -bn1 | grep 'Cpu(s)' | awk '{print $2 + $4}'").readline().strip()
    return float(cpu_usage)

def get_memory_usage():
    # 使用free命令获取内存使用率
    mem_info = os.popen("free -m | awk 'NR==2{printf \"%.2f\", $3*100/$2 }'").readline().strip()
    return float(mem_info)

def send_email_alert(subject, message):
    # 发送报警邮件（需要配置SMTP服务器和邮箱账号）
    msg = MIMEText(message)
    msg['Subject'] = subject
    msg['From'] = 'alert@example.com'
    msg['To'] = 'admin@example.com'

    with smtplib.SMTP('smtp.example.com') as server:
        server.login('alert@example.com', 'password')
        server.send_message(msg)

def main():
    cpu_threshold = 80.0  # CPU使用率阈值（%）
    mem_threshold = 80.0  # 内存使用率阈值（%）

    cpu_usage = get_cpu_usage()
    mem_usage = get_memory_usage()

    if cpu_usage > cpu_threshold:
        send_email_alert('CPU使用率报警', f'CPU使用率过高：{cpu_usage}%')

    if mem_usage > mem_threshold:
        send_email_alert('内存使用率报警', f'内存使用率过高：{mem_usage}%')

if __name__ == '__main__':
    main()