fix Error executing query (#110)

z-kasymalieva · web-flow · commit 70d2f276a6ce · 2024-02-02T11:34:13.000+01:00
diff --git a/testgres/connection.py b/testgres/connection.py
@@ -104,13 +104,13 @@ def rollback(self):
     def execute(self, query, *args):
         self.cursor.execute(query, args)
         try:
-            res = self.cursor.fetchall()
             # pg8000 might return tuples
-            if isinstance(res, tuple):
-                res = [tuple(t) for t in res]
-
+            res = [tuple(t) for t in self.cursor.fetchall()]
             return res
-        except Exception:
+        except ProgrammingError:
+            return None
+        except Exception as e:
+            print("Error executing query: {}\n {}".format(repr(e), query))
             return None
 
     def close(self):
