예제 #1
0
 def start_container(self, request, pk=None):
     """
     启动靶场
     :param request:
     :param pk:
     :return:
     """
     img_info = self.get_object()
     # 当前用户登录ID
     user = request.user
     image_id = img_info.image_id
     user_id = user.id
     container_vul = ContainerVul.objects.filter(user_id=user_id,
                                                 image_id=image_id,
                                                 time_model_id="").first()
     if not container_vul:
         container_vul = ContainerVul(
             image_id=img_info,
             user_id=user_id,
             vul_host="",
             container_status="stop",
             docker_container_id="",
             vul_port="",
             container_port="",
             time_model_id="",
             create_date=django.utils.timezone.now(),
             container_flag="")
         container_vul.save()
     task_id = tasks.create_container_task(container_vul, user,
                                           get_request_ip(request))
     return JsonResponse(R.ok(task_id))
예제 #2
0
파일: views.py 프로젝트: exexute/vulfocus
 def start_container(self, request, pk=None):
     """
     启动容器
     :param request:
     :param pk:
     :return:
     """
     user_info = request.user
     container_vul = self.get_object()
     task_id = tasks.create_container_task(container_vul=container_vul, user_info=user_info,
                                           request_ip=get_request_ip(request))
     return JsonResponse(R.ok(task_id))