【python】深度优先搜索文件夹并移动全部doc文件

input文件下有：2文件夹，4文件夹，3.docx文件
2文件夹下有：7文件夹，5.txt文件，6.docx文件
7文件夹下有：10.docx文件，11.txt文件
4文件夹下有：8.docx文件，9.txt文件
和input同级目录下，有output文件夹

import os # 导入系统库os
import shutil # 实现移动文件的功能需要

# path代表待搜索的目录路径，result存储搜索到的文件路径列表
# 函数将path目录中的全部子目录和文件找到保存至result
def search_dir(path, result):
    # 使用os中的listdir得到path下目录和文件，保存到child_files
    child_files = os.listdir(path)
    for child in child_files: # 遍历child_files
        # 通过join函数拼接子目录或文件的路径，存储至child
        child = os.path.join(path, child)
        # 将child保存至result
        result.append(child)
        if os.path.isdir(child): # 如果child是一个子目录
            search_dir(child, result) # 调用search_dir继续递归搜索child

# 输入搜索目录和doc文件保存的目录
input_dir = input("输入待搜索的目录：")
output_dir = input("输出保存文件的目录：")

# 设置保存子目录和文件路径的列表files
files = list()

# 将input_dir中的全部子目录和文件路径找到并保存到files
search_dir(input_dir, files)

for file in files: # 遍历files
    print("find %s" %(file))
    # 如果该路径是一个docx文件
    if os.path.isfile(file) and file.endswith('.docx'):
        print("move %s" %(file)) # 打印提示信息
        shutil.move(file, output_dir) # 将文件移动到output_dir

输入待搜索的目录：./input
输出保存文件的目录：./output

运行程序后：
output文件夹下有：3.docx，6.docx，8.docx，10.docx。