python计算文件夹大小——os.walk

下面提供一种利用os.walk()快捷的计算指定文件夹大小的方法

话不多说,直接上代码

简略版:

 1 import os
 2 
 3 
 4 def get_file_size(file_path, size=0):
 5     for root, dirs, files in os.walk(file_path):
 6         for f in files:
 7             size += os.path.getsize(os.path.join(root, f))
 8             # 加上下面一行打印所有文件
 9             # print(f)
10     return size
11 
12 print(get_file_size('E:\scoket_homework\db'))
13 print(get_file_size('.'))

专业版:

 1 import os
 2 
 3 
 4 def get_file_size(file_path, size=0):
 5     for root, dirs, files in os.walk(file_path):
 6         for f in files:
 7             size += os.path.getsize(os.path.join(root, f))
 8             # 加上下面一行打印所有文件
 9             # print(f)
10     return size
11 
12 
13 def main(file_path, size=0):
14     return get_file_size(file_path, size)
15 
16 
17 if __name__ == '__main__':
18     # 打印当前目录大小
19     print(main('.', 0))
20     # 如需测试可以将下面的路径换成您机器中的路径
21     path = r'E:\scoket_homework'
22     print(main(path, 0))
View Code

猜你喜欢

转载自www.cnblogs.com/huang-yc/p/9260174.html