Fixed to work with either normal DBAPI Row.cursor.description or pyOD…

…BC Row.cursor_description
isotoma · Jul 21, 2010 · 236cea6 · 236cea6
1 parent e48480a
commit 236cea6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyDBCLI/utils.py b/pyDBCLI/utils.py
@@ -170,7 +170,11 @@ def default(self, line):
                                                         # If this DBAPI doesn't provide a keys(), then
                                                         # try cursor.description
                                                         keys = []
-                                                        for r in row.cursor.description:
+                                                        if hasattr(row, 'cursor'):
+                                                                description = getattr(row, 'cursor').description
+                                                        else:
+                                                                description = getattr(row, 'cursor_description')
+                                                        for r in description:
                                                                 keys.append(r[0])
                                                 else:
                                                         keys = keys()