database cursor detection: also check for .rowcount and .fetchone

astanin · astanin · commit 86bcf75acf73 · 2016-03-21T11:30:46.000+01:00
diff --git a/tabulate.py b/tabulate.py
@@ -747,7 +747,10 @@ def _normalize_tabular_data(tabular_data, headers, showindex="default"):
                 raise ValueError('headers for a list of dicts is not a dict or a keyword')
             rows = [[row.get(k) for k in keys] for row in rows]
 
-        elif headers == "keys" and hasattr(tabular_data, "description"):
+        elif (headers == "keys"
+              and hasattr(tabular_data, "description")
+              and hasattr(tabular_data, "fetchone")
+              and hasattr(tabular_data, "rowcount")):
             # Python Database API cursor object (PEP 0249)
             # print tabulate(cursor, headers='keys')
             headers = [column[0] for column in tabular_data.description]
