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