快捷導(dǎo)航

python+jinja2實(shí)現(xiàn)接口數(shù)據(jù)批量生成工具

更新時(shí)間：2019年08月28日 08:51:13 作者：技術(shù)永無止境

這篇文章主要介紹了python+jinja2實(shí)現(xiàn)接口數(shù)據(jù)批量生成工具的操作方法，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下

在做接口測試的時(shí)候，我們經(jīng)常會(huì)遇到一種情況就是要對接口的參數(shù)進(jìn)行各種可能的校驗(yàn)，手動(dòng)修改很麻煩，尤其是那些接口參數(shù)有幾十個(gè)甚至更多的，有沒有一種方法可以批量的對指定參數(shù)做生成處理呢。

答案是肯定的！

python的jinja2模板庫可以很好的滿足我們的需求，通過維護(hù)一個(gè)原始數(shù)據(jù)模板，將我們想要?jiǎng)討B(tài)生成的變量模板化，就可以實(shí)現(xiàn)需求。

現(xiàn)在我們有這樣的一個(gè)請求數(shù)據(jù)

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

如果想對其中的smid字段進(jìn)行批量修改并生成新的請求數(shù)據(jù)就可以進(jìn)行如下操作：

1、首先新建一個(gè)名為fp_template.txt的文本文件；

2、將上面的接口請求數(shù)據(jù)復(fù)制粘貼進(jìn)這個(gè)txt文件中，我們以此作為“模板文件”；

3、新建一個(gè)predata文件夾用來存放生成后的數(shù)據(jù)文件；

4、對smid字段進(jìn)行模板化（模板格式可以參考jinja2的語法，這里不做贅述），于是上面的請求參數(shù)就變成了下面這樣：

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

實(shí)現(xiàn)代碼代碼如下：

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

class DataTemplateFaker:
 def __init__(self):
  self.aesPath = os.getcwd()#獲取啟動(dòng)路徑
  self.resultPath = self.aesPath + "/predata/"#指定用來保存生成數(shù)據(jù)的路徑
  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' #用以區(qū)分存放新生成的請求數(shù)據(jù)（也可以寫到一個(gè)文件中）
   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)#控制生成數(shù)據(jù)的范圍
 AT.makeContent(reList)

運(yùn)行程序，就能得到新生成的數(shù)據(jù)

當(dāng)然，我們也可以對其他的參數(shù)進(jìn)行指定修改，如修改p2p，只需要修改模板文件：

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

然后在代碼中加入一個(gè)方法init_p2p()

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

class DataTemplateFaker:
 def __init__(self):
  self.aesPath = os.getcwd()#獲取啟動(dòng)路徑
  self.resultPath = self.aesPath + "/predata/"#指定用來保存生成數(shù)據(jù)的路徑
  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' #用以區(qū)分存放新生成的請求數(shù)據(jù)（也可以寫到一個(gè)文件中）
   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)#控制生成數(shù)據(jù)的范圍
 AT.makeContent(reList)

這只是一個(gè)很簡單的demo，當(dāng)然還有很多可優(yōu)化的地方，比如多字段同時(shí)修改、引入faker庫進(jìn)行關(guān)聯(lián)生成偽造數(shù)據(jù)等，越是復(fù)雜且參數(shù)繁多的接口越適用，其他的方法就可以天馬行空，任君發(fā)揮了。

總結(jié)

以上所述是小編給大家介紹的python+jinja2實(shí)現(xiàn)接口數(shù)據(jù)批量生成工具,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！

您可能感興趣的文章: