【Python】使用 openpyxl 读取Excel中的测试数据

ゞ浴缸里的玫瑰 2023-07-12 11:47 33阅读 0赞

### 文章目录 ###

*   *   *  1. 安装
         *  2. openpyxl 核心三大对象
         *  3.代码示例
         *  一个exceltool类的封装

### 1. 安装 ###

使用 `pip install openpyxl`进行安装，只要不报红就是安装成功啦。  
这里可能会因为文件比较大下载不了而报错，使用国内镜像就好啦。语句如下：`pip install 包名 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com`

### 2. openpyxl 核心三大对象 ###

在使用openpyxl前先要掌握三个对象，即：

*  Workbook(工作簿，一个包含多个Sheet的Excel文件)、
 *  Worksheet（工作表，一个Workbook有多个Worksheet，表名识别，如“Sheet1”,“Sheet2”等）
 *  Cell（单元格，存储具体的数据对象）三个对象。

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70]

### 3.代码示例 ###

先示范一下基本的方法：

import openpyxl
    
    file_path = r'E:\Python_selenium_advance\homework\day15_ddt\XX项目测试用例表.xlsx'
    # 将指定的 .xlsx 文件加载为 WorkBook 对象
    wb = openpyxl.load_workbook(file_path, False)
    # 读取该工作簿中对应 sheet 中的数据
    login_sh = wb['Login']
    # 读取指定格子中的信息
    print(login_sh.cell(row = 1,column = 2).value)
    # 在指定格子中写入信息，写入数据后必须要保存这个工作簿才能够看到
    login_sh.cell(row = 3,column = 2,value = '新的数据')
    # 将这个 WorkBook 文件保存为文件，注意，该文件在占用状态下将不会被保存成功。
    wb.save(file_path)
    # 获取工作表的最大行数
    print(login_sh.max_row)
    # 获取工作表的最大列数
    print(login_sh.max_column)
    # rows:按行获取工作表中所有格子对象，每行内容放在一个元组里面
    res1 = list(login_sh.rows)
    for every_row in res1:
        for every_cell in every_row:
            print(every_cell.value,end = '|')
        print('')
    # columns:按列获取工作表中所有格子对象，每列内容放在一个元组里面
    # res2 = list(login_sh.columns)
    # for every_col in res2:
    #     for every_cell in every_col:
    #         print(every_cell.value,end='|')
    #     print()

### 一个exceltool类的封装 ###

import openpyxl
    
    '''
    对excel进行操作的工具，封装了读取excel中值和写入值的操作
    '''
    class ExcelTool(object):
    
        def __init__(self, path, sheet):
            self.path = path
            self.sheet = sheet
    
        def read_data(self):
            '''
            从给定的excel表和对应的sheet中读取出所有的数据
            :return: 每一行数据与表头聚合存储为字典，以列表形式存储每一行的字典数据
            '''
            wb = openpyxl.load_workbook(self.path)
            sh = wb[self.sheet]
            rows = list(sh.rows)
            cases = []
            # 取出表头作为每一个用例的标题
            title = []
            for row in rows[0]:
                title.append(row.value)
            # 从第二行开始的数据作为测试用例的数据,和标题聚合在一起转化为字典
            for row in rows[1:]:
                case = []
                for item in row:
                    case.append(item.value)
                cases.append(dict(zip(title, case)))
    
            return cases
    
        def write_data(self, row, col, str):
            '''
            在给定的excel表和对应的sheet中的指定位置写入数据
            :param row: 写入到指定的行
            :param col: 写入到指定的列
            :param str: 写入的数据内容
            :return: True or False
            '''
            wb = openpyxl.load_workbook(self.path)
            sh = wb[self.sheet]
            try:
                sh.cell(row, col, str)
            except:
                return False
            else:
                wb.save(self.path)
                return True
    
    
    if __name__ == '__main__':
        excel_tool = ExcelTool(r'E:\Python_selenium_advance\homework\day16_readexcel\logintestcase.xlsx', 'login')
        cases=excel_tool.read_data()
        print(cases)

最后总结一下：  
如果你对此文有任何疑问，如果你也需要接口项目实战，如果你对软件测试、接口测试、自动化测试、面试经验交流感兴趣欢迎加入：软件测试技术群：**593462778**，群里的免费资料都是笔者十多年测试生涯的精华。还有同行大神一起交流技术哦。

作者：暗潮汹涌  
原创不易，欢迎转载，但未经作者同意请保留此段声明，并在文章页面明显位置给出原文链接。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200308154313217.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3,size_16,color_FFFFFF,t_70