Python 快速获取文件夹的总大小

在本文中，我们将介绍如何使用Python快速获取文件夹的总大小。无论是在操作系统中清理磁盘空间，还是在开发中需要统计文件夹大小，这些技巧都会很有用。

使用os模块递归遍历文件夹

Python的os模块提供了许多文件和文件夹操作的函数，其中就包括了获取文件夹大小的方法。我们可以利用这些方法来递归地遍历文件夹，并累计文件的大小。

下面是一个示例代码，展示了如何使用os模块递归遍历文件夹并获取总大小：

import os

def get_folder_size(folder_path):
    total_size = 0
    for path, dirs, files in os.walk(folder_path):
        for file in files:
            file_path = os.path.join(path, file)
            total_size += os.path.getsize(file_path)
    return total_size

folder_path = "/path/to/folder"
folder_size = get_folder_size(folder_path)
print(f"The total size of the folder is: {folder_size} bytes")

在这个示例中，我们定义了一个名为get_folder_size的函数，它接受一个文件夹路径作为参数，并返回该文件夹的总大小。函数内部使用了os.walk函数来递归遍历文件夹，并通过os.path.getsize函数获取文件大小。

使用du命令的子进程

除了使用Python自带的模块来处理文件夹大小外，我们还可以利用操作系统的命令行工具来获取文件夹大小。在Unix系统中，可以使用du命令来计算文件夹的大小。在Python中，我们可以使用subprocess模块来执行命令行命令并获取输出。

下面是一个示例代码，展示了如何使用du命令的子进程来获取文件夹的总大小：

import subprocess

def get_folder_size(folder_path):
    command = ["du", "-s", folder_path]
    result = subprocess.run(command, capture_output=True, text=True)
    output = result.stdout.strip().split("\t")
    total_size = int(output[0]) * 1024   # 转换为字节
    return total_size

folder_path = "/path/to/folder"
folder_size = get_folder_size(folder_path)
print(f"The total size of the folder is: {folder_size} bytes")

在这个示例中，我们定义了一个名为get_folder_size的函数，它接受一个文件夹路径作为参数，并返回该文件夹的总大小。函数内部使用了subprocess.run函数执行du -s命令并捕获输出。我们通过拆分输出并将其转换为字节来获取文件夹的大小。

递归遍历文件夹 vs. du命令

使用递归遍历文件夹和使用du命令的子进程，各有优点和缺点。下面是它们的对比：

递归遍历文件夹：
- 优点：
- 不依赖于操作系统的特定命令，可在任何平台上运行。
- 可以自定义处理逻辑，比如忽略特定类型的文件或文件夹。
- 缺点：
- 遍历大型文件夹时，性能可能较低。
- 需要自行编写代码来实现递归遍历和文件大小的累加。
du命令的子进程：
- 优点：
- 在Unix系统上性能较好，特别是对于大型文件夹。
- 不需要自己编写递归遍历的代码。
- 缺点：
- 依赖于特定的命令行工具，不适用于所有操作系统。
- 难以自定义处理逻辑，只能获取文件夹的总大小。