Skip to content

Commit

Permalink
Add parsing of eventual settings.provider_labels for AT, Container pr…
Browse files Browse the repository at this point in the history
…oviders
  • Loading branch information
rplevka committed Apr 29, 2024
1 parent 7f00818 commit 0f62165
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 7 deletions.
19 changes: 12 additions & 7 deletions broker/providers/ansible_tower.py
Original file line number Diff line number Diff line change
Expand Up @@ -591,19 +591,24 @@ def execute(self, **kwargs): # noqa: PLR0912,PLR0915 - Possible TODO refactor
if inventory := kwargs.pop("inventory", None):
payload["inventory"] = inventory
logger.info(f"Using tower inventory: {self._translate_inventory(inventory)}")
if labels := kwargs.pop("provider_labels", None):
payload["labels"] = self._resolve_labels(labels, target)
# record labels also as extra vars - use key=value format
kwargs["provider_labels"] = kwargs.get("provider_labels", {})
kwargs["provider_labels"].update(
{label[0]: "=".join(label[1:]) for label in labels.items()}
)

elif self.inventory:
payload["inventory"] = self.inventory
logger.info(f"Using tower inventory: {self._translate_inventory(self.inventory)}")
else:
logger.info("No inventory specified, Ansible Tower will use a default.")

# provider labels handling

provider_labels = kwargs.get("provider_labels", {})
# include eventual common labels, specified at each level of configuration
# typically imported from dynaconf env vars
provider_labels.update(settings.get("provider_labels", {}))
provider_labels.update(settings.ANSIBLETOWER.get("provider_labels", {}))
if provider_labels:
payload["labels"] = self._resolve_labels(provider_labels, target)
kwargs["provider_labels"] = provider_labels

# Save custom, non-workflow extra vars to a named variable.
# The workflow can save these values to job artifacts / host facts.
workflow_extra_vars = self._pull_extra_vars(target.extra_vars)
Expand Down
5 changes: 5 additions & 0 deletions broker/providers/container.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,6 +274,11 @@ def run_container(self, container_host, **kwargs):
if origin[1]:
envars["JENKINS_URL"] = origin[1]
kwargs["environment"] = envars

# process eventual provider labels for each setting level
kwargs["provider_labels"] = kwargs.get("provider_labels", {})
kwargs["provider_labels"].update(settings.get("provider_labels", {}))
kwargs["provider_labels"].update(settings.CONTAINER.get("provider_labels", {}))
# prefix eventual label keys with 'broker.' to conform to the docker guidelines
# https://docs.docker.com/config/labels-custom-metadata/#key-format-recommendations
kwargs["provider_labels"] = {
Expand Down

0 comments on commit 0f62165

Please sign in to comment.