From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Fri, 26 Mar 2021 10:49:46 +0000 (+0000)
Subject: Shorten tracebacks executing on connection too
X-Git-Tag: 3.0.dev0~77
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=6b384ed9f642d435168846423915558baeab2114;p=thirdparty%2Fpsycopg.git

Shorten tracebacks executing on connection too
---

diff --git a/psycopg3/psycopg3/connection.py b/psycopg3/psycopg3/connection.py
index d95434b2e..3a51c605a 100644
--- a/psycopg3/psycopg3/connection.py
+++ b/psycopg3/psycopg3/connection.py
@@ -537,7 +537,10 @@ class Connection(BaseConnection):
     ) -> Cursor:
         """Execute a query and return a cursor to read its results."""
         cur = self.cursor()
-        return cur.execute(query, params, prepare=prepare)
+        try:
+            return cur.execute(query, params, prepare=prepare)
+        except e.Error as ex:
+            raise ex.with_traceback(None)
 
     def commit(self) -> None:
         """Commit any pending transaction to the database."""
@@ -713,7 +716,10 @@ class AsyncConnection(BaseConnection):
         prepare: Optional[bool] = None,
     ) -> AsyncCursor:
         cur = self.cursor()
-        return await cur.execute(query, params, prepare=prepare)
+        try:
+            return await cur.execute(query, params, prepare=prepare)
+        except e.Error as ex:
+            raise ex.with_traceback(None)
 
     async def commit(self) -> None:
         async with self.lock: