python统计文件夹下不同文件的数量和大小

import os

wjzd={}
wjsize={}
lj="wjcl/src/test"

def get_size_type(f_path):
    global lj
    files_name = os.listdir(f_path)
    for name in files_name:
        file_path = os.path.join(f_path, name)
        ty = str(os.path.splitext(file_path)[1]).strip('.')

        #print(type(ty))
        if os.path.isdir(file_path):
            get_size_type(file_path)
        if not ty or ty == 'gitkeep':
            continue
        else:
            wjzd.setdefault(ty, 0)
            wjzd[ty] += 1
            wjsize.setdefault(ty, 0)
            wjsize[ty] += os.path.getsize(file_path)

get_size_type(lj)

for x in ['txt', 'mp3', 'jpg', 'pptx']:
    print("文件类型{}文件数{}文件大小{:.2f}KB".format(x, wjzd[x], wjsize[x] / 1024))