Skip to content

Commit 24fe0f7

Browse files
committed
fix Error executing query
1 parent 0edc937 commit 24fe0f7

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

testgres/connection.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -103,14 +103,16 @@ def rollback(self):
103103

104104
def execute(self, query, *args):
105105
self.cursor.execute(query, args)
106+
if self.cursor.rowcount < 0:
107+
self.cursor.close()
108+
return None
106109
try:
107-
res = self.cursor.fetchall()
108-
# pg8000 might return tuples
109-
if isinstance(res, tuple):
110-
res = [tuple(t) for t in res]
111-
110+
res = [tuple(t) for t in self.cursor.fetchall()]
112111
return res
113-
except Exception:
112+
except ProgrammingError:
113+
return None
114+
except Exception as e:
115+
print("Error executing query: {}\n {}".format(repr(e), query))
114116
return None
115117

116118
def close(self):

0 commit comments

Comments
 (0)