def delete(self, request, pk, *args, **kwargs):
"""
停止任务
:param request:
:param pk:
:param args:
:param kwargs:
:return:
"""
try:
task = Task.objects.get(id=pk)
task.status = TASK_STATUS[2][0]
task.save()
except Task.DoesNotExist:
raise MyException("任务不存在")
return common.response_success()
def put(self, request, *args, **kwargs):
body = request.body
params = json.loads(body)
form = UserForm(params)
result = form.is_valid()
if result:
user = authenticate(username=form.cleaned_data["username"],
password=str(form.cleaned_data["password"]))
if user:
login(request, user)
return common.response_success()
else:
raise MyException("登录失败")
else:
print(form.errors.as_json())
raise MyException()
def post(self, request, *args, **kwargs):
"""
测试的 post
:param request:
:param args:
:param kwargs:
:return:
"""
body = request.body
print(request.body)
params = json.loads(body)
test = params.get("test", None)
if test is not None:
return common.response_success(params)
else:
return common.response_failed()
def put(self, request, pk, *args, **kwargs):
""" 更新单个服务 """
body = request.body
params = json.loads(body)
form = ServiceForm(params) # 参数校验
result = form.is_valid()
if result:
Service.objects.filter(id=pk).update(
name=form.cleaned_data['name'],
description=form.cleaned_data['description'],
parent=form.cleaned_data['parent'])
else:
print(form.errors.as_json())
raise MyException()
return common.response_success()
def post(self, request, *args, **kwargs):
body = request.body
params = json.loads(body)
form = UserForm(params)
result = form.is_valid()
if result:
user = User.objects.create_user(
username=form.cleaned_data['username'],
password=form.cleaned_data['password'])
if user:
login(request, user)
return common.response_success()
else:
raise MyException("注册失败")
else:
print(form.errors.as_json())
raise MyException()
def get(self, request, pk, *args, **kwargs):
"""
获取单个 任务版本列表
:param request:
:param pk:
:param args:
:param kwargs:
:return:
"""
result = TaskResult.objects.filter(task_id=pk).order_by('-version')
ret = []
for i in result:
tmp = dict()
tmp['version'] = i.version
tmp['task_id'] = i.task_id
tmp['created'] = i.created.strftime("%Y-%m-%d %H:%M")
tmp['id'] = i.id
ret.append(tmp)
return common.response_success(ret)
def post(self, request, *args, **kwargs):
""" 创建服务 """
body = request.body
params = json.loads(body)
form = ServiceForm(params)
result = form.is_valid()
if result:
service = Service.objects.create(
name=form.cleaned_data['name'],
description=form.cleaned_data['description'],
parent=form.cleaned_data['parent']
)
if service:
return common.response_success()
else:
raise MyException("创建失败")
else:
print(form.errors.as_json())
return common.response_failed()
def delete(self, request, pk, *args, **kwargs):
"""
任务删除接口
{
task_interface_id: 1,
}
:param request:
:param pk: task的 id
:param args:
:param kwargs:
:return:
"""
body = request.body
params = json.loads(body)
task_interface_id = params.get("task_interface_id", None)
if task_interface_id is None:
raise MyException()
TaskInterface.objects.filter(id=task_interface_id).delete()
return common.response_success()
def put(self, request, pk, *args, **kwargs):
"""
更新单个 mock
:param request:
:param pk:
:param args:
:param kwargs:
:return:
"""
body = request.body
params = json.loads(body)
form = MockForm(params) # 参数校验
result = form.is_valid()
if request:
Mock.objects.filter(id=pk).update(**form.cleaned_data)
else:
print(form.errors.as_json())
raise MyException()
return common.response_success()
def post(self, request, *args, **kwargs):
"""单个服务"""
body = request.body
params = json.loads(body)
form = UserForm(params)
result = form.is_valid()
if result:
user = User.objects.create_user(
username=form.cleaned_data["username"],
password=form.changed_data["password"])
if user:
login(request, user)
session = request.session.session_key
return common.response_success({"session": session})
else:
raise MyException("注册失败")
else:
print(form.errors.as_json())
raise MyException()
def post(self, request, *args, **kwargs):
"""
创建 mock
:param request:
:param args:
:param kwargs:
:return:
"""
body = request.body
params = json.loads(body)
form = MockForm(params)
result = form.is_valid()
if result:
mock = Mock.objects.create(**form.cleaned_data)
if mock:
return common.response_success()
else:
raise MyException("创建失败")
else:
print(form.errors.as_json())
return common.response_failed()
def get(self, request, pk, *args, **kwargs):
"""
获取单个任务的接口列表
:param request:
:param pk: # 任务的id
:param args:
:param kwargs:
:return:
"""
results = TaskInterface.objects.filter(task_id=pk)
last_result = TaskUtils.get_last_result(pk)
ret = []
for i in results:
tmp = model_to_dict(i.interface)
tmp["task_interface_id"] = i.id
r = "无"
if last_result is not None:
r = TaskUtils.get_last_interface_result(
last_result.id, i.interface_id)
tmp["result"] = r
ret.append(tmp)
return common.response_success(ret)
def post(self, request, *args, **kwargs):
"""
创建接口
:param request:
:param args:
:param kwargs:
:return:
"""
body = request.body
params = json.loads(body)
form = DebugForm(params)
result = form.is_valid()
if result:
url = form.cleaned_data["url"]
method = form.cleaned_data["method"]
header = form.cleaned_data["header"]
parameter = form.cleaned_data["parameter"]
parameter_type = form.cleaned_data["parameter_type"]
ret = InterfaceUtils.send_request(url, method, header, parameter, parameter_type)
return common.response_success(ret)
else:
print(form.errors.as_json())
return common.response_failed()
def get(self, request, *args, **kwargs):
"""
获取任务列表
:param request:
:param args:
:param kwargs:
:return:
"""
results = Task.objects.all()
ret = list()
for i in results:
tmp = dict()
tmp["id"] = i.id
tmp["name"] = i.name
tmp["description"] = i.description
tmp["status"] = i.get_status_display() # 获取choice的描述
tmp["interface_count"] = TaskInterface.objects.filter(
task_id=i.id).count
tmp["result_count"] = TaskInterface.objects.filter(
task_id=i.id).count
tmp["last_result"] = TaskUtils.get_result_summary(i.id)
ret.append(tmp)
return common.response_success(ret)
def post(self, request, pk, *args, **kwargs):
"""
执行任务
:param request:
:param pk:
:param args:
:param kwargs:
:return:
"""
try:
task = Task.objects.get(id=pk)
except Task.DoesNotExist:
raise MyException("任务不存在")
body = request.body
params = json.loads(body)
def fun(obj, arg):
runner = RunTaskUtils(obj, arg)
runner.run()
t = threading.Thread(target=fun, args=(task, params))
t.start()
return common.response_success()
def get(self, request, pk, *args, **kwargs):
"""
获取单个任务的信息
:param request:
:param pk:
:param args:
:param kwargs:
:return:
"""
try:
task = Task.objects.get(id=pk)
tmp = dict()
tmp["id"] = task.id
tmp["name"] = task.name
tmp["description"] = task.description
tmp["status"] = task.get_status_display() # 获取 choice 的描述
tmp["interface_count"] = TaskInterface.objects.filter(
task_id=task.id).count()
tmp["result_count"] = TaskResult.objects, filter(
task_id=task.id).count()
except Task.DoesNotExist:
return common.response_failed()
else:
return common.response_success(tmp)
def get(self, request, *args, **kwargs):
""" 获取服务列表 """
ret = ServiceUtils.get_service_tree_recursion(IS_ROOT)
return common.response_success(ret)
def delete(self, request, pk, *args, **kwargs):
""" 删除单个服务 """
Service.objects.filter(id=pk).delete()
return common.response_success()
