예제 #1
0
파일: views.py 프로젝트: r4b3rt/vulfocus
 def get_task(self, request, pk=None):
     task_info = self.get_object()
     task_msg = task_info.task_msg
     if task_info.task_status == 1:
         return JsonResponse(
             R.running(msg="执行中", data=str(task_info.task_id)))
     task_info.is_show = True
     task_info.save()
     if task_msg:
         msg = json.loads(task_msg)
         if msg["status"] == 200:
             if not msg["data"]:
                 msg["data"] = {"_now": int(timezone.now().timestamp())}
             else:
                 msg["data"]["_now"] = int(timezone.now().timestamp())
                 try:
                     HTTP_HOST = request.META.get("HTTP_REFERER")
                     # 判断前端的请求地址是IP形式或者是域名形式
                     if HTTP_HOST.count(":") < 2:
                         HTTP_HOST = HTTP_HOST.replace("http://",
                                                       "").replace(
                                                           "https://", "")
                         origin_host = msg["data"]["host"].split(":")
                         if len(origin_host) >= 2 and HTTP_HOST:
                             msg["data"][
                                 "host"] = HTTP_HOST[:
                                                     -1] + ":" + origin_host[
                                                         1]
                 except Exception as e:
                     pass
             return JsonResponse(msg, status=200)
         else:
             return JsonResponse(msg, status=200)
     return JsonResponse(R.ok())
예제 #2
0
 def get_task(self, request, pk=None):
     task_info = self.get_object()
     task_msg = task_info.task_msg
     if task_info.task_status == 1:
         return JsonResponse(
             R.running(msg="执行中", data=str(task_info.task_id)))
     task_info.is_show = True
     task_info.save()
     if task_msg:
         msg = json.loads(task_msg)
         if msg["status"] == 200:
             return JsonResponse(msg, status=200)
         else:
             return JsonResponse(msg, status=msg["status"])
     return JsonResponse(R.ok())
예제 #3
0
 def get_task(self, request, pk=None):
     task_info = self.get_object()
     task_msg = task_info.task_msg
     if task_info.task_status == 1:
         return JsonResponse(
             R.running(msg="执行中", data=str(task_info.task_id)))
     task_info.is_show = True
     task_info.save()
     if task_msg:
         msg = json.loads(task_msg)
         if msg["status"] == 200:
             if not msg["data"]:
                 msg["data"] = {"_now": int(timezone.now().timestamp())}
             else:
                 msg["data"]["_now"] = int(timezone.now().timestamp())
             return JsonResponse(msg, status=200)
         else:
             return JsonResponse(msg, status=msg["status"])
     return JsonResponse(R.ok())