From 5c063d6a89967dc4effe4e389c0342de5d397b08 Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Sat, 21 Sep 2024 13:54:47 -0500 Subject: [PATCH] Fix crash when sql execution error has no line number --- src/pgwui_sql/views/sql.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/pgwui_sql/views/sql.py b/src/pgwui_sql/views/sql.py index c169cea..b49256b 100644 --- a/src/pgwui_sql/views/sql.py +++ b/src/pgwui_sql/views/sql.py @@ -168,11 +168,14 @@ class SQLResultsHandler(pgwui_core.core.SessionDBHandler): try: cur.execute(stmt_text) except psycopg.errors.Error as err: - lineno = stmt_text.count( - '\n', 0, int(err.diag.statement_position) - 1) + 1 + stmt_pos = err.diag.statement_position + if stmt_pos is None: + lineno = '' + else: + lineno = stmt_text.count('\n', 0, int(stmt_pos) - 1) + 1 raise sql_ex.ExecutionError( err.diag.severity, - lineno, + lineno=lineno, descr=err.diag.message_primary, detail=self.format_detail(err, stmt_text)) -- 2.34.1