计算目录大小

Posted 2020-12-11 lihuafeng

import os,os.path

directory = input(‘请输入目录或者文件地址：‘)

def getSzie(path):
global size
size = 0
if os.path.isfile(path):
size1 = os.path.getsize(path)
size += size1

if os.path.isdir(path):
    for file in os.listdir(path):
        #全路径
        filename = os.path.join(path, file)
        if os.path.isfile(filename):
            size2 = os.path.getsize(filename)
            size += size2
        elif os.path.isdir(filename):
            #递归方法调用
            #最后的大小一定要记得加上之前文件的大小
            size += getSzie(filename)

return size

size = getSzie(directory)
if size <= 1024:
print(‘目录大小为:{}字节‘.format(size))
elif size > 1024 and size<10241024:
print(‘目录大小为:{:.2f}KB‘.format(size/1024))
else:
print(‘目录大小为:{:.2f}MB‘.format(size/(10241024)))