From edc04d61f3b51233a11a5b3aea14da3ac11acef0 Mon Sep 17 00:00:00 2001 From: Michael DeHaan Date: Mon, 11 Mar 2013 17:47:57 -0400 Subject: [PATCH] Added 'with_nested' lookup plugin that can easily do nested loops. --- lib/ansible/runner/lookup_plugins/nested.py | 58 +++++++++++++++++++++ 1 file changed, 58 insertions(+) create mode 100644 lib/ansible/runner/lookup_plugins/nested.py diff --git a/lib/ansible/runner/lookup_plugins/nested.py b/lib/ansible/runner/lookup_plugins/nested.py new file mode 100644 index 0000000000..40070237f2 --- /dev/null +++ b/lib/ansible/runner/lookup_plugins/nested.py @@ -0,0 +1,58 @@ +# (c) 2012, Michael DeHaan +# +# This file is part of Ansible +# +# Ansible is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Ansible is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with Ansible. If not, see . + + +def flatten(terms): + ret = [] + for term in terms: + if isinstance(term, list): + ret.extend(term) + elif isinstance(term, tuple): + ret.extend(term) + else: + ret.append(term) + return ret + +def combine(a,b): + results = [] + for x in a: + for y in b: + results.append([x,y]) + return results + +class LookupModule(object): + + def __init__(self, **kwargs): + pass + + def run(self, terms, **kwargs): + if not isinstance(terms, list): + raise errors.AnsibleError("a list is required for with_nested") + my_list = terms[:] + result = [] + if len(my_list) == 0: + raise errors.AnsibleError("with_nested requires at least one list") + result = my_list.pop() + while len(my_list) > 0: + result2 = combine(result, my_list.pop()) + result = result2 + new_result = [] + for x in result: + new_result.append(flatten(x)) + return new_result + +