def post(self, request):
samplerId = request.data.get('samplerId')
childId = request.data.get('childId')
name = request.data.get('name')
toBeanShellParam = request.data.get('toBeanShellParam')
express = request.data.get('express')
if not express or not name:
return APIRsp(msg='存在参数为空，不创建BeanShell后置处理器!')
if not toBeanShellParam:
toBeanShellParam = ""
try:
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
sampler_xpath = json.loads(jmxInfo[0].child_info)['xpath']
if childId:
child_info = json.loads(
str(SamplersChildren.objects.get(id=childId).child_info))
child_xpath = child_info['xpath']
new_child_info = ModifyJMX(str(jmx_path)).add_after_beanshell(
sampler_xpath,
name,
toBeanShellParam,
express,
after_beanshell_xpath=child_xpath)
SamplersChildren.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(new_child_info))
else:
child_info = ModifyJMX(str(jmx_path)).add_after_beanshell(
sampler_xpath, name, toBeanShellParam, express)
s = SamplersChildren(sampler_id=samplerId,
child_name=name,
child_type='after_beanshell',
child_info=json.dumps(child_info))
s.save()
return APIRsp()
except:
return APIRsp(code=400, msg='创建BeanShell后置处理器失败，参数错误！')
def get(self, request, *args, **kwargs):
rsp = self.retrieve(request, *args, **kwargs)
if rsp.status_code == 200:
if rsp.data:
id = rsp.data['id']
children = []
qs = JmxThreadGroup.objects.values(
'id', 'child_name', 'child_type',
'child_info').filter(jmx_id=id)
for q in qs:
q['child_info'] = json.loads(q['child_info'])
children.append(q)
rsp.data = {'id': id, 'children': children}
return APIRsp(data=rsp.data)
return APIRsp(code='400',
msg='无数据',
status=rsp.status_code,
data=rsp.data)
else:
return APIRsp(code='400',
msg='无数据',
status=rsp.status_code,
data=rsp.data)
def post(self, request):
samplerId = request.data.get('samplerId')
childId = request.data.get('childId')
name = request.data.get('name')
params = request.data.get('params')
if not name:
name = "HTTP信息头管理器"
if len(params) == 1 and params[0]['key'] == "" and params[0][
'value'] == "" and not childId:
return APIRsp(msg='头参数为空，不创建头信息')
try:
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
sampler_xpath = json.loads(jmxInfo[0].child_info)['xpath']
if childId:
child_info = json.loads(
str(SamplersChildren.objects.get(id=childId).child_info))
child_xpath = child_info['xpath']
new_child_info = ModifyJMX(str(jmx_path)).add_header(
sampler_xpath,
name,
headers=params,
header_xpath=child_xpath)
SamplersChildren.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(new_child_info))
else:
child_info = ModifyJMX(str(jmx_path)).add_header(
sampler_xpath, name, headers=params)
s = SamplersChildren(sampler_id=samplerId,
child_name=name,
child_type='header',
child_info=json.dumps(child_info))
s.save()
return APIRsp()
except:
return APIRsp(code=400, msg='创建header失败，参数错误！')
def post(self, request):
"""
修改csv信息
"""
childId = request.data.get('childId')
name = request.data.get('name')
variableNames = request.data.get('variableNames')
delimiter = request.data.get('delimiter')
ignoreFirstLine = request.data.get('ignoreFirstLine')
recycle = request.data.get('recycle')
stopThread = request.data.get('stopThread')
if childId and variableNames:
csv_base_info = JmxThreadGroup.objects.get(id=childId)
jmx_path = csv_base_info.jmx
child_info = json.loads(csv_base_info.child_info)
xpath = child_info['xpath']
threadType = str(csv_base_info.child_thread)
csv_path = child_info['filename']
csv_info = ModifyJMX(str(jmx_path)).add_csv(
name,
csv_path,
variableNames,
ignoreFirstLine=ignoreFirstLine,
delimiter=delimiter,
recycle=recycle,
stopThread=stopThread,
accord=threadType,
xpath=xpath)
# 更新csv信息
JmxThreadGroup.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(csv_info))
return APIRsp()
else:
return APIRsp(code=400,
msg='修改失败，参数不完整',
status=status.HTTP_400_BAD_REQUEST)
def get(self, request):
rsp = {
"menuList": [
{
"id":
101,
"authName":
"JMX管理",
"path":
"jmxs",
"children": [{
"id": 121,
"authName": "JMX列表",
"path": "jmxs",
"children": [],
"order": None
}],
"order":
1
},
{
"id":
103,
"authName":
"任务管理",
"path":
"tasks",
"children": [{
"id": 111,
"authName": "任务列表",
"path": "tasks",
"children": [],
"order": None
}, {
"id": 112,
"authName": "任务流水",
"path": "tasks/flows",
"children": [],
"order": None
}],
"order":
2
},
# {
# "id":145,
# "authName":"报告管理",
# "path":"reports",
# "children":[
# {
# "id":146,
# "authName":"报告列表",
# "path":"reports",
# "children":[
#
# ],
# "order":None
# }
# ],
# "order":5
# },
{
"id":
102,
"authName":
"参数管理",
"path":
"params",
"children": [{
"id": 107,
"authName": "参数列表",
"path": "params",
"children": [],
"order": None
}],
"order":
5
}
]
}
return APIRsp(data=rsp)
def post(self, request):
"""
修改csv信息
"""
data = {}
old_jmx_id = request.data.get('jmxId')
user = request.data.get('userId')
if old_jmx_id:
jmx_info = Jmxs.objects.get(id=old_jmx_id)
jmx_path = str(jmx_info.jmx)
jmx_alias = str(jmx_info.jmx_alias)
temp_jmx_path = settings.JMX_URL + Tools.random_str(9) + '.jmx'
shutil.copyfile(jmx_path, temp_jmx_path)
new_jmx_name = jmx_alias + '.' + Tools.random_str(9)
new_jmx_path = settings.JMX_URL + new_jmx_name + '.jmx'
# 解析JMX
jmxinfo = ReadJmx(temp_jmx_path).analysis_jmx()
if not jmxinfo:
return APIRsp(code=400,
msg='添加失败，jmx文件错误',
status=status.HTTP_400_BAD_REQUEST)
samplers_info = jmxinfo[0]
csvs_info = jmxinfo[1]
thread_info = jmxinfo[2]
# jmx路径
data['jmx'] = new_jmx_path
# jmx别名
data['jmx_alias'] = new_jmx_name
# user的id不存在时，会校验失败
data['add_user'] = user
# 线程组基础信息
data['thread_base_info'] = json.dumps(thread_info)
obj = JmxsSerializer(data=data)
# 校验数据格式
if obj.is_valid():
obj.save()
new_jmx_id = obj.data['id']
for sampler in samplers_info:
# 保存sampler信息
sampler_children = sampler['children']
del sampler['children']
sp = JmxThreadGroup(jmx_id=new_jmx_id,
child_name=sampler['name'],
child_info=json.dumps(sampler),
child_thread=sampler['thread_type'])
sp.save()
# 获取保存后得到的id
sampler_id = sp.id
for child in sampler_children:
sc = SamplersChildren(sampler_id=sampler_id,
child_name=child['child_name'],
child_type=child['child_type'],
child_info=json.dumps(child))
sc.save()
if csvs_info:
for csv in csvs_info:
old_csv_name = csv['name']
new_csv_name = Tools.filename(
Tools.filename(
csv['name'])) + '.' + Tools.random_str(9)
new_csv_path = settings.CSV_URL + new_csv_name + '.csv'
with open(new_jmx_path, 'w') as fw:
with open(temp_jmx_path, 'r') as fr:
for line in fr:
if old_csv_name in line:
line = line.replace(
csv['name'], new_csv_name)
fw.write(line)
shutil.copyfile(csv['filename'], new_csv_path)
csv['filename'] = new_csv_path
c = JmxThreadGroup(jmx_id=new_jmx_id,
child_name=csv['name'],
child_type='csv',
child_info=json.dumps(csv),
child_thread=csv['thread_type'])
c.save()
# 保存csv到数据库
co = Csvs(csv=new_csv_path,
jmx_id=new_jmx_id,
add_user_id=user)
co.save()
else:
shutil.copyfile(temp_jmx_path, new_jmx_path)
os.remove(temp_jmx_path)
return APIRsp()
os.remove(new_jmx_path)
os.remove(temp_jmx_path)
return APIRsp(code=400,
msg='添加失败，参数校验未通过',
status=status.HTTP_400_BAD_REQUEST)
else:
return APIRsp(code=400,
msg='修改失败，参数不完整',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
"""
:param request: :param request: {'csv': ,'jmx': , 'user': 1}
:return:
"""
data = {}
# request.POST.get适用于form-data请求获取参数
csv = request.FILES.get('csv')
name = request.POST.get('name')
jmx_id = request.POST.get('jmxId')
user = request.POST.get('userId')
variableNames = request.POST.get('variableNames')
delimiter = request.POST.get('delimiter')
# mult-form-data会将json中的true或者false转换为字符串
ignoreFirstLine = request.POST.get('ignoreFirstLine')
recycle = request.POST.get('recycle')
stopThread = request.POST.get('stopThread')
threadType = request.POST.get('threadType')
if csv and jmx_id and user and variableNames and delimiter and ignoreFirstLine and recycle and stopThread and threadType:
csv_name_ext = os.path.splitext(csv.name)
csv_name = csv_name_ext[0]
csv_ext = csv_name_ext[1]
if csv_ext not in settings.CSV_ALLOWED_FILE_TYPE:
return APIRsp(code=205,
msg='无效的格式，请上传.csv格式的文件',
status=status.HTTP_205_RESET_CONTENT)
csvfile = csv_name + "." + Tools.random_str(9) + csv_ext
path = settings.CSV_URL + csvfile
with open(path, 'wb') as f:
for i in csv.chunks():
f.write(i)
data['csv'] = csvfile
# csv不存在时，接口会报错
data['jmx'] = jmx_id
# user不存在时，接口会报错
data['add_user'] = user
obj = CsvSerializer(data=data)
if obj.is_valid():
obj.save()
jmx_path = Jmxs.objects.values('jmx').get(id=jmx_id)['jmx']
csv_info = ModifyJMX(jmx_path).add_csv(
name,
path,
variableNames,
ignoreFirstLine=Tools.strToBool(ignoreFirstLine),
delimiter=delimiter,
recycle=Tools.strToBool(recycle),
stopThread=Tools.strToBool(stopThread),
accord=threadType)
# 保存csv信息
s = JmxThreadGroup(jmx_id=jmx_id,
child_name=name,
child_type='csv',
child_info=json.dumps(csv_info),
child_thread=threadType)
s.save()
return APIRsp()
return APIRsp(code=500,
msg='添加失败，校验未通过',
status=status.HTTP_400_BAD_REQUEST)
else:
return APIRsp(code=400,
msg='添加失败，参数不完整',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
"""
:param request: {'jmx': 'name': , 'add_user': 1}
:return:
"""
data = {}
jmx = request.FILES.get('jmx')
jmx_alias = request.POST.get('jmxName')
user = request.POST.get('addUser')
if jmx and user:
jmx_name_ext = os.path.splitext(jmx.name)
jmx_name = jmx_name_ext[0]
jmx_ext = jmx_name_ext[1]
if jmx_ext not in settings.JMX_ALLOWED_FILE_TYPE:
return APIRsp(code=205,
msg='无效的格式，请上传.jmx格式的文件',
status=status.HTTP_205_RESET_CONTENT)
jmxfile = jmx_name + "." + Tools.random_str(9) + jmx_ext
jmxpath = settings.JMX_URL + jmxfile
with open(jmxpath, 'wb') as f:
for i in jmx.chunks():
f.write(i)
jmxinfo = ReadJmx(jmxpath).analysis_jmx()
if not jmxinfo:
return APIRsp(code=400,
msg='添加失败，jmx文件错误',
status=status.HTTP_400_BAD_REQUEST)
samplers_info = jmxinfo[0]
csvs_info = jmxinfo[1]
thread_info = jmxinfo[2]
# jmx路径
data['jmx'] = jmxpath
# jmx名称
if jmx_alias:
data['jmx_alias'] = jmx_alias
else:
data['jmx_alias'] = jmx_name
# user的id不存在时，会校验失败
data['add_user'] = user
# 线程组基础信息
data['thread_base_info'] = json.dumps(thread_info)
obj = JmxsSerializer(data=data)
# 校验数据格式
if obj.is_valid():
obj.save()
jmx_id = obj.data['id']
for sampler in samplers_info:
# 保存sampler信息
sampler_children = sampler['children']
del sampler['children']
sp = JmxThreadGroup(jmx_id=jmx_id,
child_name=sampler['name'],
child_info=json.dumps(sampler),
child_thread=sampler['thread_type'])
sp.save()
# 获取保存后得到的id
sampler_id = sp.id
for child in sampler_children:
sc = SamplersChildren(sampler_id=sampler_id,
child_name=child['child_name'],
child_type=child['child_type'],
child_info=json.dumps(child))
sc.save()
if csvs_info:
for csv in csvs_info:
# 保存csv信息
c = JmxThreadGroup(jmx_id=jmx_id,
child_name=csv['name'],
child_type='csv',
child_info=json.dumps(csv),
child_thread=csv['thread_type'])
c.save()
return APIRsp()
os.remove(jmxpath)
return APIRsp(code=400,
msg='添加失败，参数校验未通过',
status=status.HTTP_400_BAD_REQUEST)
else:
return APIRsp(code=400,
msg='添加失败，未传入文件或用户id',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
jmx_id = request.data.get('jmxId')
# 修改的时候才需要传入childid信息
child_id = request.data.get('childId')
thread_type = request.data.get('threadType')
sampler_name = request.data.get('samplerName')
method = request.data.get('method')
url = request.data.get('url')
param_type = request.data.get('paramType')
params = request.data.get('params')
jmx_path = Jmxs.objects.values('jmx').get(id=jmx_id)['jmx']
if not jmx_path:
return APIRsp(code=400,
msg='无效的jmxID！',
status=status.HTTP_400_BAD_REQUEST)
if child_id:
# 修改已有sampler
try:
child_info = JmxThreadGroup.objects.values('child_info').get(
id=child_id, child_type='sampler')['child_info']
child_xpath = json.loads(child_info)['xpath']
rst = ModifyJMX(jmx_path).add_sampler(sampler_name,
url,
method,
accord=thread_type,
param_type=param_type,
params=params,
xpath=child_xpath)
JmxThreadGroup.objects.filter(id=child_id).update(
child_name=sampler_name,
child_info=json.dumps(rst),
child_thread=thread_type)
return APIRsp(msg='修改sampler成功！')
except:
return APIRsp(code=400,
msg='修改sampler失败，sampler参数有误！',
status=status.HTTP_400_BAD_REQUEST)
else:
try:
# 新增sampler
rst = ModifyJMX(jmx_path).add_sampler(sampler_name,
url,
method,
accord=thread_type,
param_type=param_type,
params=params)
# 保存sampler信息
s = JmxThreadGroup(jmx_id=jmx_id,
child_name=sampler_name,
child_info=json.dumps(rst),
child_thread=thread_type)
s.save()
return APIRsp(msg='新增sampler成功！', data={'samplerId': s.id})
except:
return APIRsp(code=400,
msg='新增sapmler失败，sampler参数有误！',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
data = {}
jmx_name = request.data.get('jmxName')
sampler_name = request.data.get('samplerName')
method = request.data.get('method')
url = request.data.get('url')
param_type = request.data.get('paramType')
params = request.data.get('params')
user = request.data.get('addUser')
if not sampler_name or not method or not url:
return APIRsp(code=400,
msg='创建jmx失败，接口名称、方法、url必传',
status=status.HTTP_400_BAD_REQUEST)
template_path = settings.JMX_URL + 'template.jmx'
new_jmxpath = settings.JMX_URL + jmx_name + "." + Tools.random_str(
9) + '.jmx'
shutil.copyfile(template_path, new_jmxpath)
try:
ModifyJMX(new_jmxpath).add_sampler(sampler_name,
url,
method,
param_type=param_type,
params=params)
except:
os.remove(new_jmxpath)
return APIRsp(code=400,
msg='创建jmx失败，参数错误！',
status=status.HTTP_400_BAD_REQUEST)
jmxinfo = ReadJmx(new_jmxpath).analysis_jmx(upload=False)
if not jmxinfo:
return APIRsp(code=400,
msg='添加失败，jmx文件错误！',
status=status.HTTP_400_BAD_REQUEST)
sampler_info = jmxinfo[0][0]
thread_info = jmxinfo[2]
# jmx路径
data['jmx'] = new_jmxpath
# jmx名称
data['jmx_alias'] = jmx_name
# user的id不存在时，会校验失败
data['add_user'] = user
# 线程组基础信息
data['thread_base_info'] = json.dumps(thread_info)
obj = JmxsSerializer(data=data)
# 校验数据格式
if obj.is_valid():
obj.save()
jmx_id = obj.data['id']
if sampler_info:
# 保存sampler信息
del sampler_info['children']
s = JmxThreadGroup(jmx_id=jmx_id,
child_name=sampler_name,
child_info=json.dumps(sampler_info),
child_thread=sampler_info['thread_type'])
s.save()
return APIRsp(data={'samplerId': s.id}) # sampler的id
os.remove(new_jmxpath)
return APIRsp(code=400,
msg='添加失败，参数校验未通过',
status=status.HTTP_400_BAD_REQUEST)