python+jinja2实现接口数据批量生成工具


Posted in Python onAugust 28, 2019

在做接口测试的时候,我们经常会遇到一种情况就是要对接口的参数进行各种可能的校验,手动修改很麻烦,尤其是那些接口参数有几十个甚至更多的,有没有一种方法可以批量的对指定参数做生成处理呢。

答案是肯定的!

python的jinja2模板库可以很好的满足我们的需求,通过维护一个原始数据模板,将我们想要动态生成的变量模板化,就可以实现需求。

现在我们有这样的一个请求数据

{
 "abc":"123",
 "p2p":"123",
 "smid":"20180807220733939b66d80092eea34ce9e77f30bedff12345b7d5a3faa11b",
 "test":{
  "test1":"1",
  "test2":"2"
 },
 "test3":"3"
}

如果想对其中的smid字段进行批量修改并生成新的请求数据就可以进行如下操作:

1、首先新建一个名为fp_template.txt的文本文件;

2、将上面的接口请求数据复制粘贴进这个txt文件中,我们以此作为“模板文件”;

3、新建一个predata文件夹用来存放生成后的数据文件;

4、对smid字段进行模板化(模板格式可以参考jinja2的语法,这里不做赘述),于是上面的请求参数就变成了下面这样:

{
 "abc":"123",
 "p2p":"123",
 "smid":"{{ smid }}",
 "test":{
  "test1":"1",
  "test2":"2"
 },
 "test3":"3"
}

实现代码代码如下:

# -*- coding: UTF-8 -*-
from jinja2 import Environment,FileSystemLoader
import os

class DataTemplateFaker:
 def __init__(self):
  self.aesPath = os.getcwd()#获取启动路径
  self.resultPath = self.aesPath + "/predata/"#指定用来保存生成数据的路径
  self.templateFile = "fp_template.txt"

 #修改我们要批量生成smid的格式
 def init_smid(self,start,end):
  smidArg = [x for x in range(start, end)]
  re = []
  for n in smidArg:
   re.append("20180807220733939b66d80092eea34ce9e77f30bedff" + str(n) + "b7d5a3faa11b")
  return re

 #操作模板文件
 def preContent(self,arg):
  env = Environment(loader=FileSystemLoader('./'))
  tpl = env.get_template(self.templateFile)
  renderContent = tpl.render(smid=arg)
  return renderContent

 #通过修改的smid列表批量替换模板文件并写入指定文件中
 def makeContent(self,preList):
  x = 0
  for i in preList:
   x = x + 1
   filename = str(self.resultPath) + 'data_' + str(x) + '.txt' #用以区分存放新生成的请求数据(也可以写到一个文件中)
   renderContent = self.preContent(i)
   with open(filename, 'w') as f:
    f.writelines(renderContent)
    f.close()

if __name__ == "__main__":
 AT = DataTemplateFaker()
 reList = AT.init_smid(1,10)#控制生成数据的范围
 AT.makeContent(reList)

运行程序,就能得到新生成的数据

python+jinja2实现接口数据批量生成工具

当然,我们也可以对其他的参数进行指定修改,如修改p2p,只需要修改模板文件:

{
 "abc":"123",
 "p2p":"{{ p2p }}",
 "smid":"20180807220733939b66d80092eea34ce9e77f30bedff12345b7d5a3faa11b",
 "test":{
  "test1":"1",
  "test2":"2"
 },
 "test3":"3"
}

然后在代码中加入一个方法init_p2p()

# -*- coding: UTF-8 -*-
from jinja2 import Environment,FileSystemLoader
import os

class DataTemplateFaker:
 def __init__(self):
  self.aesPath = os.getcwd()#获取启动路径
  self.resultPath = self.aesPath + "/predata/"#指定用来保存生成数据的路径
  self.templateFile = "fp_template.txt"

 #修改我们要批量生成smid的格式
 def init_smid(self,start,end):
  smidArg = [x for x in range(start, end)]
  re = []
  for n in smidArg:
   re.append("20180807220733939b66d80092eea34ce9e77f30bedff" + str(n) + "b7d5a3faa11b")
  return re
 #修改我们要批量生成p2p的格式 
 def init_p2p(self,start,end):
  p2pArg = [x for x in range(start, end)]
  return p2pArg

 #操作模板文件
 def preContent(self,arg):
  env = Environment(loader=FileSystemLoader('./'))
  tpl = env.get_template(self.templateFile)
  renderContent = tpl.render(smid=arg)
  return renderContent

 #通过修改的smid列表批量替换模板文件并写入指定文件中
 def makeContent(self,preList):
  x = 0
  for i in preList:
   x = x + 1
   filename = str(self.resultPath) + 'data_' + str(x) + '.txt' #用以区分存放新生成的请求数据(也可以写到一个文件中)
   renderContent = self.preContent(i)
   with open(filename, 'w') as f:
    f.writelines(renderContent)
    f.close()

