如何检测闪存驱动器上是否有足够的存储空间来写入文件？

Question

我正在开发一个可以在闪存驱动器上生成无限文本文件的 Python 程序。该程序将是唯一在该闪存驱动器上运行的程序，我想在每次写入时检查是否有足够的存储空间。

如果有，我想将文件写入驱动器。如果没有足够的空间，我想对内容做点别的。例如：

def write_file(contents):
    if "Check if there is sufficient storage space on E:\ drive.":
        with open("filename", "w") as file:
            file.write(contents)
    else:
        # Alternative method for dealing with content.

我需要找到一个好方法来确定file.write() 操作将占用多少空间，并将其与驱动器上的可用空间进行比较。

谢谢！

Answer 1

这取决于平台；这是 Windows 的解决方案：

import ctypes
import platform

def get_free_space(dirname):
    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

if __name__ == "__main__":
    free_space = get_free_space("path\\")
    print(free_space)

如果您使用的是 Linux，我不确定，但我发现了这个：

from os import statvfs

st = statvfs("path/")
free_space = st.f_bavail * st.f_frsize / 1024

你的函数应该是这样的：

def write_file(contents):
    if free_space >= len(contents.encode("utf-8")):
        # Write to file.
        file = open("filename", "w")
        file.write(contents)
        file.close()
    else:
        # Alternative method for dealing with content.

Answer 2

可以按照here的说明获取磁盘信息：

import subprocess
df = subprocess.Popen(["df", "path/to/root/of/disk"], stdout=subprocess.PIPE)
output = df.communicate()[0]
device, size, used, available, percent, mountpoint = \
    output.split("\n")[1].split()

现在，使用used 和available 来确定磁盘是否有足够的空间。