]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
Shorten tracebacks executing on connection too
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Fri, 26 Mar 2021 10:49:46 +0000 (10:49 +0000)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Fri, 26 Mar 2021 10:49:46 +0000 (10:49 +0000)
psycopg3/psycopg3/connection.py

index d95434b2e79c551b04dd6fdfb59c82c76f72fa88..3a51c605a63fe2d51219a5bd9f0ea34f49db3587 100644 (file)
@@ -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: