python+jinja2實現介面資料批量生成工具
阿新 • • 發佈:2019-08-28
在做介面測試的時候,我們經常會遇到一種情況就是要對介面的引數進行各種可能的校驗,手動修改很麻煩,尤其是那些介面引數有幾十個甚至更多的,有沒有一種方法可以批量的對指定引數做生成處理呢。
答案是肯定的!
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庫進行關聯生成偽造資料等,越是複雜且引數繁多的介面越適用,其他的方法就可以天馬行空,任君發揮了