import requests
import json
import xlrd
import xlwt
from xlutils import copy
#
def login(username,password):
url="http://49.234.14.39:5001/user/login"
data ={
"username":username,
"password":password
}
response=requests.request('post',url,json=data)
return response.json()['access_token']
class MyToke:
def __init__(self,method,url,header,data,token):
self.method = method
self.url = url
self.header =header
self.data = data
self.token = token
def test_01(self):
data = json.loads(self.data)
if "access_token" in data:
data['access_token']=self.token
response = requests.request(self.method,self.url,json=data)
return response.json()
class Set:
def __init__(self,token):
self.excel_path='./txt.xls'
self.work_book=xlrd.open_workbook(self.excel_path,formatting_info=True)
self.work_sheet=self.work_book.sheet_by_name('示例')
self.new_work_book = copy.copy(self.work_book)
self.new_sheet=self.new_work_book.get_sheet(0)
self.token = token('admin','123456')
def read_excel(self):
for case in range(2,self.work_sheet.nrows):
case_id =self.work_sheet.cell_value(case,0)
desc=self.work_sheet.cell_value(case,1)
url = self.work_sheet.cell_value(case,2)
method = self.work_sheet.cell_value(case,3)
header = self.work_sheet.cell_value(case,4)
data = self.work_sheet.cell_value(case,5)
result= self.work_sheet.cell_value(case,6)
if not url or not data:
print(f"测试用例{case_id},{desc}url或者body为空")
continue
if header:
header = json.loads(header)
if result:
result = json.loads(result)
res =MyToke(method,url,header,data,self.token)
resp=res.test_01()
print(resp)
if __name__ == '__main__':
es=Set(login)
es.read_excel()