mirror of
https://github.com/ansible/awx.git
synced 2026-01-29 07:14:43 -03:30
c1f0a831ff
* pull the correct collection plugin for the product * remove unused import and logging line * refactor code to load entry points * reformat method * lint fix * renames for clarity and a lint fix * move function to utils * move the rest of the code into load_inventory_plugins * temp - confirm that tests will pass * revert change caught in merge * change back requirement the related PR has been merged
88 lines
3.5 KiB
Python
88 lines
3.5 KiB
Python
import os
|
|
|
|
from django.apps import AppConfig
|
|
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):
|
|
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 ready(self):
|
|
super().ready()
|
|
|
|
"""
|
|
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()
|