我需要一种在 Linux、Windows 和 OS X 上使用 python 来确定磁盘卷上剩余空间的方法。我目前正在解析各种系统调用(df、dir)的输出来实现这一点——有更好的方法吗?
11 回答
import ctypes
import os
import platform
import sys
def get_free_space_mb(dirname):
"""Return folder/drive free space (in megabytes)."""
if platform.system() == 'Windows':
free_bytes = ctypes.c_ulonglong(0)
ctypes.windll.kernel32.GetDiskFreeSpaceExW(ctypes.c_wchar_p(dirname), None, None, ctypes.pointer(free_bytes))
return free_bytes.value / 1024 / 1024
else:
st = os.statvfs(dirname)
return st.f_bavail * st.f_frsize / 1024 / 1024
请注意,您必须传递目录名称GetDiskFreeSpaceEx()
才能工作(statvfs()
适用于文件和目录)。您可以从带有os.path.dirname()
.
另请参阅 和 的os.statvfs()
文档GetDiskFreeSpaceEx
。
使用. _ _ pip install psutil
然后,您可以使用以下方法获取可用空间量(以字节为单位):
import psutil
print(psutil.disk_usage(".").free)
您可以将wmi模块用于 windows,将 os.statvfs 用于 unix
用于窗户
import wmi
c = wmi.WMI ()
for d in c.Win32_LogicalDisk():
print( d.Caption, d.FreeSpace, d.Size, d.DriveType)
对于 unix 或 linux
from os import statvfs
statvfs(path)
如果您正在运行 python3:
使用名称正则shutil.disk_usage()
化os.path.realpath('/')
作品:
from os import path
from shutil import disk_usage
print([i / 1000000 for i in disk_usage(path.realpath('/'))])
或者
total_bytes, used_bytes, free_bytes = disk_usage(path.realpath('D:\\Users\\phannypack'))
print(total_bytes / 1000000) # for Mb
print(used_bytes / 1000000)
print(free_bytes / 1000000)
为您提供以 MB 为单位的总空间、已用空间和可用空间。
如果您不喜欢添加另一个依赖项,您可以为 windows 使用 ctypes 直接调用 win32 函数调用。
import ctypes
free_bytes = ctypes.c_ulonglong(0)
ctypes.windll.kernel32.GetDiskFreeSpaceExW(ctypes.c_wchar_p(u'c:\\'), None, None, ctypes.pointer(free_bytes))
if free_bytes.value == 0:
print 'dont panic'
从 Python 3.3 开始,您可以使用来自 Windows 和 UNIX 标准库的shutil.disk_usage("/").free :)
一个很好的跨平台方式是使用 psutil:http ://pythonhosted.org/psutil/#disks (请注意,您需要 psutil 0.3.0 或更高版本)。
您可以使用df作为跨平台的方式。它是GNU 核心实用程序的一部分。这些是预期存在于每个操作系统上的核心实用程序。但是,默认情况下它们没有安装在 Windows 上(这里,GetGnuWin32派上用场了)。
df是一个命令行实用程序,因此是脚本编写所需的包装器。例如:
from subprocess import PIPE, Popen
def free_volume(filename):
"""Find amount of disk space available to the current user (in bytes)
on the file system containing filename."""
stats = Popen(["df", "-Pk", filename], stdout=PIPE).communicate()[0]
return int(stats.splitlines()[1].split()[3]) * 1024
os.statvfs ()函数是为类 Unix 平台(包括 OS X)获取该信息的更好方法。Python 文档说“可用性:Unix”,但值得检查它是否在您的 Python 版本中也适用于 Windows(即文档可能不是最新的)。
否则,您可以使用pywin32库直接调用GetDiskFreeSpaceEx函数。
下面的代码在 Windows 上返回正确的值
import win32file
def get_free_space(dirname):
secsPerClus, bytesPerSec, nFreeClus, totClus = win32file.GetDiskFreeSpace(dirname)
return secsPerClus * bytesPerSec * nFreeClus
我不知道任何跨平台的方法来实现这一点,但也许对您来说一个好的解决方法是编写一个检查操作系统并为每个系统使用最佳方法的包装类。
对于 Windows,win32 扩展中有GetDiskFreeSpaceEx方法。