mirror of
https://github.com/RustPython/RustPython.git
synced 2026-06-02 19:39:49 +09:00
00440b179a
* Update sqlite3, dbm.sqlite3 and tests to CPython 3.14.2 * Add skip decorators for failing sqlite3 tests Skip tests that fail due to unimplemented features or behavior differences: - _iterdump not implemented (test_dump.py) - Unraisable exception handling not implemented (test_hooks.py, test_userfunctions.py) - Keyword-only arguments not supported for various methods - Autocommit behavior differences (test_transactions.py) - TransactionTests skipped due to timeout parameter type issue - Various error message differences (test_dbapi.py) - SQLITE_DBCONFIG constants not implemented - Row and Connection signature inspection issues All tests now pass with 95 skipped out of 493 total tests. * Change @unittest.skip to @unittest.expectedFailure per code review - Convert @unittest.skip decorators to @unittest.expectedFailure for tests that fail without panic/hang - Keep @unittest.skip only for TransactionTests class (setUp fails with timeout=0 int type) * fixup
140 lines
4.2 KiB
Python
Vendored
140 lines
4.2 KiB
Python
Vendored
"""A simple SQLite CLI for the sqlite3 module.
|
|
|
|
Apart from using 'argparse' for the command-line interface,
|
|
this module implements the REPL as a thin wrapper around
|
|
the InteractiveConsole class from the 'code' stdlib module.
|
|
"""
|
|
import sqlite3
|
|
import sys
|
|
|
|
from argparse import ArgumentParser
|
|
from code import InteractiveConsole
|
|
from textwrap import dedent
|
|
|
|
|
|
def execute(c, sql, suppress_errors=True):
|
|
"""Helper that wraps execution of SQL code.
|
|
|
|
This is used both by the REPL and by direct execution from the CLI.
|
|
|
|
'c' may be a cursor or a connection.
|
|
'sql' is the SQL string to execute.
|
|
"""
|
|
|
|
try:
|
|
for row in c.execute(sql):
|
|
print(row)
|
|
except sqlite3.Error as e:
|
|
tp = type(e).__name__
|
|
try:
|
|
print(f"{tp} ({e.sqlite_errorname}): {e}", file=sys.stderr)
|
|
except AttributeError:
|
|
print(f"{tp}: {e}", file=sys.stderr)
|
|
if not suppress_errors:
|
|
sys.exit(1)
|
|
|
|
|
|
class SqliteInteractiveConsole(InteractiveConsole):
|
|
"""A simple SQLite REPL."""
|
|
|
|
def __init__(self, connection):
|
|
super().__init__()
|
|
self._con = connection
|
|
self._cur = connection.cursor()
|
|
|
|
def runsource(self, source, filename="<input>", symbol="single"):
|
|
"""Override runsource, the core of the InteractiveConsole REPL.
|
|
|
|
Return True if more input is needed; buffering is done automatically.
|
|
Return False if input is a complete statement ready for execution.
|
|
"""
|
|
if not source or source.isspace():
|
|
return False
|
|
if source[0] == ".":
|
|
match source[1:].strip():
|
|
case "version":
|
|
print(f"{sqlite3.sqlite_version}")
|
|
case "help":
|
|
print("Enter SQL code and press enter.")
|
|
case "quit":
|
|
sys.exit(0)
|
|
case "":
|
|
pass
|
|
case _ as unknown:
|
|
self.write("Error: unknown command or invalid arguments:"
|
|
f' "{unknown}".\n')
|
|
else:
|
|
if not sqlite3.complete_statement(source):
|
|
return True
|
|
execute(self._cur, source)
|
|
return False
|
|
|
|
|
|
def main(*args):
|
|
parser = ArgumentParser(
|
|
description="Python sqlite3 CLI",
|
|
color=True,
|
|
)
|
|
parser.add_argument(
|
|
"filename", type=str, default=":memory:", nargs="?",
|
|
help=(
|
|
"SQLite database to open (defaults to ':memory:'). "
|
|
"A new database is created if the file does not previously exist."
|
|
),
|
|
)
|
|
parser.add_argument(
|
|
"sql", type=str, nargs="?",
|
|
help=(
|
|
"An SQL query to execute. "
|
|
"Any returned rows are printed to stdout."
|
|
),
|
|
)
|
|
parser.add_argument(
|
|
"-v", "--version", action="version",
|
|
version=f"SQLite version {sqlite3.sqlite_version}",
|
|
help="Print underlying SQLite library version",
|
|
)
|
|
args = parser.parse_args(*args)
|
|
|
|
if args.filename == ":memory:":
|
|
db_name = "a transient in-memory database"
|
|
else:
|
|
db_name = repr(args.filename)
|
|
|
|
# Prepare REPL banner and prompts.
|
|
if sys.platform == "win32" and "idlelib.run" not in sys.modules:
|
|
eofkey = "CTRL-Z"
|
|
else:
|
|
eofkey = "CTRL-D"
|
|
banner = dedent(f"""
|
|
sqlite3 shell, running on SQLite version {sqlite3.sqlite_version}
|
|
Connected to {db_name}
|
|
|
|
Each command will be run using execute() on the cursor.
|
|
Type ".help" for more information; type ".quit" or {eofkey} to quit.
|
|
""").strip()
|
|
sys.ps1 = "sqlite> "
|
|
sys.ps2 = " ... "
|
|
|
|
con = sqlite3.connect(args.filename, isolation_level=None)
|
|
try:
|
|
if args.sql:
|
|
# SQL statement provided on the command-line; execute it directly.
|
|
execute(con, args.sql, suppress_errors=False)
|
|
else:
|
|
# No SQL provided; start the REPL.
|
|
console = SqliteInteractiveConsole(con)
|
|
try:
|
|
import readline # noqa: F401
|
|
except ImportError:
|
|
pass
|
|
console.interact(banner, exitmsg="")
|
|
finally:
|
|
con.close()
|
|
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|