diff --git a/mayan/apps/mayan_task_manager/views.py b/mayan/apps/mayan_task_manager/views.py index 060c7b1dc2..5c348ac593 100644 --- a/mayan/apps/mayan_task_manager/views.py +++ b/mayan/apps/mayan_task_manager/views.py @@ -1,5 +1,6 @@ from __future__ import absolute_import, unicode_literals +from django.contrib import messages from django.utils.translation import ugettext_lazy as _ from common.generics import SingleObjectListView @@ -32,9 +33,19 @@ class QueueActiveTaskListView(SingleObjectListView): def get_object(self): return CeleryQueue.get(queue_name=self.kwargs['queue_name']) - def get_queryset(self): + def get_task_list(self): return self.get_object().get_active_tasks() + def get_queryset(self): + try: + return self.get_task_list() + except Exception as exception: + messages.error( + self.request, + _('Unable to retrieve task list; %s') % exception + ) + return () + class QueueScheduledTaskListView(QueueActiveTaskListView): def get_extra_context(self): @@ -44,7 +55,7 @@ class QueueScheduledTaskListView(QueueActiveTaskListView): 'title': _('Scheduled tasks in queue: %s') % self.get_object() } - def get_queryset(self): + def get_task_list(self): return self.get_object().get_scheduled_tasks() @@ -56,5 +67,5 @@ class QueueReservedTaskListView(QueueActiveTaskListView): 'title': _('Reserved tasks in queue: %s') % self.get_object() } - def get_queryset(self): + def get_task_list(self): return self.get_object().get_reserved_tasks()