学习python的第一个小目标：通过requests+xlrd实现简单接口测试，将测试用例维护在表格中，与脚本分开。

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

点点小白的学习方式：通过确定一个小目标来想办法实现它，再通过笔记来加深印象。

面对标题中的小目标我陷入了思考。。。。嗯，首先实现利用xlrd库来取出想要的用例

首先用表格准备好用例，如图下：

先试下取number为1的一行用例试试：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import xlrd xlsurl = r"D:\RequestAPI.xlsx" # 表格文件路径
rqapi = xlrd.open_workbook(xlsurl) # 获得文件对象
sheet_name = rqapi.sheet_names()[0] # 获取表格第一个sheet名称
sheet1 = rqapi.sheet_by_name(sheet_name) # 获取表格第一个sheet对象
nrow = sheet1.nrows # 获取表格行总数
ncols = sheet1.ncols # 获取表格列总数
col_data = sheet1.col_values(0) # 获取表格第一列的数据
row_data = sheet1.row_values(1) # 获取表格第二行的数据
print "======================================================================="
print row_data
print type(row_data)

执行结果如图：

可以看到，表格中number为1的一行数据已经打印出来了，并且是list类型的，这样通过遍历list可以将数据都当个取出来，不过我没这么做，因为表格中paramete是保存的请求参数，在requests库中，请求参数是dict格式的，所以不需要把每个参数都遍历出来（requests库的使用在笔记后面别急）。

搞定取单行数据，那接下来再取每一行的数据，并把每行中需要的列也取出来，上代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import xlrdclass ApiTest:
def xlsee(self, xlsFile):
rqapi = xlrd.open_workbook(xlsFile) # 获得文件对象
sheet_name = rqapi.sheet_names()[0] # 获取第一个sheet名称
sheet = rqapi.sheet_by_name(sheet_name) # 获取第一个sheet对象
return sheetif __name__ == "__main__":
apitest = ApiTest()
xlsFile = r"D:\RequestAPI.xlsx" # 文件路径
sheet1 = apitest.xlsee(xlsFile)
nrow = sheet1.nrows # 获取行总数
ncols = sheet1.ncols # 获取列总数
col_data = sheet1.col_values(0) # 获取第一列的数据 for i in range(1, nrow): # 循环每行，并获取每行每列的值，第0行是标题，所以从第一行开始取
row_data = sheet1.row_values(i) # 获取第i行的数据
nums = int(row_data[0]) # 用例编号
rqtypes = str(row_data[2]) # 请求类型
rqurls = str(row_data[1]) # 请求地址
a = row_data[5:] # 请求参数，并且一次取出这一行所有的参数
parametes = dict(zip(a[0::2], a[1::2])) # 将list类型的参数转换为dict字典类型的
code = row_data[3] # 校验结果用的code状态码
coderesult = row_data[4] # 用于显示校验结果的参数pass或error
print "=================================================================================="
print "用例编号:", nums
print "请求类型", rqtypes
print "请求地址", rqurls
print "请求参数parametes：", parametes
print "校验结果用的code状态码", code
print "用于显示校验结果的参数", coderesult

对于上面代码，简单封装了一下，注意

a = row_data[5:] # 请求参数，并且一次取出这一行所有的参数这个利用list[a:]来取值，可取第a位后面所有的数据，这就表明，我可在表格中设置N个参数，而不会限制于不同接口参数个数不一样的限制了
上面代码运行结果如下：

从表格中读取用例的问题已经搞定了，接下来就是通过requests来执行这些用例，上代码：

class ApiTest:
# 请求主方法
def request(self, rqtype, rqurl, paramete, headers):
self.rqurl = rqurl # API地址
self.rqtype = rqtype # 请求类型get or post
self.paramete = paramete # 请求参数
self.headers = headers # 请求头 if rqtype == "get":
apirqhttp = requests.get(url=rqurl, params=paramete, headers=headers) # 发送请求
code = apirqhttp.status_code # 保存返回状态
pam = apirqhttp.text # 保存返回数据并将json转为dict
return code, pam
if rqtype == "post":
apirqhttp = requests.post(url=rqurl, data=paramete, headers=headers)
code = apirqhttp.status_code
pam = apirqhttp.text
return code, pam
else:
print "请求参数错误，请求类型只支持get+post，请求地址支持string，参数支持dict"

