在日常的系统管理工作中,监测服务器的资源占用情况至关重要,我们需要及时获得通知以便采取相应措施。我新装了一台UbuntuServer服务器,写了一个可以定期收集服务器的CPU、内存、网络和磁盘信息,并通过邮件将这些信息发送给管理员的Python脚本,分享一下。
看效果:
话不多说,直接上源码:
用到了psutil先安装一下需要的psutil库:
pip install psutil
mail.py
import psutil
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
import datetime
# 配置SMTP服务器和邮箱信息
smtp_server = # SMTP服务器地址
smtp_port = # SMTP服务器端口
sender_email = # 发件人邮箱
password = # 发件人邮箱密码
receiver_email = # 管理员邮箱
def get_system_info():
# 获取当前时间
now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# CPU信息
cpu_count = psutil.cpu_count(logical=False)
cpu_percent = psutil.cpu_percent(interval=1)
# 内存信息
memory = psutil.virtual_memory()
mem_total = round(memory.total / (1024.0 ** 3), 2)
mem_used = round(memory.used / (1024.0 ** 3), 2)
mem_percent = memory.percent
# 网络信息
net_io_counters = psutil.net_io_counters()
bytes_sent = round(net_io_counters.bytes_sent / (1024.0 ** 3), 2)
bytes_recv = round(net_io_counters.bytes_recv / (1024.0 ** 3), 2)
# 磁盘信息
disk_usage = psutil.disk_usage('/')
disk_total = round(disk_usage.total / (1024.0 ** 3), 2)
disk_used = round(disk_usage.used / (1024.0 ** 3), 2)
disk_percent = disk_usage.percent
# 格式化信息
system_info = f"""
Time: {now}
CPU Count: {cpu_count}
CPU Usage: {cpu_percent}%
Memory Total: {mem_total} GB
Memory Used: {mem_used} GB
Memory Usage: {mem_percent}%
Bytes Sent: {bytes_sent} GB
Bytes Received: {bytes_recv} GB
Disk Total: {disk_total} GB
Disk Used: {disk_used} GB
Disk Usage: {disk_percent}%
"""
return system_info
def send_email(subject, body, sender, password, receiver):
msg = MIMEMultipart()
msg['From'] = sender
msg['To'] = receiver
msg['Subject'] = subject
msg.attach(MIMEText(body, 'plain'))
server = smtplib.SMTP(smtp_server, smtp_port)
server.starttls()
server.login(sender, password)
text = msg.as_string()
server.sendmail(sender, receiver, text)
server.quit()
# 获取系统信息
system_info = get_system_info()
# 发送邮件
subject = 'UbuntuServer System Information'
send_email(subject, system_info, sender_email, password, receiver_email)
注意画框的部分要写填你自己的信息
把mail.py复制下来,现在可以运行测试一下,运行一次即可向管理员邮箱发一次信息。
再打开命令行使用corn让它定时运行:
编辑当前用户的crontab
文件
crontab -e
打开编辑器中,添加任务(写你自己的路径)
0 */2 * * * /path/to/your/script/mail.py
这行的含义是:
0
表示在每小时的第0分钟运行。*/2
表示每两个小时。*
表示每天的任意一天。*
表示每个月的任意一月。*
表示每周的任意一周。/path/to/your/script/mail.py
是你的mail.py
脚本的完整路径。
我设置的是两个小时执行一次,确保将/path/to/your/script/mail.py
替换为你的mail.py
脚本的实际路径。
保存并关闭文件后,cron
会自动加载新的定时任务。你可以使用以下命令来查看当前用户的cron
任务:
crontab -l
如果设置成功就可以等待接收邮件了