fix Error executing query

z-kasymalieva · z-kasymalieva · commit 24fe0f7c6893 · 2024-02-01T16:38:31.000+03:00
diff --git a/testgres/connection.py b/testgres/connection.py
@@ -103,14 +103,16 @@ def rollback(self):
 
     def execute(self, query, *args):
         self.cursor.execute(query, args)
+        if self.cursor.rowcount < 0:
+            self.cursor.close()
+            return None
         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):
