84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
"""This is part of the MSS Python's module.
|
|
Source: https://github.com/BoboTiG/python-mss.
|
|
"""
|
|
|
|
import os.path
|
|
import sys
|
|
from argparse import ArgumentParser
|
|
|
|
from mss import __version__
|
|
from mss.exception import ScreenShotError
|
|
from mss.factory import mss
|
|
from mss.tools import to_png
|
|
|
|
|
|
def main(*args: str) -> int:
|
|
"""Main logic."""
|
|
cli_args = ArgumentParser(prog="mss")
|
|
cli_args.add_argument(
|
|
"-c",
|
|
"--coordinates",
|
|
default="",
|
|
type=str,
|
|
help="the part of the screen to capture: top, left, width, height",
|
|
)
|
|
cli_args.add_argument(
|
|
"-l",
|
|
"--level",
|
|
default=6,
|
|
type=int,
|
|
choices=list(range(10)),
|
|
help="the PNG compression level",
|
|
)
|
|
cli_args.add_argument("-m", "--monitor", default=0, type=int, help="the monitor to screenshot")
|
|
cli_args.add_argument("-o", "--output", default="monitor-{mon}.png", help="the output file name")
|
|
cli_args.add_argument("--with-cursor", default=False, action="store_true", help="include the cursor")
|
|
cli_args.add_argument(
|
|
"-q",
|
|
"--quiet",
|
|
default=False,
|
|
action="store_true",
|
|
help="do not print created files",
|
|
)
|
|
cli_args.add_argument("-v", "--version", action="version", version=__version__)
|
|
|
|
options = cli_args.parse_args(args or None)
|
|
kwargs = {"mon": options.monitor, "output": options.output}
|
|
if options.coordinates:
|
|
try:
|
|
top, left, width, height = options.coordinates.split(",")
|
|
except ValueError:
|
|
print("Coordinates syntax: top, left, width, height")
|
|
return 2
|
|
|
|
kwargs["mon"] = {
|
|
"top": int(top),
|
|
"left": int(left),
|
|
"width": int(width),
|
|
"height": int(height),
|
|
}
|
|
if options.output == "monitor-{mon}.png":
|
|
kwargs["output"] = "sct-{top}x{left}_{width}x{height}.png"
|
|
|
|
try:
|
|
with mss(with_cursor=options.with_cursor) as sct:
|
|
if options.coordinates:
|
|
output = kwargs["output"].format(**kwargs["mon"])
|
|
sct_img = sct.grab(kwargs["mon"])
|
|
to_png(sct_img.rgb, sct_img.size, level=options.level, output=output)
|
|
if not options.quiet:
|
|
print(os.path.realpath(output))
|
|
else:
|
|
for file_name in sct.save(**kwargs):
|
|
if not options.quiet:
|
|
print(os.path.realpath(file_name))
|
|
return 0
|
|
except ScreenShotError:
|
|
if options.quiet:
|
|
return 1
|
|
raise
|
|
|
|
|
|
if __name__ == "__main__": # pragma: nocover
|
|
sys.exit(main())
|