Python 3.x基础实战检查磁盘可用空间

目录

• 引言
• 使用psutil模块实现
• 获取磁盘使用情况
• 获取磁盘挂载点
• 检查磁盘可用空间
• 使用 du 命令实现
• 使用 du 命令检查磁盘空间
• 检查多个目录的磁盘空间
• 检查磁盘空间使用率
• 发送邮件
• 结论

pip3 install psutil

import psutil
# 获取磁盘使用情况
disk_usage = psutil.disk_usage('/')
# 打印磁盘使用情况
print(f"Total: {disk_usage.total / (1024*1024*1024):.2f} GB")
print(f"Used: {disk_usage.used / (1024*1024*1024):.2f} GB")
print(f"Free: {disk_usage.free / (1024*1024*1024):.2f} GB")

import psutil
# 获取磁盘挂载点
disk_partitions = psutil.disk_partitions()
# 打印磁盘挂载点
for partition in disk_partitions:
    print(f"Device: {partition.device}")
    print(f"Mountpoint: {partition.mountpoint}")
    print(f"File system type: {partition.fstype}")
    print()

import psutil
import os
# 获取磁盘挂载点
disk_partitions = psutil.disk_partitions()
# 遍历每个挂载点
for partition in disk_partitions:
    # 获取磁盘使用情况
    disk_usage = psutil.disk_usage(partition.mountpoint)
    # 计算磁盘可用空间的百分比
    free_percent = disk_usage.free / disk_usage.total * 100
    # 如果磁盘可用空间小于10%，发送警告邮件
    if free_percent < 10:
        # 获取主机名
        hostname = os.uname()[1]
        # 构造邮件内容
        subject = f"Disk space warning on {hostname}"
        message = f"The disk {partition.device} ({partition.mountpoint}) is running out of space ({free_percent:.2f}% free)."
        # 发送邮件
        send_email(subject, message)

import subprocess
def get_directory_size(path):
    """Return the total size of the files in the given directory and subdirectories."""
    cmd = ["du", "-sh", path]
    result = subprocess.run(cmd, stdout=subprocess.PIPE)
    output = result.stdout.decode("utf-8").strip()
    size = output.split()[0]
    return size
# Example usage:
size = get_directory_size("/var/www/html")
print("Size of /var/www/html: {}".format(size))

import subprocess
def get_directory_size(path):
    """Return the total size of the files in the given directory and subdirectories."""
    cmd = ["du", "-sh", path]
    result = subprocess.run(cmd, stdout=subprocess.PIPE)
    output = result.stdout.decode("utf-8").strip()
    size = output.split()[0]
    return size
# List of directories to check
directories = ["/var/www/html", "/var/log", "/etc"]
# Loop through directories and print their sizes
for directory in directories:
    size = get_directory_size(directory)
    print("Size of {}: {}".format(directory, size))

import subprocess
def get_disk_usage():
    """Return the disk usage of the root filesystem in percent."""
    cmd = ["df", "-h", "/"]
    result = subprocess.run(cmd, stdout=subprocess.PIPE)
    output = result.stdout.decode("utf-8").strip()
    usage = int(output.split("\n")[1].split()[4].replace("%", ""))
    return usage
# Example usage:
usage = get_disk_usage()
print("Disk usage: {}%".format(usage))

import smtplib
from email.mime.text import MIMEText
from email.header import Header
def send_email(subject, message):
    # 邮件发送者和接收者
    sender = 'sender@example.com'
    receiver = 'receiver@example.com'
    # 邮件主题和内容
    msg = MIMEText(message, 'plain', 'utf-8')
    msg['Subject'] = Header(subject, 'utf-8')
    # 发送邮件
    smtp = smtplib.SMTP('smtp.example.com')
    smtp.login(sender, 'password')
    smtp.sendmail(sender, receiver, msg.as_string())
    smtp.quit()