首页 / 知识

关于windows：使用python的卷上剩余的跨平台空间

2023-04-14 16:24:00

Cross-platform space remaining on volume using python

我需要一种在 Linux、Windows 和 OS X 上使用 python 来确定磁盘卷上剩余空间的方法。我目前正在解析各种系统调用 (df, dir) 的输出以完成此操作 - 是否存在更好的方法？

1
2
3
4
5
6
7
8
9
10
11
12
13
14

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 安装 psutil。然后，您可以使用以下命令获取可用空间量(以字节为单位)：

1 2	import psutil print(psutil.disk_usage(".").free)

windows 可以使用 wmi 模块，unix 可以使用 os.statvfs

窗口

1
2
3
4
5

import wmi

c = wmi.WMI ()
for d in c.Win32_LogicalDisk():
print( d.Caption, d.FreeSpace, d.Size, d.DriveType)

适用于 unix 或 linux

1
2
3

from os import statvfs

statvfs(path)

如果您正在运行 python3：

使用 shutil.disk_usage() 和 os.path.realpath('/') 名称正则化有效：

1
2
3
4

from os import path
from shutil import disk_usage

print([i / 1000000 for i in disk_usage(path.realpath('/'))])

或

1
2
3
4
5

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)

给你总，使用，

如果你不想添加另一个依赖项，你可以为 windows 使用 ctypes 直接调用 win32 函数调用。

1
2
3
4
5
6
7
8

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 是一个命令行实用程序，因此是脚本编写所需的package器。
例如：

1
2
3
4
5
6
7

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

下面的代码在windows上返回正确的值

1
2
3
4
5

import win32file

def get_free_space(dirname):
secsPerClus, bytesPerSec, nFreeClus, totClus = win32file.GetDiskFreeSpace(dirname)
return secsPerClus * bytesPerSec * nFreeClus

os.statvfs() 函数是为类 Unix 平台(包括 OS X)获取该信息的更好方法。 Python 文档说"可用性：Unix"，但值得在您的 Python 构建中检查它是否也适用于 Windows(即文档可能不是最新的)。

否则，可以使用pywin32库直接调用GetDiskFreeSpaceEx函数。

我不知道有任何跨平台的方法来实现这一点，但也许对您来说一个好的解决方法是编写一个检查操作系统并为每个系统使用最佳方法的package类。

对于 Windows，在 win32 扩展中有 GetDiskFreeSpaceEx 方法。

空间跨平台磁盘方法