python完成excel的批量工作_Excel+Python十分钟跑完批量API，就是这么简单粗暴又好用...

现在API测试被越来越多的公司重视&＃xff0c;对于测试工程师来说每次收到需求之后首先要做的是研究需求文档。既然接口是两个独立系统之间同步数据或访问对方程序的途径&＃xff0c;我们也就是要先看看这些接口到底是连接了哪两个系统&＃xff0c;它们之间的关系到底是怎样的。

所以一般需求文档里会包括接口名称&＃xff0c;接口描述&＃xff0c;接口类型&＃xff0c;接口地址&＃xff0c;推送参数&＃xff0c;返回参数这几项内容。熟悉Postman的同学可以发现这些都是必需的部分&＃xff1a;

有了这几项内容之后&＃xff0c;我们需要把需求文档转换成Excel&＃xff0c;这样不管是做案例管理&＃xff0c;还是对比结果&＃xff0c;都相对容易一些。而后期Python读取和写入也更方便。在Excel中&＃xff0c;我们只写入变化的部分&＃xff0c;而对于相对固定的&＃xff0c;例如server信息则直接在Python代码中参数化。所以做出来的输入文档如下&＃xff1a;

有了这些内容&＃xff0c;我们还是使用xlrd和xlwt包对excel进行处理&＃xff0c;对于如何使用xlrd和xlwt就不在本文做过多解释了。为什么要新建一个workbook2&＃xff1f;因为在跑完一遍之后&＃xff0c;如果我们发现有些接口跑出来的结果有问题&＃xff0c;我们可以不用关闭workbook1&＃xff0c;直接进行修改&＃xff0c;然后rerun&＃xff0c;就会生成新的workbook2&＃xff0c;覆盖之前的结果&＃xff0c;节省打开关闭workbook的时间。

workbook1 &＃61; xlrd.open_workbook(r"C:\Demo\API\API_Cases.xlsx")

workbook2 &＃61; xlwt.Workbook(encoding&＃61;&＃39;utf-8&＃39;)

sheet1 &＃61; workbook1.sheet_by_index(0)

sheet2 &＃61; workbook2.add_sheet("response")

读取内容之后&＃xff0c;我们对需要参数化的部分进行修改&＃xff0c;一般情况下&＃xff0c;url的前半部分为http://xx.xx.xx.xx:8080不变&＃xff0c;所以我们只需要把excel中第二列取出来拼接成url即可&＃xff0c;body部分也如是&＃xff0c;如果有固定部分的body&＃xff0c;我们可以在python中直接定义&＃xff0c;然后与excel中的内容进行拼接&＃xff1a;

par_url &＃61; sheet1.col_values(1)

body &＃61; sheet1.col_values(2)

当然&＃xff0c;拼接的代码需要写在循环里面&＃xff1a;

request_url &＃61; &＃39;https://postman-echo.com&＃39; &＃43; par_url[i]

print(request_url)

执行的时候&＃xff0c;我们用到的是requests这个包&＃xff0c;如果需求中既有GET&＃xff0c;又有POST&＃xff0c;又有PUT&＃xff0c;还有DELETE&＃xff0c;由于用到的参数不同&＃xff0c;我们可以分别处理。对于不同请求需要用到的不同的header&＃xff0c;我们也可以在循环里对其进行定义&＃xff0c;如果区别很大&＃xff0c;我们也可以在excel中单独定义一列&＃xff0c;然后在代码中加入相应的读取语句即可。对于POST类型的请求&＃xff0c;我们需要以下几个参数&＃xff1a;

response &＃61; requests.post(url&＃61;request_url, headers&＃61;{}, data&＃61;body[i])

print(body[i])

在执行完之后&＃xff0c;如果直接写入excel&＃xff0c;我们打开response文件之后会发现&＃xff0c;所有的返回内容都变成了一行&＃xff0c;非常不便于分析。所以我们需要先对返回的数据进行处理&＃xff0c;再写入excel。如果返回的内容中包含中文&＃xff0c;则必须加上ensure_ascii&＃61;False, 否则就会出现编码问题。

json_dic &＃61; json.dumps(json.loads(response.content), indent&＃61;4, ensure_ascii&＃61;False)

最后&＃xff0c;我们把整理好的输出内容写入新建的excel文档并保存&＃xff0c;就大功告成了。

sheet2.write(i, 0, json_dic)

workbook2.save(r"C:\Demo\API\response.xls")

其中遇到了一些问题&＃xff0c;也在这里提示一下&＃xff0c;比如在POST请求中&＃xff0c;有些参数是随机数&＃xff0c;导致每次发送请求的时候都需要手动修改&＃xff0c;否则就会报“已存在”之类的错&＃xff0c;这时候我们需要用到random这个包。我们以生成10位随机数为例&＃xff0c;在代码中用生成的10位随机数替换掉body中{{random}}的内容&＃xff1a;

random_num &＃61; str(math.floor(1e10 * random.random()))

response &＃61; requests.post(url&＃61;request_url, headers&＃61;{}, data&＃61;body[i].replace(&＃39;{{random}}&＃39;, random_num))

运行结果如图&＃xff1a;

生成的新文件&＃xff1a;