if __name__ == "__main__":
 AT = DataTemplateFaker()
 reList = AT.init_p2p(1,10)#控制生成数据的范围
 AT.makeContent(reList)

这只是一个很简单的demo,当然还有很多可优化的地方,比如多字段同时修改、引入faker库进行关联生成伪造数据等,越是复杂且参数繁多的接口越适用,其他的方法就可以天马行空,任君发挥了。

总结

以上所述是小编给大家介绍的python+jinja2实现接口数据批量生成工具,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对三水点靠木网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

Python 相关文章推荐
Python编写生成验证码的脚本的教程
May 04 Python
python去除字符串中的换行符
Oct 11 Python
Python实现的IP端口扫描工具类示例
Feb 15 Python
pip指定python位置安装软件包的方法
Jul 12 Python
python多任务之协程的使用详解
Aug 26 Python
python基于socket实现的UDP及TCP通讯功能示例
Nov 01 Python
详解Ubuntu环境下部署Django+uwsgi+nginx总结
Apr 02 Python
Python实现汇率转换操作
May 03 Python
matplotlib 画双轴子图无法显示x轴的解决方法
Jul 27 Python
python 常用日期处理-- datetime 模块的使用
Sep 02 Python
python自动化调用百度api解决验证码
Apr 13 Python
Python绘制散点图之可视化神器pyecharts
Jul 07 Python
解决Atom安装Hydrogen无法运行python3的问题
Aug 28 #Python
opencv实现简单人脸识别
Feb 19 #Python
Atom Python 配置Python3 解释器的方法
Aug 28 #Python
python聚类算法解决方案(rest接口/mpp数据库/json数据/下载图片及数据)
Aug 28 #Python
Numpy的简单用法小结
Aug 28 #Python
Python 通过截图匹配原图中的位置(opencv)实例
Aug 27 #Python
Python 转换文本编码实现解析
Aug 27 #Python
You might like
PHP 写文本日志实现代码
2010/05/18 PHP
php mysql_real_escape_string函数用法与实例教程
2013/09/30 PHP
php导入csv文件碰到乱码问题的解决方法
2014/02/10 PHP
php使用fsockopen函数发送post,get请求获取网页内容的方法
2014/11/15 PHP
php解决crontab定时任务不能写入文件问题的方法分析
2019/09/16 PHP
phpinfo的知识点总结
2019/10/10 PHP
根据地区不同显示时间的javascript代码
2007/08/13 Javascript
input 高级限制级用法
2009/03/26 Javascript
js获取通过ajax返回的map型的JSONArray的方法
2014/01/09 Javascript
jQuery实现可以编辑的表格实例详解【附demo源码下载】
2016/07/09 Javascript
javascript 内置对象及常见API详细介绍
2016/11/01 Javascript
微信小程序实战之运维小项目
2017/01/17 Javascript
老生常谈Bootstrap媒体对象
2017/07/06 Javascript
Angular通过angular-cli来搭建web前端项目的方法
2017/07/27 Javascript
js设置随机切换背景图片的简单实例
2017/11/12 Javascript
使用vuex解决刷新页面state数据消失的问题记录
2019/05/08 Javascript
webpack中如何加载静态文件的方法步骤
2019/05/18 Javascript
JavaScript复制变量三种方法实例详解
2020/01/09 Javascript
python编码总结(编码类型、格式、转码)
2016/07/01 Python
Python如何快速上手? 快速掌握一门新语言的方法
2017/11/14 Python
python计算两个地址之间的距离方法
2018/06/09 Python
python将字符串以utf-8格式保存在txt文件中的方法
2018/10/30 Python
Python调用Windows命令打印文件
2020/02/07 Python
Python绘图之柱形图绘制详解
2020/07/28 Python
佳能英国官方网站:Canon UK
2017/08/08 全球购物
过滤器的用法
2013/10/08 面试题
专业幼师实习生自我鉴定范文
2013/12/08 职场文书
旅游管理毕业生自荐书
2014/02/02 职场文书
面试自我介绍演讲稿
2014/04/29 职场文书
乡镇四风对照检查材料
2014/08/31 职场文书
连锁超市项目计划书
2014/09/15 职场文书
学生不讲诚信检讨书
2014/09/29 职场文书
小学中等生评语
2014/12/29 职场文书
学校国庆节活动总结
2015/03/23 职场文书
公司员工违纪检讨书
2015/05/05 职场文书
离婚起诉书范文2016
2015/11/26 职场文书