在做接口測試的時候，我們經常會遇到一種情況就是要對接口的參數進行各種可能的校驗，手動修改很麻煩，尤其是那些接口參數有幾十個甚至更多的，有沒有一種方法可以批量的對指定參數做生成處理呢。
答案是肯定的！
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實現接口數據批量生成工具,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對腳本之家網站的支持！
如果你覺得本文對你有幫助，歡迎轉載，煩請注明出處，謝謝！
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
