python 获取服务器的CPU、内存、磁盘和网络速率

精选原创

wx65139165783dd 2024-05-18 21:59:05 ©著作权

©著作权归作者所有：来自51CTO博客作者wx65139165783dd的原创作品，请联系作者获取转载授权，否则将追究法律责任

以下是一个获取服务器的CPU、内存、磁盘和网络速率，并将数据写入MySQL数据库的Python脚本示例：

import psutil
import speedtest
import mysql.connector
from datetime import datetime
import time

# 连接MySQL数据库
cnx = mysql.connector.connect(
    host="localhost",
    user="your_username",
    password="your_password",
    database="your_database"
)
cursor = cnx.cursor()

# 创建数据表
create_table_query = """
CREATE TABLE IF NOT EXISTS server_stats (
    id INT AUTO_INCREMENT PRIMARY KEY,
    cpu_percent FLOAT,
    memory_percent FLOAT,
    disk_percent FLOAT,
    download_speed FLOAT,
    upload_speed FLOAT,
    timestamp DATETIME
)
"""
cursor.execute(create_table_query)

while True:
    # 获取CPU信息
    cpu_percent = psutil.cpu_percent()

    # 获取内存信息
    memory = psutil.virtual_memory()
    memory_percent = memory.percent

    # 获取磁盘信息
    disk = psutil.disk_usage('/')
    disk_percent = disk.percent

    # 获取网络速率信息
    speed_test = speedtest.Speedtest()
    download_speed = speed_test.download() / 1024 / 1024
    upload_speed = speed_test.upload() / 1024 / 1024

    # 获取当前时间
    timestamp = datetime.now()

    # 将数据插入数据库
    insert_query = """
    INSERT INTO server_stats (cpu_percent, memory_percent, disk_percent, download_speed, upload_speed, timestamp)
    VALUES (%s, %s, %s, %s, %s, %s)
    """
    insert_values = (cpu_percent, memory_percent, disk_percent, download_speed, upload_speed, timestamp)
    cursor.execute(insert_query, insert_values)
    cnx.commit()

    time.sleep(60)  # 暂停60秒，每分钟获取一次数据

# 关闭数据库连接
cursor.close()
cnx.close()

上述代码使用了mysql.connector库来连接和操作MySQL数据库。在使用之前，请确保你已经安装了该库：