200 lines
7.2 KiB
Python
200 lines
7.2 KiB
Python
#!/usr/bin/python3.7
|
||
# -*- coding: utf-8 -*-
|
||
#-------------------------------------------------------------------------------
|
||
# @Time : 2022/8/11 20:36
|
||
# @Author : mojin
|
||
# @Email : 397135766@qq.com
|
||
# @File : api_request.py
|
||
# @Software: PyCharm
|
||
#-------------------------------------------------------------------------------
|
||
|
||
from requests import Session
|
||
import requests
|
||
from common.logger import Logger
|
||
from common.exchange_data import ExchangeData
|
||
import allure,json,re
|
||
from common.read_file import ReadFile
|
||
from common.condition import Condition
|
||
from common.all_path import images_Path
|
||
|
||
class Api_Request():
|
||
|
||
|
||
@classmethod
|
||
def api_data(cls,cases,env_url):
|
||
|
||
url,env=env_url #环境，url
|
||
|
||
allure.dynamic.story(cases[0])
|
||
del cases[0]
|
||
|
||
Condition().skip_if(cases)
|
||
allure.dynamic.severity(ReadFile.read_config('$..cor_rel_case_severity')[cases[4]])
|
||
|
||
|
||
request_headers=str(ReadFile.read_config('$.request_headers')) #获取配置文件中的请求头
|
||
request_parameters=str(ReadFile.read_config('$.request_parameters')) #获取配置文件中的请求参数
|
||
#url="http://192.168.1.10:8888/api/private/v1/"
|
||
(
|
||
case_id,
|
||
case_title,
|
||
header_ex,
|
||
path,
|
||
case_severity,
|
||
skips,
|
||
method,
|
||
parametric_key,
|
||
file_obj,
|
||
data,
|
||
extra,
|
||
sql,
|
||
expect,
|
||
) = cases
|
||
|
||
# a,b=1,2
|
||
case_title=ExchangeData.rep_expr(case_title,return_type='srt')
|
||
path=ExchangeData.rep_expr(path,return_type='srt')
|
||
header_ex=ExchangeData.rep_expr(header_ex,return_type='dict')
|
||
request_headers=ExchangeData.rep_expr(request_headers,return_type='dict')
|
||
data=ExchangeData.rep_expr(data,return_type='dict')
|
||
file_obj=ExchangeData.rep_expr(file_obj,return_type='srt')
|
||
|
||
request_parameters = ExchangeData.rep_expr(request_parameters, return_type='dict')
|
||
|
||
header_ex.update(request_headers) #合并配置文件中请求头
|
||
#request_headers.update(header_ex)#配置文件中的请求头和获取excel请求头合并
|
||
data.update(request_parameters) #合并配置文件中请求参数
|
||
|
||
#
|
||
if file_obj!="":
|
||
#Logger.error(file_obj)
|
||
file_name = (file_obj.split('\\')[-1])
|
||
Suffix = file_name.split('.')[-1].lower()
|
||
if Suffix in ['jpeg', 'jpg', 'png']:
|
||
file_type = "image/%s" % Suffix
|
||
elif Suffix in ['mp4', ]:
|
||
file_type = "video/%s" % Suffix
|
||
else:
|
||
file_type = "image/%s" % Suffix
|
||
from requests_toolbelt import MultipartEncoder
|
||
data=MultipartEncoder(
|
||
fields={
|
||
"file": (file_name,
|
||
open("%s/%s"%(images_Path,file_obj), 'rb'),#'./config/1.jpg'
|
||
file_type)
|
||
}
|
||
)
|
||
header_ex["Content-Type"] = data.content_type
|
||
|
||
|
||
Logger.info(case_title)
|
||
allure.dynamic.title(case_title)
|
||
allure.dynamic.description("【用例名称】：%s\n\n【请求地址】：%s%s\n\n【请求参数】：%s"%(case_title,url,path,data))
|
||
#allure.dynamic.link('%s%s'%(url,path), name='%s%s'%(url,path)) # 关联的连接
|
||
|
||
|
||
pattern = re.compile(r'^((https|http|ftp|rtsp|mms)?:\/\/)[^\s]+')
|
||
if (pattern.search(path)) == None: # 判断读取的地址是否有前缀地址http://192.168.1.153:8562
|
||
urls ="%s/%s"%(url,path) # 无前缀读取配置文件添加前缀
|
||
else:
|
||
urls = path # 有前缀使用读取的完整地址
|
||
|
||
res=Api_Request().api_request(urls,method,parametric_key,header_ex,(data),file_obj)
|
||
|
||
ExchangeData.Extract(res,extra)
|
||
#
|
||
# ExchangeData.extra_allure(extra)#显示提取参数路径
|
||
# Logger.info('提取参数路径：%s' % extra)
|
||
# Logger.info('参数池：%s' % ExchangeData.extra_pool)
|
||
|
||
|
||
return res
|
||
|
||
|
||
|
||
|
||
|
||
|
||
def api_request(self,url, method, parametric_key, header=None, data=None, file=None) -> dict:
|
||
if parametric_key=="params":
|
||
parametric={"params":data}
|
||
elif parametric_key=="data":
|
||
parametric={"data":data}
|
||
elif parametric_key=="json":
|
||
parametric={"json":data}
|
||
else:
|
||
raise ValueError("“parametric_key”的可选关键字为params, json, data")
|
||
|
||
if file!="":
|
||
data_=str(data)
|
||
else:
|
||
data_=data
|
||
req_info = {
|
||
"请求地址": url,
|
||
"请求头": header,
|
||
"请求方法": method,
|
||
'参数类型':parametric_key,
|
||
"请求数据": data_,
|
||
"上传文件": str(file),
|
||
}
|
||
with allure.step('请求数据：'):
|
||
allure.attach(
|
||
json.dumps(req_info, ensure_ascii=False, indent=4),
|
||
"附件内容",
|
||
allure.attachment_type.JSON,
|
||
)
|
||
|
||
|
||
Logger.info('接口地址：%s' % url)
|
||
Logger.info('请求头：%s' % header)
|
||
Logger.info('请求方法：%s' % method)
|
||
Logger.info('参数类型：%s' % parametric_key)
|
||
Logger.info('请求参数：%s' % data)
|
||
Logger.info('上传文件：%s' % file)
|
||
try:
|
||
|
||
#res = requests.request(method=method, url=url, headers=header,files=file, **parametric)#files=file,
|
||
res = requests.request(method=method, url=url, headers=header, **parametric)#files=file,
|
||
# res = self.request(method=method, url=url, files=file, headers=header, data=data)
|
||
response = res.json()
|
||
|
||
|
||
except Exception as e:
|
||
Logger.error('请求发送失败：%s'%((e)))
|
||
response={'response':str(e)}
|
||
#raise '请求发送失败：%s'%((e))
|
||
|
||
|
||
Logger.info('返回响应：%s' % response)
|
||
|
||
with allure.step('响应数据：'):
|
||
allure.attach(
|
||
json.dumps(response, ensure_ascii=False, indent=4),
|
||
"附件内容",
|
||
allure.attachment_type.JSON,
|
||
)
|
||
|
||
|
||
return response
|
||
|
||
|
||
#
|
||
# if __name__ == '__main__':
|
||
# Api_Request=Api_Request()
|
||
# method = 'post' #get,post,put,delete,head,options,patch
|
||
# url = 'http://***.**.***.*:9999/stage-api/'
|
||
# header = {"Content-Type": "application/json;charset=UTF-8","Authorization":"eyJhbGciOiJIUzUxMiJ9.eyJsb2dpbl91c2VyX2tleSI6IjY2YTcxMjVjLTY4N2EtNGI1OS04ODJlLTE1YTk4ZDMxMWJiNyJ9.gfkoOQ6ZQbut71eSy1vbfZax-EZGKy58i1wnEdO2sNf_NqiM_RxXjGMz4otQq25VamKIFLiAnypmRZ6u3gykdQ"}
|
||
# """
|
||
# 1. params：类似这种：url?参数名=参数值&参数名1=参数值1
|
||
# 2. data：请求头content-type是from表单类型。
|
||
# 3. json：请求头content-type：application/json。
|
||
# """
|
||
# parametric_key="json"# params,data,json三种类型
|
||
# data ={"searchValue":'',"createBy":"ff","createTime":"2022-02-25 14:16:50","updateBy":"","updateTime":'',"remark":'',"params":{},"noticeId":4,"noticeTitle":"22222","noticeType":"1","noticeContent":"wqe ","status":"0"}
|
||
# #data ={}
|
||
# file='' #上传文件路径
|
||
# a=Api_Request.api_request(url,method,parametric_key,header,data,file)
|
||
# print(a)
|
||
#
|
||
|