You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

83 lines
2.8 KiB
Python

# Copyright (C) 2018 Intel Corporation
#
# SPDX-License-Identifier: MIT
from django.http import HttpResponseBadRequest, JsonResponse
from rules.contrib.views import permission_required, objectgetter
from cvat.apps.authentication.decorators import login_required
from cvat.apps.engine.log import slogger
from cvat.apps.engine import models
from cvat.apps.git.models import GitData
import cvat.apps.git.git as CVATGit
import django_rq
@login_required
def check_process(request, rq_id):
try:
queue = django_rq.get_queue('default')
rq_job = queue.fetch_job(rq_id)
if rq_job is not None:
if rq_job.is_queued or rq_job.is_started:
return JsonResponse({"status": "processing"})
elif rq_job.is_finished:
return JsonResponse({"status": "finished"})
else:
return JsonResponse({"status": "failed", "error": rq_job.exc_info})
else:
return JsonResponse({"status": "unknown"})
except Exception as ex:
slogger.glob.error("error occured during checking repository request with rq id {}".format(rq_id), exc_info=True)
return HttpResponseBadRequest(str(ex))
@login_required
@permission_required(perm=['engine.task.delete'],
fn=objectgetter(models.Task, 'tid'), raise_exception=True)
def push_repository(request, tid):
try:
slogger.task[tid].info("push repository request")
rq_id = "git.push.{}".format(tid)
queue = django_rq.get_queue('default')
queue.enqueue_call(func = CVATGit.push, args = (tid, request.user, request.scheme, request.get_host()), job_id = rq_id)
return JsonResponse({ "rq_id": rq_id })
except Exception as ex:
try:
slogger.task[tid].error("error occured during pushing repository request", exc_info=True)
except Exception:
pass
return HttpResponseBadRequest(str(ex))
@login_required
@permission_required(perm=['engine.task.access'],
fn=objectgetter(models.Task, 'tid'), raise_exception=True)
def get_repository(request, tid):
try:
slogger.task[tid].info("get repository request")
return JsonResponse(CVATGit.get(tid, request.user))
except Exception as ex:
try:
slogger.task[tid].error("error occured during getting repository info request", exc_info=True)
except Exception:
pass
return HttpResponseBadRequest(str(ex))
@login_required
def get_meta_info(request):
try:
db_git_records = GitData.objects.all()
response = {}
for db_git in db_git_records:
response[db_git.task_id] = db_git.status
return JsonResponse(response, safe = False)
except Exception as ex:
slogger.glob.exception("error occured during get meta request", exc_info = True)
return HttpResponseBadRequest(str(ex))