import xlwt
import os
class TxtToExcel(object):
def __init__(self, path):
self.path = path
self.workbook = xlwt.Workbook()
self.worksheet = self.workbook.add_sheet('mybook', cell_overwrite_ok=True)
def read_content(self, file):
with open(file, 'r', encoding='utf-8') as f:
return f.readlines()
def file_list(self):
return os.listdir(self.path)
def write_to_excel(self, row, content):
content_list = content.split(" ")
for con in content_list:
self.worksheet.write(row, content_list.index(con), con)
# self.workbook.save('a.xls')
def get_file_path_list(self):
file_path_list = [os.path.join(self.path, file) for file in self.file_list()]
# content = []
# for file in file_path_list:
# content.extend(self.read_content(file))
gen = (self.read_content(file) for file in file_path_list)
index = 0
for g in gen:
for i in g:
self.write_to_excel(index + 1, i)
index += 1
self.workbook.save('a.xls')
# for c in content:
# self.write_to_excel(content.index(c)+1, c)
if __name__ == '__main__':
path = r"E:\qqq\test_excel"
TxtToExcel(path).get_file_path_list()
|