From: Tony Mack <tmack@cs.princeton.edu>
Date: Wed, 4 Feb 2009 22:46:05 +0000 (+0000)
Subject: added whitelist, blacklist params to filter method
X-Git-Tag: sfa-0.9-0@14641~669
X-Git-Url: http://git.onelab.eu/?a=commitdiff_plain;h=31f218cf548f6471bb0fe7c8d92203d6891bb282;p=sfa.git

added whitelist, blacklist params to filter method
---

diff --git a/geni/util/rspec.py b/geni/util/rspec.py
index 86b50fd9..25e43a48 100644
--- a/geni/util/rspec.py
+++ b/geni/util/rspec.py
@@ -176,7 +176,7 @@ class Rspec():
         return tempdict
 
 
-    def filter(self, tagname, attribute, valuelist, dom = None):
+    def filter(self, tagname, attribute, blacklist = [], whitelist = [], dom = None):
         """
         Removes all elements where:
         1. tagname matches the element tag
@@ -188,12 +188,14 @@ class Rspec():
         if not dom:
             dom = self.rootNode
        
-        if dom.localName in [tagname] and dom.attributes.has_key(attribute) and \
-           dom.attributes.get(attribute).value in valuelist:
-            dom.parentNode.removeChild(dom)
+        if dom.localName in [tagname] and dom.attributes.has_key(attribute):
+            if whitelist and dom.attributes.get(attribute).value not in whitelist:
+                dom.parentNode.removeChild(dom)
+            if blacklist and dom.attributes.get(attribute).value in blacklist:
+                dom.parentNode.removeChild(dom)
            
         if dom.hasChildNodes():
             for child in dom.childNodes:
-                self.filter(tagname, attribute, valuelist, child) 
+                self.filter(tagname, attribute, blacklist, whitelist, child) 
 
 # vim:ts=4:expandtab