python定义一个统计指定文件夹大小的函数

题目：使用os模块，定义一个统计指定文件夹大小的函数，返回文件夹大小。

import os
def dirSize(dir):
    sum = 0
#判断是目录还是文件，
    if os.path.isfile(dir):
        sum+=os.path.getsize(dir)
    if os.path.isdir(dir):
        ldir = os.listdir(dir)
        #遍历目录
        for i in ldir:
            file = os.path.join(dir,i)
            if os.path.isdir(file):
                sum+= dirSize(file)
            if os.path.isfile(file):
                sum+=os.path.getsize(file)      
    return sum
a = dirSize("C:\\Users\\Desktop\\文件路径")   #输入统计大小的文件路径
print(a/1024/1024)  #转换为MB单位