allow us to specify which columns to return
authorTony Mack <tmack@cs.princeton.edu>
Mon, 30 Nov 2009 16:35:45 +0000 (16:35 +0000)
committerTony Mack <tmack@cs.princeton.edu>
Mon, 30 Nov 2009 16:35:45 +0000 (16:35 +0000)
sfa/util/genitable.py

index 2edbeba..457d36b 100644 (file)
@@ -136,8 +136,10 @@ class GeniTable(list):
         else:
             return pgdb._quote(value)
 
-    def find(self, record_filter = None):
-        sql = "SELECT * FROM %s WHERE True " % self.tablename
+    def find(self, record_filter = None, columns=None):
+        if not columns:
+            columns = "*"
+        sql = "SELECT %s FROM %s WHERE True " % (columns, self.tablename)
         
         if isinstance(record_filter, (list, tuple, set)):
             ints = filter(lambda x: isinstance(x, (int, long)), record_filter)