| #!/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, |
| help="CRTC id (default first screen)") |
| argparser.add_argument("path", help="output image location") |
| |
| 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.gbm import crtcScreenshot |
| image = crtcScreenshot(args.crtc) |
| image.save(args.path) |