Files
mayan-edms/apps/job_processor/views.py

154 lines
4.5 KiB
Python

from __future__ import absolute_import
from django.shortcuts import render_to_response
from django.template import RequestContext
from django.utils.translation import ugettext_lazy as _
from django.shortcuts import get_object_or_404
from django.contrib.contenttypes.models import ContentType
from django.db.models.loading import get_model
from django.http import Http404
from django.core.exceptions import PermissionDenied
from permissions.models import Permission
from common.utils import encapsulate
from acls.models import AccessEntry
from clustering.permissions import PERMISSION_NODES_VIEW
from clustering.models import Node
from .models import JobQueue
from .permissions import PERMISSION_JOB_QUEUE_VIEW
def node_workers(request, node_pk):
node = get_object_or_404(Node, pk=node_pk)
try:
Permission.objects.check_permissions(request.user, [PERMISSION_NODES_VIEW])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_NODES_VIEW, request.user, node)
context = {
'object_list': node.workers().all(),
'title': _(u'workers for node: %s') % node,
'object': node,
'hide_link': True,
'extra_columns': [
{
'name': _(u'created'),
'attribute': 'creation_datetime',
},
{
'name': _(u'heartbeat'),
'attribute': 'heartbeat',
},
{
'name': _(u'state'),
'attribute': 'get_state_display',
},
{
'name': _(u'job queue item'),
'attribute': 'job_queue_item',
},
{
'name': _(u'job type'),
'attribute': 'job_queue_item.get_job_type',
},
{
'name': _(u'job queue'),
'attribute': 'job_queue_item.job_queue',
},
],
}
return render_to_response('generic_list.html', context,
context_instance=RequestContext(request))
def job_queues(request):
# TODO: permissiong list filtering
Permission.objects.check_permissions(request.user, [PERMISSION_JOB_QUEUE_VIEW])
context = {
'object_list': JobQueue.objects.all(),
'title': _(u'job queue'),
'hide_link': True,
'extra_columns': [
{
'name': _(u'pending jobs'),
'attribute': 'pending_jobs.count',
},
{
'name': _(u'active jobs'),
'attribute': 'active_jobs.count',
},
{
'name': _(u'error jobs'),
'attribute': 'error_jobs.count',
},
],
}
return render_to_response('generic_list.html', context,
context_instance=RequestContext(request))
def job_queue_items(request, job_queue_pk, pending_jobs=False, error_jobs=False, active_jobs=False):
job_queue = get_object_or_404(JobQueue, pk=job_queue_pk)
try:
Permission.objects.check_permissions(request.user, [PERMISSION_JOB_QUEUE_VIEW])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_JOB_QUEUE_VIEW, request.user, job_queue)
jobs = set()
if pending_jobs:
jobs = job_queue.pending_jobs.all()
title = _(u'pending jobs for queue: %s') % job_queue
if error_jobs:
jobs = job_queue.error_jobs.all()
title = _(u'error jobs for queue: %s') % job_queue
if active_jobs:
jobs = job_queue.active_jobs.all()
title = _(u'active jobs for queue: %s') % job_queue
context = {
'object_list': jobs,
'title': title,
'object': job_queue,
'hide_link': True,
'extra_columns': [
{
'name': _(u'created'),
'attribute': 'creation_datetime',
},
{
'name': _(u'job type'),
'attribute': 'get_job_type',
},
{
'name': _(u'arguments'),
'attribute': 'kwargs',
},
],
}
if active_jobs:
context['extra_columns'].append(
{
'name': _(u'worker'),
'attribute': encapsulate(lambda x: x.worker or _(u'Unknown')),
}
)
if error_jobs:
context['extra_columns'].append(
{
'name': _(u'result'),
'attribute': 'result',
}
)
return render_to_response('generic_list.html', context,
context_instance=RequestContext(request))