python 计算文件夹里所有内容的大小总和

计算文件夹里所有内容的大小总和

递归方法

'''计算文件夹的大小'''
import os
def dir_file_size(path):
    if os.path.isdir(path):
        file_size = 0
        dir_list = os.listdir(path)
        for dir_name in dir_list:
            file_path = os.path.join(path,dir_name)
            if os.path.isfile(dir_name):
                file_size += os.path.getsize(file_path)
            else:
                ret = dir_file_size(file_path)
                file_size += ret
        return file_size
    elif os.path.isfile(path):
        return os.path.getsize(path)
    else:print('找不到%s文件'%path)

path = r'D:\oldbaoy_python_project\day04'
ret = dir_file_size(path)
print('{0} 的大小为 {1}字节'.format(path,ret))

栈方法

import os

def dir_file_size(path):
    if os.path.isdir(path):
        file_size, dir_list = 0, [path]
        while dir_list:
            path = dir_list.pop()
            dirs = os.listdir(path)
            for name in dirs:
                file_path = os.path.join(path, name)
                if os.path.isfile(file_path):
                    file_size += os.path.getsize(file_path)
                else:
                    dir_list.append(file_path)
        return file_size
    elif os.path.isfile(path):
        return os.path.getsize(path)
    else:
        print('找不到%s文件' % path)

path = r'D:\oldbaoy_python_project\day04'
ret = dir_file_size(path)
print('{0} 的大小为 {1}字节'.format(path, ret))