Check value of command-line options

codespell_lib/_codespell.py

@@ -348,6 +348,20 @@ def _supports_ansi_colors() -> bool:
     return False
 
 
+def _check_positive_int(value: str) -> int:
+    try:
+        i = int(value)
+    except ValueError:
+        # same message as argparse type
+        message = f"invalid {int.__name__} value: {value!r}"
+        raise argparse.ArgumentTypeError(message)
+    if i < 0:
+        # message similar to argparse choices
+        message = f"invalid choice: {value} (choose a positive integer)"
+        raise argparse.ArgumentTypeError(message)
+    return i
+
+
 def parse_options(
     args: Sequence[str],
 ) -> Tuple[argparse.Namespace, argparse.ArgumentParser, List[str]]:
@@ -557,21 +571,21 @@ def parse_options(
     parser.add_argument(
         "-A",
         "--after-context",
-        type=int,
+        type=_check_positive_int,
         metavar="LINES",
         help="print LINES of trailing context",
     )
     parser.add_argument(
         "-B",
         "--before-context",
-        type=int,
+        type=_check_positive_int,
         metavar="LINES",
         help="print LINES of leading context",
     )
     parser.add_argument(
         "-C",
         "--context",
-        type=int,
+        type=_check_positive_int,
         metavar="LINES",
         help="print LINES of surrounding context",
     )

codespell_lib/tests/test_basic.py

@@ -237,7 +237,6 @@ def test_interactivity(
     try:
         assert cs.main(fname) == 0, "empty file"
         fname.write_text("abandonned\n")
-        assert cs.main("-i", "-1", fname) == 1, "bad"
         with FakeStdin("y\n"):
             assert cs.main("-i", "3", fname) == 1
         with FakeStdin("n\n"):