import xlrd
import xlwt
from xlutils.copy import copyclass XlsUtil():'''execl操作工具类'''def path_is_exist(self, path):'''判断是否存在工作簿:param path: 工作簿名称:return: 存在 返回True; 不存在 返回False'''try:xlrd.open_workbook(path, formatting_info=True)except FileNotFoundError as e:return Falsereturn Truedef sheet_name_is_exist(self, path, sheet_name):'''判断是否存在表名:param path: 工作簿:param sheet_name: 表名:return: 存在工作簿和表名 返回True； 否则 返回False'''path_exist = self.path_is_exist(path)if path_exist:workbook = xlrd.open_workbook(path) # 打开工作簿sheets = workbook.sheet_names()if sheet_name in sheets:return Truereturn Falsedef create_a_sheet(self, path, sheet_name, value):'''创建一张表:param path: 工作簿:param sheet_name: 表名:param value: 表头:return: 存在工作簿 存在表名 创建失败 返回False; 存在工作簿 不存在表名 创建成功 返回True； 不存在工作簿 创建成功 返回True'''if self.path_is_exist(path):workbook = xlrd.open_workbook(path, formatting_info=True)if self.sheet_name_is_exist(path, sheet_name):print(f'表名为[{sheet_name}]已经才在[{path}]中')return Falseelse:index = len(value)wb = copy(workbook)# add sheet to workbook with existing sheetssheet = wb.add_sheet(sheet_name)for i in range(0, index):for j in range(0, len(value[i])):sheet.write(i, j, value[i][j]) # 像表格中写入数据（对应的行和列）wb.save(path) # 保存工作簿print("xls格式表格写入数据成功！")return Trueelse:index = len(value) # 获取需要写入数据的行数workbook = xlwt.Workbook() # 新建一个工作簿sheet = workbook.add_sheet(sheet_name) # 在工作簿中新建一个表格for i in range(0, index):for j in range(0, len(value[i])):sheet.write(i, j, value[i][j]) # 像表格中写入数据（对应的行和列）workbook.save(path) # 保存工作簿print("xls格式表格写入数据成功！")return Truedef sheet_append_data(self, path, sheet_name, value):'''向一张表中追加数据:param path: 工作簿:param sheet_name: 表名:param value: 数据:return:'''if self.path_is_exist(path):if self.sheet_name_is_exist(path, sheet_name):index = len(value) # 获取需要写入数据的行数workbook = xlrd.open_workbook(path) # 打开工作簿sheets = workbook.sheet_names() # 获取工作簿中的所有表格worksheet = workbook.sheet_by_name(sheet_name) # 获取工作簿中所有表格中的的第一个表格rows_old = worksheet.nrows # 获取表格中已存在的数据的行数new_workbook = copy(workbook) # 将xlrd对象拷贝转化为xlwt对象new_worksheet = new_workbook.get_sheet(sheet_name) # 获取转化后工作簿中的第一个表格for i in range(0, index):for j in range(0, len(value[i])):new_worksheet.write(i + rows_old, j, value[i][j]) # 追加写入数据，注意是从i+rows_old行开始写入new_workbook.save(path) # 保存工作簿print(f"[{sheet_name}]表格【追加】写入数据成功！")else:print(f'[{sheet_name}]表不存在！')else:print(f'[{path}]工作簿不存在！')def sheet_read(self, path, sheet_name):'''输出指定表的数据:param path: 工作簿:param sheet_name:表:return:'''if self.path_is_exist(path):if self.sheet_name_is_exist(path, sheet_name):workbook = xlrd.open_workbook(path) # 打开工作簿# sheets = workbook.sheet_names() # 获取工作簿中的所有表格worksheet = workbook.sheet_by_name(sheet_name) # 获取工作簿中所有表格中的的第一个表格for i in range(0, worksheet.nrows):for j in range(0, worksheet.ncols):print(worksheet.cell_value(i, j), "\t", end="") # 逐行逐列读取数据print()else:print(f'[{sheet_name}]表不存在！')else:print(f'[{path}]工作簿不存在！')xls_util = XlsUtil()
book_name = '测试工作簿1.xls'sheet_name = '测试表3'value_title = [["姓名", "性别", "年龄", "城市", "职业"], ]value1 = [["张三", "男", "19", "杭州", "研发工程师"],["李四", "男", "22", "北京", "医生"],["王五", "女", "33", "珠海", "出租车司机"], ]value2 = [["Tom", "男", "21", "西安", "测试工程师"],["Jones", "女", "34", "上海", "产品经理"],["Cat", "女", "56", "上海", "教师"], ]def main():xls_util.create_a_sheet(book_name, sheet_name, value_title)xls_util.sheet_append_data(book_name, sheet_name, value1)xls_util.sheet_append_data(book_name, sheet_name, value2)xls_util.sheet_read(book_name, sheet_name)if __name__ == '__main__':main()