[yum-cvs] yum/yum transactioninfo.py,1.35,1.36

[Jeremy Katz]

Update of /home/groups/yum/cvs/yum/yum
In directory login1.linux.duke.edu:/tmp/cvs-serv9944/yum

Modified Files:
	transactioninfo.py 
Log Message:
allow getting tsinfo members that are just in a specific state.  this is a 
bit ugly right now


Index: transactioninfo.py
===================================================================
RCS file: /home/groups/yum/cvs/yum/yum/transactioninfo.py,v
retrieving revision 1.35
retrieving revision 1.36
diff -u -r1.35 -r1.36
--- transactioninfo.py	16 Feb 2007 06:06:32 -0000	1.35
+++ transactioninfo.py	19 Feb 2007 22:40:04 -0000	1.36
@@ -51,19 +51,24 @@
             print msg
 
 
-    def getMembers(self, pkgtup=None):
+    def getMembers(self, pkgtup=None, output_states=None):
         """takes an optional package tuple and returns all transaction members 
            matching, no pkgtup means it returns all transaction members"""
         
         if pkgtup is None:
             returnlist = []
             for key in self.pkgdict.keys():
-                returnlist.extend(self.pkgdict[key])
-                
+                for p in self.pkgdict[key]:
+                    if not output_states or p.output_state in output_states:
+                        returnlist.append(p)
             return returnlist
 
         if self.pkgdict.has_key(pkgtup):
-            return self.pkgdict[pkgtup]
+            ret = []
+            for p in self.pkgdict[pkgtup]:
+                if not output_states or p.output_state in output_states:
+                    ret.append(p)
+            return ret
         else:
             return []