Merge pull request #204 from dhashe/fix-rows-affected

Don't display rows affected when the connector tells us -1

Author: amjith

CHANGELOG.md

@@ -1,3 +1,10 @@
+## TBD
+
+### Bug Fixes
+
+* Fix [misleading "0 rows affected" status for CTEs](https://github.com/dbcli/litecli/issues/203)
+  by never displaying rows affected when the connector tells us -1
+
 ## 1.14.2 - 2025-01-26
 
 ### Bug Fixes

litecli/sqlexecute.py

@@ -138,16 +138,19 @@ def get_result(self, cursor):
         # e.g. SELECT.
         if cursor.description is not None:
             headers = [x[0] for x in cursor.description]
-            status = "{0} row{1} in set"
+            status = "{count} row{s} in set"
             cursor = list(cursor)
             rowcount = len(cursor)
         else:
             _logger.debug("No rows in result.")
-            status = "Query OK, {0} row{1} affected"
-            rowcount = 0 if cursor.rowcount == -1 else cursor.rowcount
+            if cursor.rowcount == -1:
+                status = "Query OK"
+            else:
+                status = "Query OK, {count} row{s} affected"
+            rowcount = cursor.rowcount
             cursor = None
 
-        status = status.format(rowcount, "" if rowcount == 1 else "s")
+        status = status.format(count=rowcount, s="" if rowcount == 1 else "s")
 
         return (title, cursor, headers, status)