fix task manager priority

breatheco-de · Feb 17, 2024 · cdbaf20 · cdbaf20
1 parent acd4c26
commit cdbaf20
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/src/task_manager/__about__.py b/src/task_manager/__about__.py
@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present jefer94 <[email protected]>
 #
 # SPDX-License-Identifier: LGPL-3.0-or-later
-__version__ = "1.0.0b3"
+__version__ = "1.0.0b4"
diff --git a/src/task_manager/core/settings.py b/src/task_manager/core/settings.py
@@ -1,3 +1,4 @@
+import os
 from datetime import timedelta
 
 from .exceptions import ProgrammingError
@@ -9,8 +10,12 @@
     "RETRY_AFTER": timedelta(seconds=5),
     "DEFAULT": 5,
     "SCHEDULER": 10,
+    "TASK_MANAGER": 6,
 }
 
+if p := os.environ.get("TASK_MANAGER_PRIORITY"):
+    settings["TASK_MANAGER"] = int(p)
+
 
 def get_setting(key, default=None):
     return settings.get(key, default)

diff --git a/src/task_manager/django/tasks.py b/src/task_manager/django/tasks.py
@@ -5,7 +5,7 @@
 from celery import shared_task
 from django.utils import timezone
 
-from task_manager.core.decorators import TaskPriority
+from task_manager.core.settings import get_setting
 
 from .models import TaskManager
 
@@ -15,7 +15,7 @@
 
 
 # do not use our own task decorator
-@shared_task(bind=False, priority=TaskPriority.TASK_MANAGER.value)
+@shared_task(bind=False, priority=get_setting("TASK_MANAGER"))
 def mark_task_as_cancelled(task_manager_id):
     logger.info(f"Running mark_task_as_cancelled for {task_manager_id}")