python - 遍历文件夹解压zip文件

# 
import os
import zipfile

def unzip_file(zip_src, dst_dir):
    r = zipfile.is_zipfile(zip_src)
    if r:     
        fz = zipfile.ZipFile(zip_src, 'r')
        for file in fz.namelist():
            fz.extract(file, dst_dir)       
    else:
        print('This is not zip')

def func():
    import os 
    for root, dirs, files in os.walk('hahaha'): 
        # 遍历dirs
        for dir in dirs:
            # 遍历现在的路径下的文件
            temp_dir = root+'/'+dir
            for _, _, files in os.walk(temp_dir): 
                for f in files:
                    # print(temp_dir+'/'+f)
                    # 解压这个zip文件
                    unzip_file(temp_dir+'/'+f, temp_dir)

def delete_zip():
    import os 
    for root, dirs, files in os.walk('hahaha'): 
        # 遍历dirs
        for dir in dirs:
            # 遍历现在的路径下的文件
            temp_dir = root+'/'+dir
            for _, _, files in os.walk(temp_dir): 
                for f in files:
                    # 删除这个zip文件
                    r = zipfile.is_zipfile(temp_dir+'/'+f)
                    if r:
                        os.remove(temp_dir+'/'+f)

func()
delete_zip()