其中有个小插曲，就是接口中常见的身份验证问题，需要获取token值，将token值作为herders请求头来进行身份校验

于是再通过登录接口来获取这个token（注意：token值校验不同项目可能不一样，比如我们公司就在touken值前面加了个brarer ，所以需要再将token值拼接一下）值并参数化，上代码：

# 获取B2B分销商token值方法
def seltoken(self):
rqtypes = "post"
rqurls = "http://xxxxx.dddd.com//account/authorize"
parametes = {"username": "Wbfxs001", "password": "111111Qq", "grant_type": "password"}
headers = None
cod, pam = ApiTest().request(rqtypes, rqurls, parametes, headers) # 掉用request方法请求登录
pam = json.loads(pam) # 保存返回数据并将json转为dict
access_token = pam["access_token"] # 截取dic中access_token键的value
access_token = "bearer " + str(access_token) # 拼接access_token为最终需要的token值
return access_token

最后将xlrd+requests方法都封装好，上代码（上面代码都整合在这里了）：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2017-04-13 18:39
import requests, xlrd, xlrd, jsonclass ApiTest:
# 请求主方法
def request(self, rqtype, rqurl, paramete, headers):
self.rqurl = rqurl # API地址
self.rqtype = rqtype # 请求类型get or post
self.paramete = paramete # 请求参数
self.headers = headers # 请求头 if rqtype == "get":
apirqhttp = requests.get(url=rqurl, params=paramete, headers=headers) # 发送请求
code = apirqhttp.status_code # 保存返回状态
pam = apirqhttp.text # 保存返回数据并将json转为dict
return code, pam
if rqtype == "post":
apirqhttp = requests.post(url=rqurl, data=paramete, headers=headers)
code = apirqhttp.status_code
pam = apirqhttp.text
return code, pam
else:
print "请求参数错误，请求类型只支持get+post，请求地址支持string，参数支持dict" # 获取B2B分销商token值方法
def seltoken(self):
rqtypes = "post"
rqurls = "http://xxxxx.dddd.com//account/authorize"
parametes = {"username": "Wbfxs001", "password": "111111Qq", "grant_type": "password"}
headers = None
cod, pam = ApiTest().request(rqtypes, rqurls, parametes, headers) # 掉用request方法请求登录
pam = json.loads(pam) # 保存返回数据并将json转为dict
access_token = pam["access_token"] # 截取dic中access_token键的value
access_token = "bearer " + str(access_token) # 拼接access_token为最终需要的token值
return access_token def xlsee(self, xlsFile):
rqapi = xlrd.open_workbook(xlsFile) # 获得文件对象
sheet_name = rqapi.sheet_names()[0] # 获取第一个sheet名称
sheet = rqapi.sheet_by_name(sheet_name) # 获取第一个sheet对象
return sheetif __name__ == "__main__":
apitest = ApiTest()
xlsFile = r"D:\RequestAPI.xlsx" # 文件路径
sheet1 = apitest.xlsee(xlsFile)
nrow = sheet1.nrows # 获取行总数
ncols = sheet1.ncols # 获取列总数
col_data = sheet1.col_values(0) # 获取第一列的数据 for i in range(1, nrow): # 循环每行，并获取每行每列的值
row_data = sheet1.row_values(i) # 获取第i行的数据
nums = int(row_data[0]) # 获取第i行的某个数据
rqtypes = str(row_data[2])
rqurls = str(row_data[1])
a = row_data[5:]
parametes = dict(zip(a[0::2], a[1::2]))
code = row_data[3]
coderesult = row_data[4] access_token = apitest.seltoken() # 获取token
headers = {"Authorization": access_token}
codetest, pamtest = apitest.request(rqtypes, rqurls, parametes, headers)
print "用例编号：", nums, "code码：", codetest
print pamtest