mirror of
https://github.com/ansible/awx.git
synced 2026-02-12 23:24:48 -03:30
271383d018
* Add dispatcherctl command * Add tests for dispatcherctl command * Exit early if sqlite3 * Switch to dispatcherd mgmt cmd * Move unwanted command options to run_dispatcher * Add test for new stuff * Update the SOS report status command * make docs always reference new command * Consistently error if given config file
113 lines
4.4 KiB
Python
113 lines
4.4 KiB
Python
import os
|
|
|
|
from dispatcherd.config import setup as dispatcher_setup
|
|
|
|
from django.apps import AppConfig
|
|
from django.db import connection
|
|
from django.utils.translation import gettext_lazy as _
|
|
from awx.main.utils.common import bypass_in_test, load_all_entry_points_for
|
|
from awx.main.utils.migration import is_database_synchronized
|
|
from awx.main.utils.named_url_graph import _customize_graph, generate_graph
|
|
from awx.conf import register, fields
|
|
|
|
from awx_plugins.interfaces._temporary_private_licensing_api import detect_server_product_name
|
|
|
|
|
|
class MainConfig(AppConfig):
|
|
name = 'awx.main'
|
|
verbose_name = _('Main')
|
|
|
|
def load_named_url_feature(self):
|
|
models = [m for m in self.get_models() if hasattr(m, 'get_absolute_url')]
|
|
generate_graph(models)
|
|
_customize_graph()
|
|
register(
|
|
'NAMED_URL_FORMATS',
|
|
field_class=fields.DictField,
|
|
read_only=True,
|
|
label=_('Formats of all available named urls'),
|
|
help_text=_('Read-only list of key-value pairs that shows the standard format of all available named URLs.'),
|
|
category=_('Named URL'),
|
|
category_slug='named-url',
|
|
)
|
|
register(
|
|
'NAMED_URL_GRAPH_NODES',
|
|
field_class=fields.DictField,
|
|
read_only=True,
|
|
label=_('List of all named url graph nodes.'),
|
|
help_text=_(
|
|
'Read-only list of key-value pairs that exposes named URL graph topology.'
|
|
' Use this list to programmatically generate named URLs for resources'
|
|
),
|
|
category=_('Named URL'),
|
|
category_slug='named-url',
|
|
)
|
|
|
|
def _load_credential_types_feature(self):
|
|
"""
|
|
Create CredentialType records for any discovered credentials.
|
|
|
|
Note that Django docs advise _against_ interacting with the database using
|
|
the ORM models in the ready() path. Specifically, during testing.
|
|
However, we explicitly use the @bypass_in_test decorator to avoid calling this
|
|
method during testing.
|
|
|
|
Django also advises against running pattern because it runs everywhere i.e.
|
|
every management command. We use an advisory lock to ensure correctness and
|
|
we will deal performance if it becomes an issue.
|
|
"""
|
|
from awx.main.models.credential import CredentialType
|
|
|
|
if is_database_synchronized():
|
|
CredentialType.setup_tower_managed_defaults(app_config=self)
|
|
|
|
@bypass_in_test
|
|
def load_credential_types_feature(self):
|
|
from awx.main.models.credential import load_credentials
|
|
|
|
load_credentials()
|
|
return self._load_credential_types_feature()
|
|
|
|
def load_inventory_plugins(self):
|
|
from awx.main.models.inventory import InventorySourceOptions
|
|
|
|
is_awx = detect_server_product_name() == 'AWX'
|
|
extra_entry_point_groups = () if is_awx else ('inventory.supported',)
|
|
entry_points = load_all_entry_points_for(['inventory', *extra_entry_point_groups])
|
|
|
|
for entry_point_name, entry_point in entry_points.items():
|
|
cls = entry_point.load()
|
|
InventorySourceOptions.injectors[entry_point_name] = cls
|
|
|
|
def configure_dispatcherd(self):
|
|
"""This implements the default configuration for dispatcherd
|
|
|
|
If running the tasking service like awx-manage dispatcherd,
|
|
some additional config will be applied on top of this.
|
|
This configuration provides the minimum such that code can submit
|
|
tasks to pg_notify to run those tasks.
|
|
"""
|
|
from awx.main.dispatch.config import get_dispatcherd_config
|
|
|
|
if connection.vendor != 'postgresql':
|
|
config_dict = get_dispatcherd_config(mock_publish=True)
|
|
else:
|
|
config_dict = get_dispatcherd_config()
|
|
|
|
dispatcher_setup(config_dict)
|
|
|
|
def ready(self):
|
|
super().ready()
|
|
|
|
self.configure_dispatcherd()
|
|
|
|
"""
|
|
Credential loading triggers database operations. There are cases we want to call
|
|
awx-manage collectstatic without a database. All management commands invoke the ready() code
|
|
path. Using settings.AWX_SKIP_CREDENTIAL_TYPES_DISCOVER _could_ invoke a database operation.
|
|
"""
|
|
if not os.environ.get('AWX_SKIP_CREDENTIAL_TYPES_DISCOVER', None):
|
|
self.load_credential_types_feature()
|
|
self.load_named_url_feature()
|
|
self.load_inventory_plugins()
|