python基础：使用文件和目录操作，定义一个统计指定目录大小的函数

import os

def dirsiz(di):
    '''
    定义统计指定目录大小的函数
    '''
    sum=0
    #判断目录是否为文件
    if os.path.isfile(di):
        sum+=os.path.getsize(di)
    #判断指定目录是否为目录
    if os.path.isdir(di):
        dirLi=os.listdir(di)
        for i in dirLi:
            file=os.path.join(di,i)  #拼接目录下文件或者文件夹的地址
            if os.path.isdir(file):
                sum+=dirsiz(file)  #递归统计
            if os.path.isfile(file):
                sum+=os.path.getsize(file)
    return sum

    #判断循环，用来输入信息
while  True:
    print("{0:5}{1:5}".format('1、输入文件目录','2、退出'))
    key=input(">请输入对应的选择：")

    if key == "1":
        add_r=input("请输入目录或文件地址:")
        dirsiz(add_r)
        print("{}字节".format(dirsiz(add_r)))
        input("按回车键继续：")
    elif key == "2":
        print("已退出")
        break
    else :
        print("输入有误")