Python3 operation Execl tool class

import xlrd
import xlwt
from xlutils.copy import copy


class 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 False
        return True

    def 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 True
        return False

    def 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 False
            else:
                index = len(value)
                wb = copy(workbook)
                # add sheet to workbook with existing sheets
                sheet = 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 True
        else:
            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 True

    def 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()

 

Guess you like

Origin blog.csdn.net/someby/article/details/107727835