[python] 统计一个目录下所有文件/文件夹的大小

import os
from tqdm import tqdm
from tabulate import tabulate

def getFlodSize(path:str)->float:
    """
    统计文件夹的大小
    path: 路径
    return: 文件/文件夹的大小，单位B
    """
    size = 0

    if os.path.isfile(path):
        return os.path.getsize(path)
    else:
        for file in os.listdir(path):
            size += getFlodSize(os.path.join(path,file))

    return size

def RechData(x:float, deci :int= 3)->str:
    """格式化数据
    x(float): data
    deci(int):精度
    """
    s = ['TB','GB','MB','KB','B'][::-1]
    flag = 0
    while True:
        if x>=1024:
            flag +=1
            x = x/1024.
        
        if flag>3 or x<1024:
            break
        
    return f'{round(x,deci):0>4.3f} {s[flag]}'

def getData(path:str)->list:
    """
    path(str):
    data(list):[[size1,fold1],[size2,fold2],...]
    """
    data = []
    for fold in tqdm(os.listdir('.')):
        data.append([getFlodSize(fold), fold])

    return data

root = '.'  # 目录地址
big2small = True  # 排序方式
deci = 3
data = getData(root)
data = sorted(data, key=lambda x: x[0], reverse=big2small)
bdata = [[RechData(i, deci), j] for i,j in data]
print(tabulate(bdata))