获取指定目录下所有的doc & docx 文件名

写周月报时，经常用到需要统计一个文件夹下有n个子文件夹需要统计文件名的情况可以用到

import os
import glob
import openpyxl


def get_doc_files(path):
    """
    获取指定目录下所有doc文件的名称
    """
    doc_files = []
    for dirpath, dirnames, filenames in os.walk(path):
        for filename in filenames:
            if filename.endswith('.doc') or filename.endswith('.docx'):
                doc_files.append(os.path.join(dirpath, filename))
    return doc_files


def write_to_excel(data, filename):
    """
    将数据写入Excel文件
    """
    wb = openpyxl.Workbook()
    sheet = wb.active
    for row, item in enumerate(data, start=1):
        sheet.cell(row=row, column=1, value=item)
    wb.save(filename)


filedp='D:\\2023年10月周月报\\'
doc_files = get_doc_files(filedp)

# 将结果写入xlsx文件
write_to_excel(doc_files, '文件统计.xlsx')