Python實現按目錄層級輸出文件名並保存為excel

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

path = r'E:\py\python3.7\test'
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):
            file_dict[file_name] = find_dir_name(path+'\\'+file_name, file_name)
        else:
            pass
    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)
            # 如果value有值，則進行一次遍歷
            edit_excel(worksheet,value)
        else:
            # 若value值為空，則將key寫入sheet
            # 進行寫入excle
            col = 0
            
            for name in row_data:
                print('位置：%d，%d；值：%s' % (row_number,col,name))
                worksheet.write(row_number,col,name)
                col = col+1
                print('位置：%d，%d；值：%s' % (row_number,col,name))
            worksheet.write(row_number,col,key)
            row_number = row_number+1
    if row_data:
        row_data.pop()
    # time.sleep(1)
    return True

if __name__ == '__main__':
    # mm = str(find_dir_name(path,key))
    # f = open('file_name.txt', 'wb')
    # f.write(mm.encode('utf-8'))
    # f.close()
    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文件名稱.xls')