python项目5-扫描当前目录和所有子目录并显示大小

import os
import sys

try:
    directory = sys.argv[1]  #Sys.argv[ ]其实就是一个列表，里边的项为用户输入的参数，关键就是要明白这参数是从程序外部输入的，而非代码本身的什么地方，要想看到它的效果就应该 将程序保存了，从外部来运行程序并给出参数
except IndexError:
    sys.exit("Must provide an argument.")

dir_size = 0
fsizedicr = {'Bytes': 1,
             'Kilobytes': float(1) / 1024,
             'Megabytes': float(1) / (1024 * 1024),
             'Gigabytes': float(1) / (1024 * 1024 * 1024)}
#os.walk()用于通过在目录树中游走输出在目录中的文件名，向上或者向下
for (path, dirs, files) in os.walk(directory):  # Walk through all the directories. For each iteration, os.walk returns the folders, subfolders and files in the dir.
    for file in files:  # Get all the files
        filename = os.path.join(path, file)
        dir_size += os.path.getsize(filename)  # 获取指定路径 path 下的文件的总大小，以字节（Byte）为单位

#round()返回浮点数x的四舍五入值
fsizeList = [str(round(fsizedicr[key] * dir_size, 2)) + " " + key for key in fsizedicr]

if dir_size == 0:
    print("File Empty")
else:
    for units in sorted(fsizeList)[::-1]:  # 对单位列表进行反向排序，以便于打印最小单位
        print("Folder Size: " + units)

在parameter中输入test可以得到test目录及其子目录的总共大小