bug fix (--tables --columns)

stamparm · stamparm · commit 9c694ce3ecd6 · 2011-07-12T23:27:47.000Z
diff --git a/plugins/generic/enumeration.py b/plugins/generic/enumeration.py
@@ -1006,7 +1006,10 @@ def getColumns(self, onlyColNames=False, colTuple=None):
             self.getTables()
 
             if len(kb.data.cachedTables) > 0:
-                tblList = kb.data.cachedTables.values()
+                if conf.db in kb.data.cachedTables:
+                    tblList = kb.data.cachedTables[conf.db]
+                else:
+                    tblList = kb.data.cachedTables.values()
 
                 if isinstance(tblList[0], (set, tuple, list)):
                     tblList = tblList[0]
