python实现Linux磁盘查看|清理

# 系统默认模块
# Create your tests here.

import os
import time


def get_device(path="/mnt/data"):
    import os

    day = 7
    disk = os.statvfs(path)

    disk_size = disk.f_bsize * disk.f_blocks / (1024 ** 3)  # 1G = 1024M  1M = 1024KB 1KB = 1024bytes
    print("磁盘总容量: %s" % format(disk_size, '.2f'))

    disk_used = disk.f_bsize * (disk.f_blocks - disk.f_bfree) / (1024 ** 3)
    print("磁盘已使用: %s" % format(disk_used, '.2f'))
    print("磁盘空闲: %s" % format((disk_size - disk_used), '.2f'))

    disk_used = format((disk_used / float(disk_size)), '.2f')
    print("磁盘使用率: %s" % disk_used)

    if float(disk_used) > 0.5:
        print("磁盘使用率已超过 %s" % disk_used)
        print("执行清理指令")
        cmd = 'sudo find /mnt/data/fastdfs/files/snap/ -mtime +%s -type d -name "*"' % day
        os.system(cmd)
    else:
        print("磁盘空间充足，还剩 %s" % (1 - float(disk_used)))


class DeviceInfo:
    """获取服务器磁盘信息"""
    def __init__(self, path, day):
        self.path = path
        self.day = day
        self.rate = 0.75

    @staticmethod
    def get_disk_info(path):
        """获取磁盘挂载信息"""
        return os.statvfs(path)

    def get_disk_size(self):
        """获取磁盘总容量"""
        disk = self.get_disk_info(self.path)
        disk_size = disk.f_bsize * disk.f_blocks / (1024 ** 3)  # 1G = 1024M  1M = 1024KB 1KB = 1024bytes
        print("磁盘总容量: %s" % format(disk_size, '.2f'))
        return disk_size

    def get_disk_used(self):
        """获取磁盘已使用容量|磁盘空闲容量"""
        disk = self.get_disk_info(self.path)
        disk_size = self.get_disk_size()
        disk_used = disk.f_bsize * (disk.f_blocks - disk.f_bfree) / (1024 ** 3)
        print("磁盘已使用: %s" % format(disk_used, '.2f'))
        print("磁盘空闲: %s" % format((disk_size - disk_used), '.2f'))
        return disk_size, disk_used, disk_size - disk_used

    def get_disk_rate(self):
        """返回磁盘使用率|剩余率"""
        disk_size, disk_used, _ = self.get_disk_used()
        print("磁盘使用率: %s" % format((disk_used / disk_size), '.2f'))
        print("磁盘空闲率: %s" % format((1 - (disk_used / disk_size)), '.2f'))
        return disk_used / disk_size, 1 - (disk_used / disk_size)

    def check(self):
        """磁盘清理主进程"""
        disk_rate, _ = self.get_disk_rate()  # 磁盘使用率
        if disk_rate > self.rate:  # 磁盘使用率大于默认配置(0.75)
            clear_cmd = 'sudo find /mnt/data/files/snap/ -mtime +%d -type d -name "*" -exec rm -rf {} \'
            while self.day:
                cmd = clear_cmd % self.day
                print(cmd)
                # os.system(cmd)
                disk_rate, _ = self.get_disk_rate()  # 磁盘使用率

                if disk_rate < self.rate:  # 使用率小于默认配置，break
                    break
        else:
            print('已超标')


if __name__ == '__main__':
    device = DeviceInfo(path='/Users/zhangjianping', day=7)
    device.check()

 还有一种基于psutil实现的方式，代码更为简洁

class ClearDisk4psUtil:
    """采用PSUtil模块进行磁盘清理"""

    def __init__(self, path="/mnt/data", day=180):
        self.path = path
        self.day = day
        self.rate = 70

    def get_disk_info(self):
        """获取磁盘信息"""
        total, used, free, percent = psutil.disk_usage(self.path)
        total = round(total / (1024 ** 3), 2)
        used = round(used / (1024 ** 3), 2)
        free = round(free / (1024 ** 3))

        print("磁盘总容量: ", total)
        print("磁盘已使用:", used)
        print("磁盘空闲:", free)
        print("磁盘使用百分比:", percent)
        return total, used, free, percent

    def clear_disk(self):
        """磁盘清理主程序"""
        _, _, _, percent = self.get_disk_info()
        clear_cmd = 'sudo find /mnt/data/fastdfs/files/snap/ -mtime +%d -type d -name "*" -exec rm -rf {} \'
        while percent > self.rate:  # 使用率大于默认配置，触发清理程序
            cmd = clear_cmd % self.day
            logger.info(cmd)
            # os.system(cmd)
            time.sleep(1)
            self.day -= 1
            _, _, _, percent = self.get_disk_info()
            if percent < self.rate:
                break
        else:
            logger.info("Disk health!")