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

更新时间：2019年08月28日 08:51:13 作者：技术永无止境

这篇文章主要介绍了python+jinja2实现接口数据批量生成工具的操作方法，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下

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

答案是肯定的！

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)

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

当然，我们也可以对其他的参数进行指定修改，如修改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基于回溯法子集树模板解决取物搭配问题,简单描述了搭配问题并结合实例形式分析了Python使用回溯法子集树模板解决取物搭配问题的具体步骤与相关操作技巧,需要的朋友可以参考下
2017-09-09
利用信号如何监控Django模型对象字段值的变化详解
这篇文章主要给大家介绍了关于利用信号如何监控Django模型对象字段值变化的相关资料，文中通过示例代码介绍的非常详细，需要的朋友可以参考借鉴，下面随着小编来一起学习学习吧。
2017-11-11
Python 工具类实现大文件断点续传功能详解
用python进行大文件下载的时候，一旦出现网络波动问题，导致文件下载到一半。如果将下载不完全的文件删掉，那么又需要从头开始，如果连续网络波动，是不是要头秃了。本文提供断点续传下载工具方法，希望可以帮助到你
2021-10-10
Python3+pycuda实现执行简单GPU计算任务
GPU的加速技术在深度学习、量子计算领域都已经被广泛的应用。这篇文章就来和大家聊聊Python3如何利用pycuda执行简单GPU计算任务，感兴趣的可以了解一下
2023-03-03
使用Python发送邮件附件以定时备份MySQL的教程
这篇文章主要介绍了使用Python发送邮件附件以定时备份MySQL的教程,本文的示例基于CentOS,需要的朋友可以参考下
2015-04-04
python dataframe获得指定行列实战代码
对于一个DataFrame,常常需要筛选出某列为指定值的行,下面这篇文章主要给大家介绍了关于python dataframe获得指定行列的相关资料,文中通过代码介绍的非常详细,需要的朋友可以参考下
2023-12-12
python 实现两个线程交替执行
这篇文章主要介绍了python 实现两个线程交替执行，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-05-05
关于Python中循环器 itertools的介绍
循环器是对象的容器，包含有多个对象。通过调用循环器的next()方法 (__next__()方法，在Python 3.x中)，循环器将依次返回一个对象。直到所有的对象遍历穷尽，循环器将举出StopIteration错误。这篇文章将对此做一个详细介绍，感兴趣的小伙伴请参考下面文字内容
2021-09-09
python中pdb模块实例用法
在本篇文章里小编给大家整理的是一篇关于python中pdb模块实例用法的相关内容，有兴趣点朋友们可以学习下。
2021-01-01
Python实现注册、登录小程序功能
本文通过实例代码给大家介绍了Python实现登录、注册小程序功能，代码简单易懂非常不错，具有一定的参考借鉴价值，需要的朋友参考下吧
2018-09-09

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

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具