Better handling of SystemExit arguments

akaihola · Jul 31, 2024 · 33f44dc · 33f44dc
1 parent d69e28c
commit 33f44dc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/darkgraylib/command_line.py b/src/darkgraylib/command_line.py
@@ -135,8 +135,11 @@ def parse_args(parser: ArgumentParser, argv: list[str]) -> Namespace:
     try:
         return parser.parse_args(argv)
     except SystemExit as exc_info:
-        exit_code, = exc_info.args
-        sys.exit(EXIT_CODE_CMDLINE_ERROR if exit_code == 2 else exit_code)
+        if exc_info.args == (2,):
+            # Change all exceptions from argparse to exit code 3 (cmdline error)
+            sys.exit(EXIT_CODE_CMDLINE_ERROR)
+        # For other exit codes, exit with the original code
+        raise
 
 
 def parse_command_line(