mirror of
https://github.com/ansible/awx.git
synced 2026-02-14 01:34:45 -03:30
0a47d05d26
each call to schedule_task_manager becomes one of ScheduleTaskManager ScheduleDependencyManager ScheduleWorkflowManager
36 lines
907 B
Python
36 lines
907 B
Python
# Python
|
|
import logging
|
|
|
|
# Django
|
|
from django.conf import settings
|
|
|
|
# AWX
|
|
from awx import MODE
|
|
from awx.main.scheduler import TaskManager, DependencyManager, WorkflowManager
|
|
from awx.main.dispatch.publish import task
|
|
from awx.main.dispatch import get_local_queuename
|
|
|
|
logger = logging.getLogger('awx.main.scheduler')
|
|
|
|
|
|
def run_manager(manager, prefix):
|
|
if MODE == 'development' and settings.AWX_DISABLE_TASK_MANAGERS:
|
|
logger.debug(f"Not running {prefix} manager, AWX_DISABLE_TASK_MANAGERS is True. Trigger with GET to /api/debug/{prefix}_manager/")
|
|
return
|
|
manager().schedule()
|
|
|
|
|
|
@task(queue=get_local_queuename)
|
|
def task_manager():
|
|
run_manager(TaskManager, "task")
|
|
|
|
|
|
@task(queue=get_local_queuename)
|
|
def dependency_manager():
|
|
run_manager(DependencyManager, "dependency")
|
|
|
|
|
|
@task(queue=get_local_queuename)
|
|
def workflow_manager():
|
|
run_manager(WorkflowManager, "workflow")
|