From de82f4426d349494747dcb46115464203a93d619 Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Wed, 13 Jan 2021 21:54:27 -0600 Subject: [PATCH] Catch only the exceptions we expect to be raised --- src/pgwui_bulk_upload/views/bulk_upload.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/pgwui_bulk_upload/views/bulk_upload.py b/src/pgwui_bulk_upload/views/bulk_upload.py index b2c0a4e..b897fd5 100644 --- a/src/pgwui_bulk_upload/views/bulk_upload.py +++ b/src/pgwui_bulk_upload/views/bulk_upload.py @@ -111,13 +111,14 @@ class SaveBulkLine(DataLineProcessor): filepath = data.filepath try: udl = thunk() - self.cur.execute( - self.insert_map[filepath], udl.tuples) except (core_ex.DataLineError, core_ex.DBError) as exp: relation = data.relation exp.color(map_description(filepath, relation), filepath, relation) raise exp + try: + self.cur.execute( + self.insert_map[filepath], udl.tuples) except psycopg2.DatabaseError as exp: relation = data.relation myexp = core_ex.DBDataLineError(udl, exp) -- 2.34.1