Python-接口自动化（九）

python操作Excel处理测试数据（九）

(十)python操作Excel读/写测试数据

1、夹心饼干

setUp：在每一条测试用例执行之前执行

tearDown：在每一条测试用例执行之后执行

上面的这两个东东就叫做夹心饼干，在必要的时候写， 比如说在执行测试用例之前你有数据需要提前准备好，在测试用例执行结束之后有东西需要清除掉，就可以用夹心饼干。在之后的代码中可以看到这个夹心饼干的作用和用法。

2、解决用例相互关联性或依赖性太强 

方法1：写到setUp里面，每次都重新对第一条用例进行请求；

方法2：全局变量

方法3：反射

3、Excel处理测试数据

在这一部分会用到两个东西：pip install openpyxl，pip install ddt （需要安装好）

注意：openpyxl只支持后缀为.xlsx的格式，其余的均不支持；新建Excel的方式一：可以在桌面先创建一个后缀为.xlsx格式的Excel，再复制到写代码所在的文件夹中即可；方法二：找到写代码的当前目录路径，在目录路径下创建一个后缀为.xlsx的Excel即可。

（1）新建Excel并读取Excel单元格中的数据

首先我在桌面创建了一个名为xz.xlsx的Excel，在Excel中第一行第一列写了一个小猪，表单名是python

 from openpyxl import load_workbook

 # 打开Excel
 wb=load_workbook("xz.xlsx") # 返回打开的工作薄

 # 定位表单
 sheet=wb["python"] # 传表单名

 # 定位单元格:根据行列值去定位
 res=sheet.cell(1,1).value
 print(res)

打印出来的结果是：小猪

（2）打印行和列的最大值（我举的例子是存放了5行6列的数据）

 from openpyxl import load_workbook

 # 打开Excel
 wb=load_workbook("xz.xlsx") # 返回打开的工作薄

 # 定位表单
 sheet=wb["python"] # 传表单名 返回一个表单对象

 # 定位单元格:根据行列值去定位
 res=sheet.cell(1,1).value

 # 表单的最大行
 print(sheet.max_row)

 # 表单的最大列
 print(sheet.max_column)

打印出来的结果是：

5

6

（3）数据从Excel中读取出来之后的类型：数字还是数字，其余都是字符串

例如：（我举的例子是在Excel第一行中写入了4个数据）

 from openpyxl import load_workbook

 # 打开Excel
 wb=load_workbook("xz.xlsx") # 返回打开的工作薄

 # 定位表单
 sheet=wb["python"] # 传表单名 返回一个表单对象

 # 定位单元格:根据行列值去定位
 # res=sheet.cell(1,1).value

 # 表单的最大行
 print(sheet.max_row)

 # 表单的最大列
 print(sheet.max_column)

 # print(res)

 # 从Excel中读取数据
 print("url:{0}".format(sheet.cell(1,1).value))
 print("data:{0}".format(sheet.cell(1,2).value))
 print("code:{0}".format(sheet.cell(1,3).value))
 print("method:{0}".format(sheet.cell(1,4).value))

打印出来的结果是：

url:case_id
data:module
code:title
method:http_method

这里可以用type()函数打印出这些数据的类型

（4）eval()：把数据类型转换成原本的数据类型

例如：

 s='{"age":18}'
 print(eval(s),type(eval(s)))

打印出来的结果是：{'age':18} <class 'dict'>

(5)读取数据的三种方式

方式一：一次性读取所有的数据，对内存的要求要高一些（推荐使用）

 from openpyxl import load_workbook

 class DoExcel:
     def __init__(self, file_name, sheet_name):
         self.file_name = file_name
         self.sheet_name = sheet_name

     def get_data(self):
         wb = load_workbook (self.file_name)

         sheet = wb[self.sheet_name]
         #
         # res = sheet.cell (1, 1).value

         # 取到第一行的所有数据
         test_data = []
         for i in range (1, sheet.max_row + 1):
             sub_data = {}

             sub_data['method'] = sheet.cell (i, 1).value

             sub_data['url'] = sheet.cell (i, 2).value

             sub_data['data'] = sheet.cell (i, 3).value

             sub_data['expected'] = sheet.cell (i, 4).value

             test_data.append (sub_data)

         return test_data

 if __name__ == '__main__':
     print (DoExcel ("xz.xlsx", 'python').get_data ())

