import os


# 目錄大小統計
def size(file):
    # 判斷是否存在
    if not os.path.exists(file):
        print(file, '檔案不存在，無法統計')
        return None
    # 是普通檔案
    if os.path.isfile(file):
        return os.path.getsize(file)
    # 是目錄，遞迴統計裡面的檔案，最終得出目錄的大小
    total = 0
    dirs = os.listdir(file)
    for f in dirs:
        file_name = os.path.join(file, f)
        if os.path.isfile(file_name):
            total += os.path.getsize(file_name)
        else:
            total += size(file_name)
    return total


print(size('test'))