利用Python与Jinja2实现接口测试数据批量生成

纸上星河

发布时间：2026-10-13| 3人浏览过

原创

本文介绍如何使用Python结合Jinja2模板引擎，解决接口测试中参数校验繁琐的问题。通过定义模板文件，对smid、p2p等字段进行批量替换与生成，实现自动化构造大量测试数据，显著提升测试效率。

接口测试数据批量生成的需求背景

在进行接口测试时，经常需要对接口参数进行各种可能的校验。当接口包含数十个甚至更多参数时，手动逐一修改显得极其繁琐。是否存在一种方法，能够批量处理并生成指定参数的数据呢？

答案是肯定的。

Python的Jinja2模板库能够很好地满足这一需求。通过维护一个原始数据模板，将需要动态生成的变量进行模板化处理，即可实现批量生成数据的想法。

基于Jinja2模板的实现方案

假设我们拥有如下请求数据：


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

若需对其中smid字段进行批量修改并生成新的请求数据，可按以下步骤操作：

首先，新建一个名为fp_template.txt的文本文件。

其次，将上述接口请求数据复制并粘贴至该txt文件中，将其作为“模板文件”。

接着，新建一个predata文件夹，用于存放生成后的数据文件。

最后，对smid字段进行模板化处理（模板格式可参考Jinja2语法，此处不再赘述）。处理后的请求参数示例如下：


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

Python代码实现与运行结果

实现代码代码如下：


# -*- 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实现接口数据批量生成工具，希望对大家有所帮助。如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对本站网站的支持！如果你觉得本文对你有帮助，欢迎转载，烦请注明出处，谢谢！