说说如何使用 Python 获取文件夹中的文件列表和计算文件或文件夹大小

os.path 模块提供了一些函数，获取文件夹中的文件列表以及文件大小：

• os.path.getsize(path) 会返回 path 文件的字节数。
• os.listdir(path) 会返回 path 所包含的文件或文件名列表。注意： 这个方法直接挂在 os 模块底下哦O(∩_∩)O

folder = os.path.join('c:', 'Program Files', 'mat')
file = os.path.join(folder, 'MemoryAnalyzer.exe')

print(str(os.path.getsize(file) / 1024) + 'KB')
print(os.listdir(folder))

运行结果：

407.5KB
[‘configuration’, ‘eclipsec.exe’, ‘epl-v10.html’, ‘features’, ‘MemoryAnalyzer.exe’, ‘MemoryAnalyzer.ini’, ‘notice.html’, ‘p2’, ‘ParseHeapDump.bat’, ‘plugins’, ‘workspace’]

综合使用 os.path.getsize() 和 os.listdir()，就可以获取文件夹中所有文件总字节数：

total_size = 0
for file_name in os.listdir(folder):
    total_size = total_size + os.path.getsize(os.path.join(folder, file_name))
print(str(total_size/1024) + 'KB')

运行结果：

644.81640625KB

如果需要计算包括子孙文件夹中的所有文件大小，就必须定义一个函数，递归实现：

def calculate_total_size(folder_path):
    '''
    计算当前文件夹下所有文件大小
    :param folder_path:  文件夹路径
    :return:
    '''
    total_size = 0
    for file_name in os.listdir(folder_path):
        path = os.path.join(folder_path, file_name)
        if os.path.isdir(path):  # 递归调用，计算子孙文件夹中所包含的文件总大小
            total_size = total_size + calculate_total_size(path)
        if os.path.isfile(path):  # 直接计算
            total_size = total_size + os.path.getsize(path)
    return total_size


# 计算当前文件夹下所有文件大小（包含子孙文件夹中的所有文件）
total_size = calculate_total_size(folder)
print(str(total_size / 1024) + 'KB')

运行结果：

113761.3232421875KB