打印出来的结果是：

[{'method': 'case_id', 'url': 'module', 'data': 'title', 'expected': 'http_method'}, {'method': 1, 'url': 'login', 'data': '正常登录', 'expected': 'post'}, {'method': 2, 'url': 'login', 'data': '输入错误密码', 'expected': 'get'}, {'method': 3, 'url': 'recharge', 'data': '正常充值', 'expected': 'post'}, {'method': 4, 'url': 'recharge', 'data': '充值输入负数', 'expected': 'get'}]

方式二：在需要用的时候读取所有的数据（会对磁盘的读写要求高一些）

 from openpyxl import load_workbook

 class DoExcel:
     def __init__(self, file_name, sheet_name):
         self.file_name = file_name
         self.sheet_name = sheet_name
         # 获取一个表单对象
         self.sheet_obj=load_workbook(self.file_name)[self.sheet_name]
         self.max_row=self.sheet_obj.max_row

     def get_data(self,i,j):
         '''根据传入的坐标来获取值'''
         return self.sheet_obj.cell(i,j).value

 if __name__ == '__main__':
     res=DoExcel ("xz.xlsx", 'python').get_data (1,1)
     print(res)

打印出来的结果是：case_id

方式三：利用嵌套循环读取数据（以我自己创建的Excel为例，仅供参考）

 from openpyxl import load_workbook

 class DoExcel:
     def __init__(self, file_name, sheet_name):
         self.file_name = file_name
         self.sheet_name = sheet_name

     def get_header(self):
         '''获取第一行的标题'''
         wb=load_workbook(self.file_name)
         sheet=wb[self.sheet_name]
         # 存储标题行
         header=[]
         for j in range(1,sheet.max_column+1):
             header.append(sheet.cell(1,j).value)
         return header

     def get_data(self):
         '''根据嵌套循环读取数据'''

         wb = load_workbook (self.file_name)

         sheet = wb[self.sheet_name]

         header=self.get_header()# 拿到header
         test_data = []

         for i in range (2, sheet.max_row + 1):
             sub_data = {}
             for j in range(1,sheet.max_column+1):
                 sub_data[header[j-1]]=sheet.cell(i,j).value
             test_data.append (sub_data)

         return test_data # 返回获取到的数据

 if __name__ == '__main__':
     print (DoExcel ("xz.xlsx", 'python').get_data ())

打印出来的结果是：

[{'case_id': 'case_id', 'module': 'module', 'title': 'title', 'http_method': 'http_method', 'url': 'url', 'data': 'data', 'expected': 'expected', None: None}, {'case_id': 1, 'module': 'login', 'title': '正常登录', 'http_method': 'post', 'url': 'http://test.xxx.com/xxx/mvc/api/member/login', 'data': '{"mobilephone": "test", "pwd": "test"}', 'expected': 10001, None: None}, {'case_id': 2, 'module': 'login', 'title': '输入错误密码', 'http_method': 'get', 'url': 'http://test.xxx.com/xxx/mvc/api/member/login', 'data': '{"mobilephone": "test", "pwd": "test111"}', 'expected': 20111, None: None}, {'case_id': 3, 'module': 'recharge', 'title': '正常充值', 'http_method': 'post', 'url': 'http://test.xxx.com/xxx/mvc/api/member/recharge', 'data': '{"mobilephone": "test", "amount": "100"}', 'expected': 10001, None: None}, {'case_id': 4, 'module': 'recharge', 'title': '充值输入负数', 'http_method': 'get', 'url': 'http://test.xxx.com/xxx/mvc/api/member/recharge', 'data': '{"mobilephone": "test", "amount": "-100"}', 'expected': 20117, None: None}]

以上，第九部分到此结束～