Adjust for SQL that returns no result
authorKarl O. Pinc <kop@karlpinc.com>
Sun, 18 Aug 2024 19:08:53 +0000 (14:08 -0500)
committerKarl O. Pinc <kop@karlpinc.com>
Sun, 18 Aug 2024 19:08:53 +0000 (14:08 -0500)
src/pgwui_sql/views/sql.py

index bab967cab00a300685912145c7722bd65d0e6f2f..c269048b73770f8d2346f08ff85d45706a71fbd7 100644 (file)
@@ -158,12 +158,13 @@ class SQLHandler(pgwui_core.core.SessionDBHandler):
         sql_results = self.sql_results
         nextset = True
         while nextset is True:
-            first = True
-            while (row := cur.fetchone()) is not None:
-                if first:
-                    sql_results.append(SQLResult().build_header_row(cur))
-                    first = False
-                sql_results.append(SQLResult().build_data_row(row))
+            if cur.rownumber is not None:
+                first = True
+                while (row := cur.fetchone()) is not None:
+                    if first:
+                        sql_results.append(SQLResult().build_heading_row(cur))
+                        first = False
+                    sql_results.append(SQLResult().build_data_row(row))
             sql_results.append(SQLResult().build_statusmessage_row(cur))
             sql_results.append(SQLResult().build_rowcount_row(cur))
             nextset = cur.nextset()