screenshot.py revision f2a4b9f6e2697ef7f6c82b7f276e9e102b61e62b
1#!/usr/bin/env python
2
3import argparse
4import sys
5
6argparser = argparse.ArgumentParser(description="Take a screenshot!",
7    epilog="I can output PNG, JPEG, GIF, and other PIL-supported formats.")
8argparser.add_argument("-c", "--crtc", type=int, default=0)
9argparser.add_argument("path")
10
11args = argparser.parse_args()
12
13# Do some evil.
14sys.path.insert(0, "/usr/local/autotest")
15
16# This import can't be moved to before the sys.path alteration.
17from cros.graphics.drm import crtcScreenshot
18
19image = crtcScreenshot(args.crtc)
20image.save(args.path)
21