Merge branch 'main' into sqlite

python · Wulian233 · Mar 27, 2025 · Jun 6, 2025 · Jun 6, 2025 · Jun 6, 2025
commit 4373d47a432e1fc56dc8e766504edfa85e69386c
@@ -7,6 +7,8 @@
 import sqlite3
 import sys
 
+from _colorize import get_theme, theme_no_color
+
 
 def execute(c, sql, suppress_errors=True, theme=theme_no_color):
     """Helper that wraps execution of SQL code.
@@ -37,12 +39,13 @@ def execute(c, sql, suppress_errors=True, theme=theme_no_color):
 class SqliteInteractiveConsole:
     """A simple SQLite REPL."""
 
-    def __init__(self, connection):
+    def __init__(self, connection, use_color=False):
         from code import InteractiveConsole
         self._con = connection
         self._cur = connection.cursor()
         self._console = InteractiveConsole()
         self._console.runsource = self.runsource
+        self._use_color = use_color
 
     def runsource(self, source, filename="<input>", symbol="single"):
         """Override runsource, the core of the InteractiveConsole REPL.