missing implementation of jsonpath library

This commit is contained in:
aubin
2021-05-03 10:55:24 +02:00
parent d29f8c1eb7
commit 4a9eca576d
6 changed files with 57 additions and 16 deletions

View File

@@ -64,4 +64,10 @@ options:
- The possible reasons in a condition are specific to each resource type in Kubernetes.
- See the API documentation of the status field for a given resource to see possible choices.
type: dict
wait_for:
description:
- Specifies a property on the resource to wait for.
- Ignored if C(wait) is not set or is set to False.
default: 120
type: int
'''

View File

@@ -70,7 +70,8 @@ WAIT_ARG_SPEC = dict(
status=dict(default=True, choices=[True, False, "Unknown"]),
reason=dict()
)
)
),
wait_for=dict(type='list')
)
# Map kubernetes-client parameters to ansible parameters

View File

@@ -37,6 +37,7 @@ from ansible.module_utils.six import iteritems, string_types
from ansible.module_utils._text import to_native, to_bytes, to_text
from ansible.module_utils.common.dict_transformations import dict_merge
from ansible.module_utils.parsing.convert_bool import boolean
from ansible.errors import AnsibleError
K8S_IMP_ERR = None
try:
@@ -105,6 +106,13 @@ except ImportError as e:
k8s_import_exception = e
K8S_IMP_ERR = traceback.format_exc()
try:
import jsonpath_rw as jsonpath
HAS_JSONPATH_RW = True
jsonpath_import_exception = None
except ImportError as e:
HAS_JSONPATH_RW = False
jsonpath_import_exception = e
def configuration_digest(configuration):
m = hashlib.sha256()
@@ -246,7 +254,7 @@ class K8sAnsibleMixin(object):
self.fail(msg='Failed to find exact match for {0}.{1} by [kind, name, singularName, shortNames]'.format(api_version, kind))
def kubernetes_facts(self, kind, api_version, name=None, namespace=None, label_selectors=None, field_selectors=None,
wait=False, wait_sleep=5, wait_timeout=120, state='present', condition=None):
wait=False, wait_sleep=5, wait_timeout=120, state='present', condition=None, wait_for=None):
resource = self.find_resource(kind, api_version)
api_found = bool(resource)
if not api_found:
@@ -302,7 +310,7 @@ class K8sAnsibleMixin(object):
for resource_instance in resource_list:
success, res, duration = self.wait(resource, resource_instance,
sleep=wait_sleep, timeout=wait_timeout,
state=state, condition=condition)
state=state, condition=condition, wait_for=wait_for)
if not success:
self.fail(msg="Failed to gather information about %s(s) even"
" after waiting for %s seconds" % (res.get('kind'), duration))
@@ -365,7 +373,7 @@ class K8sAnsibleMixin(object):
def fail(self, msg=None):
self.fail_json(msg=msg)
def _wait_for(self, resource, name, namespace, predicate, sleep, timeout, state):
def _wait_for(self, resource, name, namespace, predicates, sleep, timeout, state):
start = datetime.now()
def _wait_for_elapsed():
@@ -375,7 +383,7 @@ class K8sAnsibleMixin(object):
while _wait_for_elapsed() < timeout:
try:
response = resource.get(name=name, namespace=namespace)
if predicate(response):
if all([ predicate(response) for predicate in predicates ]):
if response:
return True, response.to_dict(), _wait_for_elapsed()
return True, {}, _wait_for_elapsed()
@@ -387,7 +395,7 @@ class K8sAnsibleMixin(object):
response = response.to_dict()
return False, response, _wait_for_elapsed()
def wait(self, resource, definition, sleep, timeout, state='present', condition=None):
def wait(self, resource, definition, sleep, timeout, state='present', condition=None, wait_for=None):
def _deployment_ready(deployment):
# FIXME: frustratingly bool(deployment.status) is True even if status is empty
@@ -434,22 +442,36 @@ class K8sAnsibleMixin(object):
return True
return False
def _wait_for_property(resource):
return all([ jsonpath.match("$.{}".format(item),json_data) for item in wait_for ])
def _resource_absent(resource):
return not resource
# wait_for requires jsonpath-rw library
if wait_for is not None and not HAS_JSONPATH_RW:
if hasattr(self, 'fail_json'):
self.fail_json(msg=missing_required_lib('jsonpath_rw'), error=to_native(jsonpath_import_exception))
raise AnsibleError("wait_for option requires 'jsonpath_rw' library")
waiter = dict(
Deployment=_deployment_ready,
DaemonSet=_daemonset_ready,
Pod=_pod_ready
)
kind = definition['kind']
if state == 'present' and not condition:
predicate = waiter.get(kind, lambda x: x)
elif state == 'present' and condition:
predicate = _custom_condition
predicates = []
if state == 'present':
if condition:
predicates.append(_custom_condition)
if wait_for:
# json path predicate
predicates.append(_wait_for_property)
if condition is None and wait_for is None:
predicates.append(waiter.get(kind, lambda x: x))
else:
predicate = _resource_absent
return self._wait_for(resource, definition['metadata']['name'], definition['metadata'].get('namespace'), predicate, sleep, timeout, state)
predicates.append(_resource_absent)
return self._wait_for(resource, definition['metadata']['name'], definition['metadata'].get('namespace'), predicates, sleep, timeout, state)
def set_resource_definitions(self, module):
resource_definition = module.params.get('resource_definition')
@@ -602,6 +624,7 @@ class K8sAnsibleMixin(object):
continue_on_error = self.params.get('continue_on_error')
if self.params.get('wait_condition') and self.params['wait_condition'].get('type'):
wait_condition = self.params['wait_condition']
wait_for = self.params.get('wait_for')
def build_error_msg(kind, name, msg):
return "%s %s: %s" % (kind, name, msg)
@@ -710,7 +733,7 @@ class K8sAnsibleMixin(object):
success = True
result['result'] = k8s_obj
if wait and not self.check_mode:
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition)
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition, wait_for=wait_for)
if existing:
existing = existing.to_dict()
else:
@@ -763,7 +786,7 @@ class K8sAnsibleMixin(object):
success = True
result['result'] = k8s_obj
if wait and not self.check_mode:
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition)
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition, wait_for=wait_for)
result['changed'] = True
result['method'] = 'create'
if not success:
@@ -798,7 +821,7 @@ class K8sAnsibleMixin(object):
success = True
result['result'] = k8s_obj
if wait and not self.check_mode:
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition)
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition, wait_for=wait_for)
match, diffs = self.diff_objects(existing.to_dict(), result['result'])
result['changed'] = not match
result['method'] = 'replace'
@@ -836,7 +859,7 @@ class K8sAnsibleMixin(object):
success = True
result['result'] = k8s_obj
if wait and not self.check_mode:
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition)
success, result['result'], result['duration'] = self.wait(resource, definition, wait_sleep, wait_timeout, condition=wait_condition, wait_for=wait_for)
match, diffs = self.diff_objects(existing.to_dict(), result['result'])
result['changed'] = not match
result['method'] = 'patch'

View File

@@ -136,6 +136,7 @@ requirements:
- "python >= 2.7"
- "openshift >= 0.6"
- "PyYAML >= 3.11"
- "jsonpath-rw"
'''
EXAMPLES = r'''
@@ -246,6 +247,13 @@ EXAMPLES = r'''
type: Progressing
status: Unknown
reason: DeploymentPaused
# Wait for this service to have acquired an External IP
- name: Deploy the dashboard service (lb)
kubernetes.core.k8s:
template: dash-service.yaml
wait: yes
wait_for: .status.loadBalancer.ingress[*].ip
'''
RETURN = r'''

View File

@@ -52,6 +52,7 @@ requirements:
- "python >= 2.7"
- "openshift >= 0.6"
- "PyYAML >= 3.11"
- "jsonpath-rw"
'''
EXAMPLES = r'''
@@ -164,6 +165,7 @@ def execute_module(module, k8s_ansible_mixin):
wait_sleep=module.params["wait_sleep"],
wait_timeout=module.params["wait_timeout"],
condition=module.params["wait_condition"],
wait_for=module.params["wait_for"]
)
module.exit_json(changed=False, **facts)

View File

@@ -34,6 +34,7 @@ requirements:
- "python >= 2.7"
- "openshift >= 0.6"
- "PyYAML >= 3.11"
- "jsonpath-rw"
'''
EXAMPLES = r'''