From 6b384ed9f642d435168846423915558baeab2114 Mon Sep 17 00:00:00 2001 From: Daniele Varrazzo Date: Fri, 26 Mar 2021 10:49:46 +0000 Subject: [PATCH] Shorten tracebacks executing on connection too --- psycopg3/psycopg3/connection.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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: -- 2.47.2