import os

empty_dir_number = 0
file_type_number = {}  # 默认这个字典里没有值


def check_empty_dir(dirs, files):
    global empty_dir_number
    if len(dirs) == len(files) == 0:
        empty_dir_number += 1


# def check_file_type(files):
#     for file in files:
#         file_names = file.split(".")
#         if len(file_names) > 1:
#             file_type = file_names[-1]
#             number = file_type_number.get(file_type, 0)
#             number += 1
#             file_type_number[file_type] = number
#     else:
#         pass
def check_file_type(files):
    for file in files:
        file_names = file.split(".")
        if len(file_names) > 1:
            file_type = file_names[-1]
            if file_type in target_show_type:
                number = file_type_number.get(file_type, 0)
                number += 1
                file_type_number[file_type] = number


def main(target_dir):
    for path, dirs, files in os.walk(target_dir):
        check_empty_dir(dirs, files)


if __name__ == "__main__":
    target_dir = "C:/"

    main(target_dir)
    print("空文件夹数量是{}个".format(empty_dir_number))
    print("统计结果", file_type_number)
    target_show_type = ["pdf", "mp3", "mp4", "png", "jpeg", "xls", "xlsx", "doc", "docs", "ppt"]
    for t in target_show_type:
        print("{}:{}".format(t, file_type_number.get(t, 0)))