xref: /external/autotest/client/bin/screenshot.py

#!/usr/bin/env python

import argparse
import sys

argparser = argparse.ArgumentParser(description="Take a screenshot!",
    epilog="I can output PNG, JPEG, GIF, and other PIL-supported formats.")
argparser.add_argument("-c", "--crtc", type=int, default=0)
argparser.add_argument("path")

args = argparser.parse_args()

# Do some evil.
sys.path.insert(0, "/usr/local/autotest")

# This import can't be moved to before the sys.path alteration.
from cros.graphics.drm import crtcScreenshot

image = crtcScreenshot(args.crtc)
image.save(args.path)