Python小工具-复制嵌套目录下的多个word文档到指定目录

# 1.导入库
import os
import shutil
# 2.输入需要遍历的路径
file_path = r'F:\公司\20221124三峡大学培训\学号+姓名+实习报告\20201762+实习报告'
# 3.输入需要遍历的层级
dir_level = 3
# 4.输入需要找到的文件名后缀
file_types = [".doc",".docx"]
# 5.输入不需要读取的目录 包含这个目录就不读取 
filter_dirs = ['data','code'] 
# 6.把文件存放的位置 如果指定目录不存在，则新建文件的存放目录
dst_dir = r'F:\公司\20221124三峡大学培训\学号+姓名+实习报告\20201762+实习报告\汇总'
if not os.path.exists(dst_dir):
    os.mkdir(dst_dir)

# 定义函数 完成判断是指定格式的文档，并复制到指定的存放位置
def file_type_to_reduce_dir(filename,dst_dir,file_types,copy_or_remove="copy"):
    """
    filename当前文件的路径
    dst_dir要迁移到的路径
    file_types要过滤的文件类型
    copy_or_remove复制或移动
    """
    # os.path.split切分路径为路径和文件名 如 c:/ 和a.txt
    head_tail = os.path.split(filename)
    # 如果是复制模式
    if copy_or_remove =="copy":
        # 遍历需要复制的文件类型
        for file_type in file_types:
            # 如果文件类型是需要复制的
            if str(head_tail[1]).endswith(file_type):
                print(f"当前需要复制的文件名为{filename}")
                # 构建移动后的文件全路径
                file_path_new = dst_dir+'\\' + head_tail[1]
                # 复制文件到指定目录下
                shutil.copyfile(filename, file_path_new)
                print(f"复制：原文件{filename}--->新文件{file_path_new}")
    # 如果是移动模型            
    elif copy_or_remove=="remove":
         # 遍历需要复制的文件类型
        for file_type in file_types:
            print(f"当前需要移动的文件名为{filename}")
            # 如果文件类型是需要复制的
            if str(head_tail[1]).endswith(file_type):
                # 构建移动后的文件全路径
                file_path_new = dst_dir+'\\' + head_tail[1]
                # 复制文件到指定目录下
                # os.remove(file_path, file_path_new)
                print(f"移动：原文件{file_path}--->新文件{file_path_new}")
    else:
        print("copy_or_remove的值为copy 或是 remove")

# 开始遍历
# 查看初级目录下的所有文件以及目录
continue_flag = False
def list_dir_by_level(file_path,dst_dir,file_types,copy_or_remove="copy"):
    global continue_flag
    # 遍历当前路径
    dir_one = os.listdir(file_path)
    for file in dir_one:
        # 拼接绝对路径
        abs_file_name = os.path.join(file_path,file)
        # 如果路径为目标地址 跳过
        if abs_file_name == dst_dir:
            continue
        # 跳过 指定路径
        for filter_dir in filter_dirs:
            if filter_dir in abs_file_name:
                print(f"跳过包含{filter_dir}的目录-->{abs_file_name}")
                continue_flag = True # 设置外层循环的标记
                continue
        if continue_flag:
            continue_flag = False # 修改标记为False
            continue
        # if "data" in abs_file_name:
        #     # print(f"跳过包含data的目录-->{abs_file_name}")
        #     continue
        # 过滤提取初级目录下的子目录
        if os.path.isdir(abs_file_name):
            # print(f"当前目录为{abs_file_name}")
            list_dir_by_level(abs_file_name,dst_dir,file_types,copy_or_remove=copy_or_remove)
        elif os.path.isfile(abs_file_name):
            file_type_to_reduce_dir(abs_file_name,dst_dir,file_types,copy_or_remove=copy_or_remove)

list_dir_by_level(file_path,dst_dir,file_types)

