python复制指定文件夹下指定格式的文件（保留文件夹目录）

import os
import shutil

# 要复制的目录
file_path = r'E:/@_projects/EMIS.WEB/src'
# 指定目标目录
new_path = r'D:/download/png'
# 文件格式
format_name = '.png'


# 判断文件夹是否存在,不存在创建
def create_folder(folder_name):
    folder_name = folder_name.replace('\\', '/')
    folder_name = folder_name.replace(file_path, new_path)
    if os.path.exists(folder_name):
        return
    folder_list = folder_name.split('/')
    index = 2
    for item in folder_list:
        path = '/'.join(folder_list[0:index])
        if not os.path.exists(path):
            print('不存在目录:' + path + ',对其进行创建')
            os.mkdir(path)
        index += 1


# 复制文件夹下指定文件格式的文件(保留文件夹格式)
def copy_file():
    for root, dirs, files in os.walk(file_path):
        for file_name in files:
            if file_name.endswith(format_name):
                create_folder(root)
                new_file_path = root.replace('\\', '/').replace(file_path, new_path)
                shutil.copyfile(os.path.join(root, file_name), os.path.join(new_file_path, file_name))
                print(file_name + '复制成功')


if __name__ == '__main__':
    copy_file()