From ae8d6ac303e9dc2e8d62543c76f760a4ae7cc97e Mon Sep 17 00:00:00 2001 From: Daniel Hokka Zakrisson Date: Thu, 28 Feb 2013 14:27:42 +0100 Subject: [PATCH] Ensure complex_args is considered in all action_plugins --- lib/ansible/module_common.py | 3 ++- lib/ansible/runner/action_plugins/add_host.py | 5 ++++- lib/ansible/runner/action_plugins/async.py | 2 +- lib/ansible/runner/action_plugins/copy.py | 9 ++++++--- lib/ansible/runner/action_plugins/debug.py | 5 ++++- lib/ansible/runner/action_plugins/fail.py | 5 ++++- lib/ansible/runner/action_plugins/fetch.py | 5 ++++- lib/ansible/runner/action_plugins/group_by.py | 5 ++++- lib/ansible/runner/action_plugins/pause.py | 5 ++++- lib/ansible/runner/action_plugins/template.py | 9 ++++++--- 10 files changed, 39 insertions(+), 14 deletions(-) diff --git a/lib/ansible/module_common.py b/lib/ansible/module_common.py index 82b0b5371de..635c87925f4 100644 --- a/lib/ansible/module_common.py +++ b/lib/ansible/module_common.py @@ -379,7 +379,8 @@ class AnsibleModule(object): return changed try: # FIXME: support English modes - mode = int(mode, 8) + if not isinstance(mode, int): + mode = int(mode, 8) except Exception, e: self.fail_json(path=path, msg='mode needs to be something octalish', details=str(e)) diff --git a/lib/ansible/runner/action_plugins/add_host.py b/lib/ansible/runner/action_plugins/add_host.py index a5995000611..23d5e223660 100644 --- a/lib/ansible/runner/action_plugins/add_host.py +++ b/lib/ansible/runner/action_plugins/add_host.py @@ -39,7 +39,10 @@ class ActionModule(object): if self.runner.check: return ReturnData(conn=conn, comm_ok=True, result=dict(skipped=True, msg='check mode not supported for this module')) - args = parse_kv(module_args) + args = {} + if complex_args: + args.update(complex_args) + args.update(parse_kv(module_args)) if not 'hostname' in args and not 'name' in args: raise ae("'name' is a required argument.") diff --git a/lib/ansible/runner/action_plugins/async.py b/lib/ansible/runner/action_plugins/async.py index 72266d3a1fe..dcee36684d4 100644 --- a/lib/ansible/runner/action_plugins/async.py +++ b/lib/ansible/runner/action_plugins/async.py @@ -33,7 +33,7 @@ class ActionModule(object): module_name = 'command' module_args += " #USE_SHELL" - (module_path, is_new_style, shebang) = self.runner._copy_module(conn, tmp, module_name, module_args, inject) + (module_path, is_new_style, shebang) = self.runner._copy_module(conn, tmp, module_name, module_args, inject, complex_args=complex_args) self.runner._low_level_exec_command(conn, "chmod a+rx %s" % module_path, tmp) return self.runner._execute_module(conn, tmp, 'async_wrapper', module_args, diff --git a/lib/ansible/runner/action_plugins/copy.py b/lib/ansible/runner/action_plugins/copy.py index d7cd001c7d5..60e29ff885b 100644 --- a/lib/ansible/runner/action_plugins/copy.py +++ b/lib/ansible/runner/action_plugins/copy.py @@ -32,7 +32,10 @@ class ActionModule(object): ''' handler for file transfer operations ''' # load up options - options = utils.parse_kv(module_args) + options = {} + if complex_args: + options.update(complex_args) + options.update(utils.parse_kv(module_args)) source = options.get('src', None) dest = options.get('dest', None) @@ -93,7 +96,7 @@ class ActionModule(object): # run the copy module module_args = "%s src=%s" % (module_args, tmp_src) - return self.runner._execute_module(conn, tmp, 'copy', module_args, inject=inject) + return self.runner._execute_module(conn, tmp, 'copy', module_args, inject=inject, complex_args=complex_args) else: # no need to transfer the file, already correct md5, but still need to call @@ -103,7 +106,7 @@ class ActionModule(object): module_args = "%s src=%s" % (module_args, tmp_src) if self.runner.check: module_args = "%s CHECKMODE=True" % module_args - return self.runner._execute_module(conn, tmp, 'file', module_args, inject=inject) + return self.runner._execute_module(conn, tmp, 'file', module_args, inject=inject, complex_args=complex_args) def _get_diff_data(self, conn, tmp, inject, destination, source): peek_result = self.runner._execute_module(conn, tmp, 'file', "path=%s diff_peek=1" % destination, inject=inject, persist_files=True) diff --git a/lib/ansible/runner/action_plugins/debug.py b/lib/ansible/runner/action_plugins/debug.py index 2b1c50e385a..61ed4496242 100644 --- a/lib/ansible/runner/action_plugins/debug.py +++ b/lib/ansible/runner/action_plugins/debug.py @@ -29,7 +29,10 @@ class ActionModule(object): self.runner = runner def run(self, conn, tmp, module_name, module_args, inject, complex_args=None, **kwargs): - args = utils.parse_kv(module_args) + args = {} + if complex_args: + args.update(complex_args) + args.update(utils.parse_kv(module_args)) if not 'msg' in args: args['msg'] = 'Hello world!' diff --git a/lib/ansible/runner/action_plugins/fail.py b/lib/ansible/runner/action_plugins/fail.py index ce8cbdd8d3a..7c3a58cfb91 100644 --- a/lib/ansible/runner/action_plugins/fail.py +++ b/lib/ansible/runner/action_plugins/fail.py @@ -33,7 +33,10 @@ class ActionModule(object): # note: the fail module does not need to pay attention to check mode # it always runs. - args = utils.parse_kv(module_args) + args = {} + if complex_args: + args.update(complex_args) + args.update(utils.parse_kv(module_args)) if not 'msg' in args: args['msg'] = 'Failed as requested from task' diff --git a/lib/ansible/runner/action_plugins/fetch.py b/lib/ansible/runner/action_plugins/fetch.py index 96c77067241..58aa627b0e5 100644 --- a/lib/ansible/runner/action_plugins/fetch.py +++ b/lib/ansible/runner/action_plugins/fetch.py @@ -40,7 +40,10 @@ class ActionModule(object): return ReturnData(conn=conn, comm_ok=True, result=dict(skipped=True, msg='check mode not (yet) supported for this module')) # load up options - options = utils.parse_kv(module_args) + options = {} + if complex_args: + options.update(complex_args) + options.update(utils.parse_kv(module_args)) source = options.get('src', None) dest = options.get('dest', None) if source is None or dest is None: diff --git a/lib/ansible/runner/action_plugins/group_by.py b/lib/ansible/runner/action_plugins/group_by.py index a4f9df67c21..72df4e626c2 100644 --- a/lib/ansible/runner/action_plugins/group_by.py +++ b/lib/ansible/runner/action_plugins/group_by.py @@ -37,7 +37,10 @@ class ActionModule(object): # the group_by module does not need to pay attention to check mode. # it always runs. - args = parse_kv(self.runner.module_args) + args = {} + if complex_args: + args.update(complex_args) + args.update(utils.parse_kv(module_args)) if not 'key' in args: raise ae("'key' is a required argument.") diff --git a/lib/ansible/runner/action_plugins/pause.py b/lib/ansible/runner/action_plugins/pause.py index 3a7708ce761..109f41bfce4 100644 --- a/lib/ansible/runner/action_plugins/pause.py +++ b/lib/ansible/runner/action_plugins/pause.py @@ -53,7 +53,10 @@ class ActionModule(object): # flag, it always runs hosts = ', '.join(self.runner.host_set) - args = parse_kv(template(self.runner.basedir, module_args, inject)) + args = {} + if complex_args: + args.update(complex_args) + args.update(parse_kv(template(self.runner.basedir, module_args, inject))) # Are 'minutes' or 'seconds' keys that exist in 'args'? if 'minutes' in args or 'seconds' in args: diff --git a/lib/ansible/runner/action_plugins/template.py b/lib/ansible/runner/action_plugins/template.py index 2eed900479b..dc444856790 100644 --- a/lib/ansible/runner/action_plugins/template.py +++ b/lib/ansible/runner/action_plugins/template.py @@ -37,7 +37,10 @@ class ActionModule(object): raise errors.AnsibleError("in current versions of ansible, templates are only usable in playbooks") # load up options - options = utils.parse_kv(module_args) + options = {} + if complex_args: + options.update(complex_args) + options.update(utils.parse_kv(module_args)) source = options.get('src', None) dest = options.get('dest', None) @@ -106,9 +109,9 @@ class ActionModule(object): if self.runner.check: return ReturnData(conn=conn, comm_ok=True, result=dict(changed=True), diff=dict(before_header=dest, after_header=source, before=dest_contents, after=resultant)) else: - res = self.runner._execute_module(conn, tmp, 'copy', module_args, inject=inject) + res = self.runner._execute_module(conn, tmp, 'copy', module_args, inject=inject, complex_args=complex_args) res.diff = dict(before=dest_contents, after=resultant) return res else: - return self.runner._execute_module(conn, tmp, 'file', module_args, inject=inject) + return self.runner._execute_module(conn, tmp, 'file', module_args, inject=inject, complex_args=complex_args)