CLI: fix arguments handling

Author: BoboTiG

CHANGELOG

@@ -3,7 +3,7 @@ History:
 <see Git checking messages for history>
 
 8.0.2   2023/xx/xx
-      - 
+      - CLI: fixed arguments handling
 
 8.0.1   2023/04/09
       - MSS: ensure --with-cursor, and with_cursor argument & attribute, are simple NOOP on platforms not supporting the feature

mss/__main__.py

@@ -4,15 +4,14 @@
 """
 import os.path
 from argparse import ArgumentParser
-from typing import List
 
 from . import __version__
 from .exception import ScreenShotError
 from .factory import mss
 from .tools import to_png
 
 
-def main(args: List[str], /) -> int:
+def main(*args: str) -> int:
     """Main logic."""
 
     cli_args = ArgumentParser()
@@ -83,4 +82,4 @@ def main(args: List[str], /) -> int:
 if __name__ == "__main__":  # pragma: nocover
     import sys
 
-    sys.exit(main(sys.argv[1:]))
+    sys.exit(main(*sys.argv[1:]))

mss/tests/test_implementation.py

@@ -87,7 +87,16 @@ def test_entry_point(with_cursor: bool, capsys):
     def main(*args: str, ret: int = 0) -> None:
         if with_cursor:
             args = args + ("--with-cursor",)
-        assert entry_point(args) == ret
+        assert entry_point(*args) == ret
+
+    # No arguments
+    main()
+    out, _ = capsys.readouterr()
+    for mon, line in enumerate(out.splitlines(), 1):
+        filename = f"monitor-{mon}.png"
+        assert line.endswith(filename)
+        assert os.path.isfile(filename)
+        os.remove(filename)
 
     for opt in ("-m", "--monitor"):
         main(opt, "1")
@@ -147,7 +156,7 @@ def test_entry_point_error(quiet: bool, capsys):
     def main(*args: str) -> int:
         if quiet:
             args = args + ("--quiet",)
-        return entry_point(args)
+        return entry_point(*args)
 
     if quiet:
         assert main() == 1