mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-09 20:26:33 +00:00
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Local action plugin that extends ansible's uri action plugin with centralized
|
|
# retry defaults while preserving full uri feature compatibility.
|
|
|
|
from __future__ import annotations
|
|
|
|
import time
|
|
|
|
from ansible.plugins.action.uri import ActionModule as UriActionModule
|
|
|
|
|
|
class ActionModule(UriActionModule):
|
|
# Default policy when no task keywords are provided.
|
|
DEFAULT_RETRIES = 30
|
|
DEFAULT_DELAY = 2
|
|
|
|
def run(self, tmp=None, task_vars=None):
|
|
if task_vars is None:
|
|
task_vars = {}
|
|
|
|
# Respect explicit Ansible until/retry logic to avoid nested loops.
|
|
if self._task_keyword_is_set("until"):
|
|
return super(ActionModule, self).run(tmp=tmp, task_vars=task_vars)
|
|
|
|
if self._task.async_val:
|
|
return super(ActionModule, self).run(tmp=tmp, task_vars=task_vars)
|
|
|
|
retries = self._task_keyword_int("retries", self.DEFAULT_RETRIES)
|
|
delay = self._task_keyword_int("delay", self.DEFAULT_DELAY)
|
|
|
|
attempts_total = max(1, retries + 1)
|
|
last_result = None
|
|
for attempt in range(1, attempts_total + 1):
|
|
last_result = super(ActionModule, self).run(tmp=tmp, task_vars=task_vars)
|
|
if not last_result.get("failed", False):
|
|
last_result["attempts"] = attempt
|
|
return last_result
|
|
|
|
if attempt < attempts_total:
|
|
time.sleep(delay)
|
|
|
|
last_result["attempts"] = attempts_total
|
|
return last_result
|
|
|
|
def _task_keyword_is_set(self, key):
|
|
ds = self._task.get_ds()
|
|
return isinstance(ds, dict) and key in ds
|
|
|
|
def _task_keyword_int(self, key, fallback):
|
|
# `retries` can be present on the parsed task object even if it is not
|
|
# visible in get_ds(); prefer the task attribute for reliable override.
|
|
if key == "retries":
|
|
retries_value = getattr(self._task, "retries", None)
|
|
if retries_value is None:
|
|
return fallback
|
|
return self._as_int(retries_value, fallback)
|
|
|
|
if not self._task_keyword_is_set(key):
|
|
return fallback
|
|
|
|
return self._as_int(getattr(self._task, key, None), fallback)
|
|
|
|
@staticmethod
|
|
def _as_int(value, fallback):
|
|
try:
|
|
return int(value)
|
|
except (TypeError, ValueError):
|
|
return fallback
|