From: Karl O. Pinc <kop@karlpinc.com>
Date: Sat, 21 Sep 2024 18:54:47 +0000 (-0500)
Subject: Fix crash when sql execution error has no line number
X-Git-Url: https://papio.biology.duke.edu/gitweb/?a=commitdiff_plain;h=5c063d6a89967dc4effe4e389c0342de5d397b08;p=pgwui_sql

Fix crash when sql execution error has no line number
---

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))