def get(self, request, pk, size_code, format=None): user = self.request.user try: site = get_object_or_404(Site, pk=pk) except ObjectDoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND, data={"detail": "Not found."}) size = "-small" if size_code == '1': size = "-medium" elif size_code == '2': size = "-large" task_obj = CeleryTaskProgress.objects.create(user=user, content_object=site, task_type=6) if task_obj: task = site_download_zipfile.delay(task_obj.pk, size) task_obj.task_id = task.id task_obj.save() status, data = 200, { 'status': 'true', 'message': 'Sucess, the Zip file is being generated. You will be notified after the file is generated.' } else: status, data = 401, { 'status': 'false', 'message': 'Error occured please try again.' } return Response(data, status=status)
def get(self, request, pk, size_code, *args, **kwargs): user = self.request.user site = get_object_or_404(Site, pk=pk) size = "-small" if size_code == '1': size = "-medium" elif size_code == '2': size = "-large" task_obj = CeleryTaskProgress.objects.create(user=user, content_object=site, task_type=6) if task_obj: task = site_download_zipfile.delay(task_obj.pk, size) task_obj.task_id = task.id task_obj.save() status, data = 200, { 'status': 'true', 'message': 'Sucess, the Zip file is being generated. You will be notified after the file is generated.' } else: status, data = 401, { 'status': 'false', 'message': 'Error occured please try again.' } return JsonResponse(data, status=status)