 使用xlsxwriter库import os
import xlwt
import openpyxl
import time
import xlsxwriter #使用这个库创建excel
def search_file(start_dir, target) :
os.chdir(start_dir)
for each_file in os.listdir(os.curdir) :
ext = os.path.splitext(each_file)[1]
file_name = os.path.splitext(each_file)[0]
if ext in target :
file_dict[file_name] = os.getcwd() + os.sep + each_file + os.linesep # 使用os.sep是程序更标准
if os.path.isdir(each_file) :
search_file(each_file, target) # 递归调用
os.chdir(os.pardir) # 递归调用后切记返回上一层目录
#start_dir = input('请输入待查找的初始目录:')
start_dir = 'D:\python代码'
#orig_dir = os.getcwd()
target = ['.txt']
file_dict = {}
#写入excel表
line = 1
row = 0
timest = '文件台账'+str(time.localtime().tm_min)+str(time.localtime().tm_sec)+'.xlsx'
workbook = xlsxwriter.Workbook(timest)
worksheet = workbook.add_worksheet('文件清单')
#设置第一列第二列的宽度
worksheet.set_column('A:A', 20)
worksheet.set_column('B:B', 50)
search_file(start_dir, target)
#设置excel表第一行标题格式
style = workbook.add_format({'bold': True}) #加粗
worksheet.write(0,0,'文件名',style)
worksheet.write_url(0,1,'链接',style)
#从第1行开始,向excel表中写入文件名及其路径
for each in file_dict.keys():
worksheet.write(line,0,each)
worksheet.write_url(line,1,file_dict[each].strip())
line+=1
workbook.close()
|