Validating CLI arguments

src/sql/magic.py

@@ -43,6 +43,23 @@
 
 
 SUPPORT_INTERACTIVE_WIDGETS = ["Checkbox", "Text", "IntSlider", ""]
+recognized_arguments = [
+    "--connections",
+    "--close",
+    "--creator",
+    "--section",
+    "--persist",
+    "--persist-replace",
+    "--no-index",
+    "--append",
+    "--connection_arguments",
+    "--file",
+    "--save",
+    "--with",
+    "--no-execute",
+    "--alias",
+    "--interact",
+]
 
 
 @magics_class
@@ -311,6 +328,24 @@ def interactive_execute_wrapper(**kwargs):
 
         # %%sql {line}
         # {cell}
+
+        # Split the line into tokens separated by whitespace
+        tokens = line.split()
+        arguments = []
+
+        # Iterate through the tokens to separate arguments and SQL code
+        # If the token starts with "--", it is an argument
+        # Otherwise, it is part of the SQL code
+        for token in tokens:
+            if token.startswith("--"):
+                arguments.append(token)
+
+        for check_argument in arguments:
+            if check_argument not in recognized_arguments:
+                raise exceptions.UsageError(
+                    "Unrecognized argument(s): {}".format(check_argument)
+                )
+
         if local_ns is None:
             local_ns = {}
 

src/tests/test_magic.py

@@ -187,6 +187,11 @@ def test_persist_no_index(ip):
     assert persisted == [(1, "foo"), (2, "bar")]
 
 
+def test_unrecognized_arguments(ip):
+    result = ip.run_cell("%sql --stuff sqlite://")
+    assert "Unrecognized argument(s)" in str(result.error_in_exec)
+
+
 def test_persist_invalid_identifier(ip):
     result = ip.run_cell("%sql --persist sqlite:// not an identifier")
     assert "not a valid identifier" in str(result.error_in_exec)