打印子目录到Excel中

# encoding=utf-8
import os
import xlwt
from xlwt.Worksheet import Worksheet
import time
from openpyxl import load_workbook

path = r'E:\资源\处理\全科资料\小学数学资料'
file_dicts = {}

# 将目录名、文件名读取为字典数据
# def find_dir_name(path,key=''):
#     file_names = os.listdir(path)
#     file_dict = {}
#     for file_name in file_names:
#         # print('当前目录：%s' % path+'\\'+file_name)
#         file_dict[file_name] = {}
#         if os.path.isdir(path+'\\'+file_name):
#             print(path+'\\'+file_name)
#             file_dict[file_name] = find_dir_name(path+'\\'+file_name, file_name)
#         else:
#             pass
#     return file_dict
def find_dir_name(path, key=''):  
    file_names = os.listdir(path)  
    file_dict = {}  
    for file_name in file_names:  
        # 构建完整的文件/文件夹路径  
        full_path = os.path.join(path, file_name)  
          
        # 如果是文件夹，则递归调用此函数  
        if os.path.isdir(full_path):  
            print(full_path)  
            file_dict[file_name] = find_dir_name(full_path, file_name)  
        # 如果是文件，则不做任何操作  
        else:  
            continue  
    return file_dict



# 将字典数据存入Excel
def edit_excel(worksheet,data):
    global row_number, row_data
    for key,value in data.items():
        if value:
            row_data.append(key)
            edit_excel(worksheet,value)
        else:
            col = 0
            for name in row_data:
                worksheet.write(row_number,col,name)
                col = col+1
            worksheet.write(row_number,col,key)
            row_number = row_number+1
    if row_data:
        row_data.pop()
    return True


      
#     return True
if __name__ == '__main__':
    data = find_dir_name(path)
    # 创建工作表
    workbook = xlwt.Workbook(encoding='utf-8')
    worksheet = workbook.add_sheet('sheet1',cell_overwrite_ok=True)
    # 临时记录一行数据
    row_data = []
    row_number = 0
    edit_excel(worksheet,data)
    workbook.save('转化后的excel文件名称.xlsx')