From 2272985a2dde3201bc5c2a6db617edaaa7c98534 Mon Sep 17 00:00:00 2001 From: Kenny Woodson Date: Thu, 29 Jan 2015 15:18:35 -0500 Subject: Add filters to oo_collect filter so that we can pick and choose what we want to receive back based on more than a single attribute. --- filter_plugins/oo_filters.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) (limited to 'filter_plugins') diff --git a/filter_plugins/oo_filters.py b/filter_plugins/oo_filters.py index 703506b88..b57056375 100644 --- a/filter_plugins/oo_filters.py +++ b/filter_plugins/oo_filters.py @@ -31,10 +31,16 @@ def get_attr(data, attribute=None): return ptr -def oo_collect(data, attribute=None): +def oo_collect(data, attribute=None, filters={}): ''' This takes a list of dict and collects all attributes specified into a list - Ex: data = [ {'a':1,'b':5}, {'a':2}, {'a':3} ] + If filter is specified then we will include all items that match _ALL_ of filters. + Ex: data = [ {'a':1, 'b':5, 'z': 'z'}, # True, return + {'a':2, 'z': 'z'}, # True, return + {'a':3, 'z': 'z'}, # True, return + {'a':4, 'z': 'b'}, # FAILED, obj['z'] != obj['z'] + ] attribute = 'a' + filters = {'z': 'z'} returns [1, 2, 3] ''' @@ -44,7 +50,10 @@ def oo_collect(data, attribute=None): if not attribute: raise errors.AnsibleFilterError("|failed expects attribute to be set") - retval = [get_attr(d, attribute) for d in data] + if filters: + retval = [get_attr(d, attribute) for d in data if all([ d[key] == filters[key] for key in filters ]) ] + else: + retval = [get_attr(d, attribute) for d in data] return retval -- cgit v1.2.1