'''
Descripttion: 
version: 
Author: sueRimn
Date: 2022-12-27 07:37:32
LastEditors: sueRimn
LastEditTime: 2022-12-27 20:25:36
'''
# 导入库
import os
import shutil
# 输入需要遍历的路径
file_path = r'F:\公司\20221124三峡大学培训\学号+姓名+实习报告\20201762+实习报告'
# 输入需要遍历的层级
dir_level = 3
# 输入需要找到的文件名后缀
file_types = [".doc",".docx"]
# 5.输入不需要读取的目录 包含这个目录就不读取 
filter_dirs = ['data','code'] 
# 把文件存放的位置 如果指定目录不存在，则新建文件的存放目录
dst_dir = r'F:\公司\20221124三峡大学培训\学号+姓名+实习报告\20201762+实习报告\汇总'
if not os.path.exists(dst_dir):
    os.mkdir(dst_dir)
# 定义函数 完成判断是指定格式的文档，并复制到指定的存放位置
def file_type_to_reduce_dir(filename,dst_dir,file_types,copy_or_remove="copy"):
    """
    filename当前文件的路径
    dst_dir要迁移到的路径
    file_types要过滤的文件类型
    copy_or_remove复制或移动
    """
    # os.path.split切分路径为路径和文件名 如 c:/ 和a.txt
    head_tail = os.path.split(filename)
    # 如果是复制模式
    if copy_or_remove =="copy":
        # 遍历需要复制的文件类型
        for file_type in file_types:
            # 如果文件类型是需要复制的
            if str(head_tail[1]).endswith(file_type):
                print(f"当前需要复制的文件名为{filename}")
                # 构建移动后的文件全路径
                file_path_new = dst_dir+'\\' + head_tail[1]
                # 复制文件到指定目录下
                shutil.copyfile(filename, file_path_new)
                print(f"复制：原文件{filename}--->新文件{file_path_new}")
    # 如果是移动模型            
    elif copy_or_remove=="remove":
         # 遍历需要复制的文件类型
        for file_type in file_types:
            print(f"当前需要移动的文件名为{filename}")
            # 如果文件类型是需要复制的
            if str(head_tail[1]).endswith(file_type):
                # 构建移动后的文件全路径
                file_path_new = dst_dir+'\\' + head_tail[1]
                # 复制文件到指定目录下
                # os.remove(file_path, file_path_new)
                print(f"移动：原文件{file_path}--->新文件{file_path_new}")
    else:
        print("copy_or_remove的值为copy 或是 remove")
# 开始遍历
# 查看初级目录下的所有文件以及目录
continue_flag = False
def list_dir_by_level(file_path,dst_dir,file_types,copy_or_remove="copy"):
    global continue_flag
    # 遍历当前路径
    dir_one = os.listdir(file_path)
    for file in dir_one:
        # 拼接绝对路径
        abs_file_name = os.path.join(file_path,file)
        # 如果路径为目标地址 跳过
        if abs_file_name == dst_dir:
            continue
        # 跳过 指定路径
        for filter_dir in filter_dirs:
            if filter_dir in abs_file_name:
                print(f"跳过包含{filter_dir}的目录-->{abs_file_name}")
                continue_flag = True # 设置外层循环的标记
                continue
        if continue_flag:
            continue_flag = False # 修改标记为False
            continue
        # if "data" in abs_file_name:
        #     # print(f"跳过包含data的目录-->{abs_file_name}")
        #     continue
        # 过滤提取初级目录下的子目录
        if os.path.isdir(abs_file_name):
            # print(f"当前目录为{abs_file_name}")
            list_dir_by_level(abs_file_name,dst_dir,file_types,copy_or_remove=copy_or_remove)
        elif os.path.isfile(abs_file_name):
            file_type_to_reduce_dir(abs_file_name,dst_dir,file_types,copy_or_remove=copy_or_remove)
list_dir_by_level(file_path,dst_dir,file_types)