Python计算文件夹的大小

Python计算文件夹的大小

#!/usr/bin/env python
# @Time    : 2018/3/18 21:56
# @Author  : KiritoLiu
# @Contact : kiritoliuyhsky@gmail.com
# @Site    : 
# @File    : 计算文件夹的大小.py
# @Software: PyCharm

import os
def gmkb(bites):        #换算成相应的单位便于查看
    if  bites >= 1024 :
        kb = bites/1024
        bb = bites%1024
        if kb >= 1024 :
            mb = kb/1024
            kb = kb%1024
            if mb >= 1024 :
                gb = mb/1024
                mb = mb%1024
            else:
                gb = 0
        else:
            mb = 0
    else:
        kb = 0
        bb = bites
    print("文件夹的大小为:{:.0f}Gb {:.0f}Mb {:.0f}kb {:.0f}b".format(gb, mb, kb, bb))      #取消小数点输出

def getFileSize(filePath, size=0):
    for root, dirs, files in os.walk(filePath):
        for f in files:
            size += os.path.getsize(os.path.join(root, f))
    gmkb(size)

getFileSize(".")