Support spaces after dot

Author: tanloong

Lib/sqlite3/__main__.py

@@ -48,18 +48,18 @@ def runsource(self, source, filename="<input>", symbol="single"):
         Return True if more input is needed; buffering is done automatically.
         Return False if input is a complete statement ready for execution.
         """
-        source = source.rstrip()
-        match source:
-            case ".version":
-                print(f"{sqlite3.sqlite_version}")
-            case ".help":
-                print("Enter SQL code and press enter.")
-            case ".quit":
-                sys.exit(0)
-            case _:
-                if not sqlite3.complete_statement(source):
-                    return True
-                execute(self._cur, source)
+        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)
+        else:
+            if not sqlite3.complete_statement(source):
+                return True
+            execute(self._cur, source)
         return False
 
 

Lib/test/test_sqlite3/test_cli.py

@@ -116,12 +116,12 @@ def test_interact_version(self):
         self.assertEqual(out.count(self.PS2), 0)
         self.assertIn(sqlite3.sqlite_version, out)
 
-    def test_interact_whitespace_stripping(self):
-        out, err = self.run_cli(commands=(".version" + " ",))
+    def test_interact_dot_commands_whitespace(self):
+        out, err = self.run_cli(commands=(".version ", ". version"))
         self.assertIn(self.MEMORY_DB_MSG, err)
-        self.assertIn(sqlite3.sqlite_version + "\n", out)
+        self.assertEqual(out.count(sqlite3.sqlite_version + "\n"), 2)
         self.assertEndsWith(out, self.PS1)
-        self.assertEqual(out.count(self.PS1), 2)
+        self.assertEqual(out.count(self.PS1), 3)
         self.assertEqual(out.count(self.PS2), 0)
         self.assertIn(sqlite3.sqlite_version, out)