python读取指定目录下所有txt文件，并生成xls表目录

import os
import xlwt
import re

'''
1.读取指定目录下的所有文件，获取TXT文件，根据文件大小进行倒序排序，生成xls表
'''
def createNovelDirectory():
    path = r'D:\文件夹'
    fileMap = {}

    for dirpath, dirnames, filenames in os.walk(path):
        for filename in filenames:
            # 判断文件是否是txt文件
            if ".txt" not in filename:
                continue
            # 获取文件大小
            size = os.path.getsize(os.path.join(dirpath, filename))
            # 文件大小采用MB
            size = round(size / (1024 * 1024), 4)
            list = [filename, size, os.path.join(dirpath, filename)]
            fileMap.setdefault(filename, list)
    # 按文件大小进行倒序排序
    fileList = sorted(fileMap.items(), key=lambda d: d[1][1], reverse=True)
    work_book = xlwt.Workbook(encoding='utf-8')
    sheet = work_book.add_sheet('文件目录')
    for i in range(len(fileList)):
        if "《" in fileList[i][0]:
            patten = re.compile('《(.*?)》', re.S)
            lists = patten.findall(fileList[i][0])
            sheet.write(i, 0, str(lists[0])+".txt")
        else:
            sheet.write(i, 0, fileList[i][0])
        sheet.write(i, 1, str(fileList[i][1][1]) + "MB")
        sheet.write(i, 2, fileList[i][1][2])
    work_book.save("文件目录.xls")

createNovelDirectory()