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.

1 files changed, 12 insertions, 3 deletions

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