Make minor changes to add needed imports

This commit is contained in:
Alan Rominger 2021-07-12 15:32:17 -04:00
parent b10a8b0fa9
commit 39e23db523
No known key found for this signature in database
GPG Key ID: C2D7EAAA12B63559
2 changed files with 4 additions and 1 deletions

View File

@ -2,6 +2,7 @@
# All Rights Reserved.
from decimal import Decimal
import random
from django.core.validators import MinValueValidator
from django.db import models, connection

View File

@ -106,7 +106,7 @@ from awx.main.utils.common import (
get_cpu_capacity,
get_system_task_capacity,
)
from awx.main.utils.execution_environments import get_default_pod_spec, CONTAINER_ROOT, to_container_path
from awx.main.utils.execution_environments import get_default_execution_environment, get_default_pod_spec, CONTAINER_ROOT, to_container_path
from awx.main.utils.ansible import read_ansible_config
from awx.main.utils.external_logging import reconfigure_rsyslog
from awx.main.utils.safe_yaml import safe_dump, sanitize_jinja
@ -408,6 +408,7 @@ def cleanup_execution_environment_images():
if process.returncode != 0:
logger.debug(f"Failed to delete image {image_name}")
@task(queue=get_local_queuename)
def check_heartbeat(node):
AWXReceptorJob.check_heartbeat(node)
@ -2835,6 +2836,7 @@ class RunAdHocCommand(BaseTask):
params['process_isolation'] = False if MODE == 'development' else True
return params
@task(queue=get_local_queuename)
class RunSystemJob(BaseTask):