|
# 系统默认模块# 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 |