python接口自动化测试 - openpyxl基本使用

前言

• 当你做接口自动化测试时，测试用例逐渐变多情况下，如果所有测试用例都通过代码管理将会使得代码十分臃肿，而且维护成本会很高；
• 所以我们一般会通过Excel去管理所有的测试用例，而openpyxl库提供了访问Excel的方法

openpyxl简单介绍

• 一个Python库，用于读取/写入Excel 2010  xlsx 、xlsm 、xltx 、xltm 文件
• 不能操作 xls 文件

openpyxl简单概念

• Workbook：excel工作表
• Sheet：工作表中的一张表
• Cell：其中的一个单元格
• 简单步骤：打开Workbook，选中Sheet，操作Cell

openpyxl简单使用

 import openpyxl

 if __name__ == '__main__':
     path = 'F:/imocInterface/case/imooc.xlsx'
     # 读取excel文件
     workbook = openpyxl.load_workbook(path)
     # 读取所有sheet
     sheet = workbook.get_sheet_names()
     # 获取某个sheet
     sheet = workbook[sheet[0]]
     # 获取某个cell的值
     cell_val = sheet.cell(row=2, column=2).value
     print(cell_val)

包含知识点

• 调用 load_workbook() 等同于调用 open()
• 第8、10行代码可能浓缩成一行代码 workbook.get_sheet_by_name("sheet的名字") ，前提是你得知道sheet的命名
• cell(row, column, value=None) 三个参数分别是：行，列，值；若设置了value相当于赋值操作，会覆盖原本的值

openpyxl操作单元格

访问单个cell

     # 方式一：获取A4单元格的值
     cell_val = sheet['A4'].value
     # 方式二：获取第二行,第二列的单元格的值
     cell_val = sheet.cell(row=2, column=2).value

访问多个cell

     # A1-B3的单元格 共6个
     cell_range = sheet['A1':'B3']

     # A1-A3的单元格 共3个
     cell_range = sheet['A1:A3']

     # 第十行的单元格
     cell_range = sheet[10]

     # 第1、2行的单元格
     cell_range = sheet[1:2]

 注意：以上方法返回的是都是cell对象组成tuple

获取指定范围的cell

     # 返回行
     for col in sheet.iter_rows(min_row=1, min_col=1, max_col=3, max_row=2):
         for cell in col:
             print(cell)

     # 返回列
     for col in sheet.iter_cols(min_row=1, min_col=1, max_col=3, max_row=2):
         for cell in col:
             print(cell)

获取指定范围cell的测试结果

# 返回行
<Cell 'Sheet1'.A1>
<Cell 'Sheet1'.B1>
<Cell 'Sheet1'.C1>
<Cell 'Sheet1'.A2>
<Cell 'Sheet1'.B2>
<Cell 'Sheet1'.C2>

# 返回列
<Cell 'Sheet1'.A1>
<Cell 'Sheet1'.A2>
<Cell 'Sheet1'.B1>
<Cell 'Sheet1'.B2>
<Cell 'Sheet1'.C1>
<Cell 'Sheet1'.C2>

官方提醒：出于性能考虑， sheet.iter_cols() 方法不支持在只读模式使用

获取sheet内所有行和列的cell

     # 以列的形式,获取sheet的全部cell
     data = tuple(sheet.columns)

     # 以行的形式,获取sheet的全部cell
     data = tuple(sheet.rows)

     # 获取所有数据
     data = tuple(sheet.values)
     # 指定返回某一行数据
     print(data[2])

注意： sheet.rows 返回的是一个对象，需要用 tuple() 才能将对象转换成tuple

官方提醒：出于性能考虑， sheet.cloumns 方法不支持在只读模式使用

获取sheet的行数、列数

     # 获取sheet最大行
     data = sheet.max_row

     # 获取sheet最多列
     data = sheet.max_column

     # 获取sheet最小行
     data = sheet.min_row

     # 获取sheet最小列
     data = sheet.min_column

openpyxl写入数据

     workbook = openpyxl.load_workbook(path)
     sheet = workbook.active
     sheet.cell(row=2, col=2, value="")
     workbook.save(filename=path)

包含知识点

• 写入数据的文件需要是可写文件，需要已关闭；若打开着excel的话无法写入会报 PermissionError: [Errno 13] Permission denied: 'XXXXX.xlsx'
• 其实就是四部曲：获取excel - 执行 workbook.active - 赋值操作 - 保存文件
• 赋值操作还可以是  sheet["B2"] = "" ，等价于上面第三行代码
• save() 会覆盖原有文件，不会有提醒