use a global const to store the EOF key

Signed-off-by: Keming <[email protected]>

Author: kemingy

Lib/sqlite3/__main__.py

@@ -15,6 +15,9 @@
 from ._completer import completer
 
 
+EOF_KEY = "CTRL-Z" if sys.platform == "win32" else "CTRL-D"
+
+
 def execute(c, sql, suppress_errors=True, theme=theme_no_color):
     """Helper that wraps execution of SQL code.
 
@@ -41,13 +44,6 @@ def execute(c, sql, suppress_errors=True, theme=theme_no_color):
             sys.exit(1)
 
 
-def _eof_key():
-    if sys.platform == "win32" and "idlelib.run" not in sys.modules:
-        return "CTRL-Z"
-    else:
-        return "CTRL-D"
-
-
 class SqliteInteractiveConsole(InteractiveConsole):
     """A simple SQLite REPL."""
 
@@ -76,7 +72,7 @@ def runsource(self, source, filename="<input>", symbol="single"):
                     print(f"Enter SQL code or one of the below commands, and press enter.\n\n"
                           f"{t.builtin}.version{t.reset}    Print underlying SQLite library version\n"
                           f"{t.builtin}.help{t.reset}       Print this help message\n"
-                          f"{t.builtin}.quit{t.reset}       Exit the CLI, equivalent to {_eof_key()}\n")
+                          f"{t.builtin}.quit{t.reset}       Exit the CLI, equivalent to {EOF_KEY}\n")
                 case "quit":
                     sys.exit(0)
                 case "":
@@ -129,7 +125,7 @@ def main(*args):
         Connected to {db_name}
 
         Each command will be run using execute() on the cursor.
-        Type ".help" for more information; type ".quit" or {_eof_key()} to quit.
+        Type ".help" for more information; type ".quit" or {EOF_KEY} to quit.
     """).strip()
 
     theme = get_theme()