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))