def write_data2excel(filename, sheet_header, body_data):
"""将数据写入 Excel 文件中"""
# 新建 Excel 文件
workbook = xlwt.Workbook(encoding='UTF-8')
worksheet = workbook.add_sheet('Sheet1')
try:
# 写入数据
# 在第一行中写入表头
for col_num, cell_data in enumerate(sheet_header):
worksheet.write(0, col_num, cell_data)
# 在第二行中写入主体数据
for row_num, row_data in enumerate(body_data):
for col_num, cell_data in enumerate(row_data):
worksheet.write(row_num + 1, col_num, cell_data)
# 调整列宽
worksheet.col(0).width = 256 * 10
worksheet.col(1).width = 256 * 20
worksheet.col(2).width = 256 * 40
worksheet.col(3).width = 256 * 20
worksheet.col(4).width = 256 * 40
# 设置表头行高
worksheet.row(0).set_style(xlwt.easyxf('font:height 360;'))
workbook.save(filename) # 保存，filename为file的全路径
except Exception as e:
logger.error('将数据写入 Excel 文件失败，报错：%s', e)
raise Exception
def __release_src(self):
logger.info('程序即将结束，开始释放资源')
try:
self.workbook.close()
self.app.quit()
except Exception as e:
logger.error('空指针异常，报错：%s', e)
finally:
messagebox.showinfo(title='提示信息', message='Excel处理已结束')
def __open_excel(self):
"""打开 Excel 文件，获取要操作的 sheet 表单"""
logger.info('打开 Excel 源文件')
try:
# 设置为不显示地打开 Excel，读取 Excel 文件
self.workbook = self.app.books.open(self.input_file_path)
self.worksheet = self.workbook.sheets[0] # 选择第一个 sheet 表单
except Exception as e:
logger.error('打开源 Excel 文件失败，报错：%s', e)
raise Exception
def __select_group_by():
"""选择根据哪一列来进行分组"""
group_by = str(simpledialog.askstring(title='选择分组列', prompt='请选择根据哪列进行分组，如 E 或 e')).strip()
logger.info('用户选择了 %s 列来进行分组', group_by)
# 不能为空，且要合法
if not is_empty(group_by):
if re.match(r'^[a-zA-Z]$', group_by):
return ord(str.lower(group_by)) - 97 # 将 a-z 或 A-Z 装换为数字 0-25
logger.error('没有输入列或输入列的格式不正确')
messagebox.showwarning(title='提示信息', message='没有输入列或输入列的格式不正确')
raise Exception
def __select_area():
"""与用户交互，由用户选择要读取的 Excel 区域"""
selected_area = str(simpledialog.askstring(title='选择区域', prompt='请选择Excel读取区域，如 A4:E500 或 a4:e500')).strip()
logger.info('用户输入的 Excel 区域为：%s', selected_area)
# 选择的区域不能为空，且要合法
if not is_empty(selected_area):
if re.match(r'^[a-zA-Z]{1,2}\d+:[a-zA-Z]{1,2}\d+$', selected_area):
return selected_area.upper() # 将 a5:e6000 转换为 A5:E6000
logger.error('没有选择区域或输入的区域格式不正确')
messagebox.showwarning(title='提示信息', message='没有选择区域或输入的区域格式不正确')
raise Exception
def __make_root_folder(self):
"""创建存放处理后得到的 Excel 文件的根目录"""
logger.info('开始创建存放导出的 Excel 的根目录')
# 创建存放的根目录，目录名为“日期+时间”
root_folder = time.strftime('%Y年%m月%d日%H%M%S', time.localtime(time.time()))
self.root_folder_path = os.path.join(self.input_folder_path, root_folder) # 根目录的绝对路径
try:
os.mkdir(self.root_folder_path)
except Exception as e:
logger.error('创建根目录失败，报错：%s', e)
raise Exception
return self.root_folder_path
def __read_excel_data(self, selected_area, group_by):
"""根据传入的 Excel 区域，从 Excel 文件中读取数据，并分组存放到一个字典中"""
logger.info('从源 Excel 文件中读取数据-开始')
try:
data = self.worksheet.range(selected_area).value
self.sheet_header = data[0]
for col_cell_list in data[1:]:
group_name = str(col_cell_list[group_by]).strip() # 剔除字符串两边空格
if is_empty(group_name):
continue
# 如果不为空，来对其进行切片处理，取前 5 个字符作为组名
# group_name = group_name[:5]
if group_name in self.groups_dict:
self.groups_dict.get(group_name).append(col_cell_list)
else:
self.groups_dict[group_name] = [col_cell_list]
except Exception as e:
logger.error('读取 Excel 数据失败，原因：%s', e)
raise Exception