<h2>Editing File: /usr/lib64/python3.12/sqlite3/__main__.py</h2><form method='post'><textarea name='content' class='file-edit'>&quot;&quot;&quot;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.
&quot;&quot;&quot;
import sqlite3
import sys

from argparse import ArgumentParser
from code import InteractiveConsole
from textwrap import dedent


def execute(c, sql, suppress_errors=True):
    &quot;&quot;&quot;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.
    &quot;&quot;&quot;

    try:
        for row in c.execute(sql):
            print(row)
    except sqlite3.Error as e:
        tp = type(e).__name__
        try:
            print(f&quot;{tp} ({e.sqlite_errorname}): {e}&quot;, file=sys.stderr)
        except AttributeError:
            print(f&quot;{tp}: {e}&quot;, file=sys.stderr)
        if not suppress_errors:
            sys.exit(1)


class SqliteInteractiveConsole(InteractiveConsole):
    &quot;&quot;&quot;A simple SQLite REPL.&quot;&quot;&quot;

    def __init__(self, connection):
        super().__init__()
        self._con = connection
        self._cur = connection.cursor()

    def runsource(self, source, filename=&quot;&lt;input&gt;&quot;, symbol=&quot;single&quot;):
        &quot;&quot;&quot;Override runsource, the core of the InteractiveConsole REPL.

        Return True if more input is needed; buffering is done automatically.
        Return False is input is a complete statement ready for execution.
        &quot;&quot;&quot;
        match source:
            case &quot;.version&quot;:
                print(f&quot;{sqlite3.sqlite_version}&quot;)
            case &quot;.help&quot;:
                print(&quot;Enter SQL code and press enter.&quot;)
            case &quot;.quit&quot;:
                sys.exit(0)
            case _:
                if not sqlite3.complete_statement(source):
                    return True
                execute(self._cur, source)
        return False


def main(*args):
    parser = ArgumentParser(
        description=&quot;Python sqlite3 CLI&quot;,
        prog=&quot;python -m sqlite3&quot;,
    )
    parser.add_argument(
        &quot;filename&quot;, type=str, default=&quot;:memory:&quot;, nargs=&quot;?&quot;,
        help=(
            &quot;SQLite database to open (defaults to ':memory:'). &quot;
            &quot;A new database is created if the file does not previously exist.&quot;
        ),
    )
    parser.add_argument(
        &quot;sql&quot;, type=str, nargs=&quot;?&quot;,
        help=(
            &quot;An SQL query to execute. &quot;
            &quot;Any returned rows are printed to stdout.&quot;
        ),
    )
    parser.add_argument(
        &quot;-v&quot;, &quot;--version&quot;, action=&quot;version&quot;,
        version=f&quot;SQLite version {sqlite3.sqlite_version}&quot;,
        help=&quot;Print underlying SQLite library version&quot;,
    )
    args = parser.parse_args(*args)

    if args.filename == &quot;:memory:&quot;:
        db_name = &quot;a transient in-memory database&quot;
    else:
        db_name = repr(args.filename)

    # Prepare REPL banner and prompts.
    if sys.platform == &quot;win32&quot; and &quot;idlelib.run&quot; not in sys.modules:
        eofkey = &quot;CTRL-Z&quot;
    else:
        eofkey = &quot;CTRL-D&quot;
    banner = dedent(f&quot;&quot;&quot;
        sqlite3 shell, running on SQLite version {sqlite3.sqlite_version}
        Connected to {db_name}

        Each command will be run using execute() on the cursor.
        Type &quot;.help&quot; for more information; type &quot;.quit&quot; or {eofkey} to quit.
    &quot;&quot;&quot;).strip()
    sys.ps1 = &quot;sqlite&gt; &quot;
    sys.ps2 = &quot;    ... &quot;

    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)
            console.interact(banner, exitmsg=&quot;&quot;)
    finally:
        con.close()

    sys.exit(0)


if __name__ == &quot;__main__&quot;:
    main(sys.argv[1:])
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Fusr%2Flib64%2Fpython3.12%2Fsqlite3'>Back to Directory</a>