Add .tables for sqlite3 command-line interface

Author: tanloong

Lib/sqlite3/__main__.py

@@ -62,11 +62,28 @@ def runsource(self, source, filename="<input>", symbol="single"):
             return False
         if source[0] == ".":
             match source[1:].strip():
+                case "tables":
+                    schema_names = tuple(row[1]
+                        for row in self._cur.execute("PRAGMA database_list"))
+                    select_clauses = (f"""SELECT
+                        CASE '{schema}'
+                            WHEN 'main' THEN name
+                            ELSE CONCAT('{schema}.', name)
+                        END
+                        FROM "{schema}".sqlite_master
+                        WHERE type IN ('table', 'view')
+                        AND name NOT LIKE 'sqlite_%'"""
+                        for schema in schema_names
+                    )
+                    command = " UNION ALL ".join(select_clauses) + " ORDER BY 1"
+                    for row in self._cur.execute(command):
+                        print(row[0])
                 case "version":
                     print(sqlite3.sqlite_version)
                 case "help":
                     t = theme.syntax
                     print(f"Enter SQL code or one of the below commands, and press enter.\n\n"
+                          f"{t.builtin}.tables{t.reset}     List names of tables\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 CTRL-D\n")

Lib/test/test_sqlite3/test_cli.py

@@ -125,6 +125,27 @@ def test_interact_version(self):
         self.assertEqual(out.count(self.PS2), 0)
         self.assertIn(sqlite3.sqlite_version, out)
 
+    def test_interact_tables(self):
+        out, err = self.run_cli(commands=(
+            "CREATE TABLE table_ (id INTEGER);",
+            "CREATE TEMP TABLE temp_table (id INTEGER);",
+            "CREATE VIEW view_ AS SELECT * FROM table_;",
+            "CREATE TEMP VIEW temp_view As SELECT * FROM table_;",
+            "ATTACH ':memory:' AS attach_;",
+            "CREATE TABLE attach_.table_ (id INTEGER);",
+            "CREATE VIEW attach_.view_ AS SELECT * FROM table_;",
+            "ATTACH ':memory:' AS 123;",
+            "CREATE TABLE \"123\".table_ (id INTEGER);",
+            "CREATE VIEW \"123\".view_ AS SELECT * FROM table_;",
+            ".tables",
+            ))
+        self.assertIn(self.MEMORY_DB_MSG, err)
+        self.assertEndsWith(out, self.PS1)
+        self.assertEqual(out.count(self.PS1), 12)
+        self.assertEqual(out.count(self.PS2), 0)
+        self.assertIn("123.table_\n123.view_\nattach_.table_\nattach_.view_\n"
+                      "table_\ntemp.temp_table\ntemp.temp_view\nview_\n", out)
+
     def test_interact_empty_source(self):
         out, err = self.run_cli(commands=("", " "))
         self.assertIn(self.MEMORY_DB_MSG, err)