Return task type instance to caller

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
2019-05-14 17:03:13 -04:00
parent b687a7f519
commit 6ef9b04d57
1 changed files with 3 additions and 1 deletions
--- a/mayan/apps/task_manager/classes.py
+++ b/mayan/apps/task_manager/classes.py
@@ -128,7 +128,9 @@ class CeleryQueue(object):
        return result

    def add_task_type(self, *args, **kwargs):
-        self.task_types.append(TaskType(*args, **kwargs))
+        task_type = TaskType(*args, **kwargs)
+        self.task_types.append(task_type)
+        return task_type

    def get_active_tasks(self):
        return self._process_task_dictionary(