16d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpson#!/usr/bin/env python 26d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpson 3f2a4b9f6e2697ef7f6c82b7f276e9e102b61e62bCorbin Simpsonimport argparse 46d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpsonimport sys 56d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpson 6d94feb65407b07bd2c17ffd94cfcace57f9d0650Ilja H. Friedelargparser = argparse.ArgumentParser( 7d94feb65407b07bd2c17ffd94cfcace57f9d0650Ilja H. Friedel description="Take a screenshot!", 8f2a4b9f6e2697ef7f6c82b7f276e9e102b61e62bCorbin Simpson epilog="I can output PNG, JPEG, GIF, and other PIL-supported formats.") 9d94feb65407b07bd2c17ffd94cfcace57f9d0650Ilja H. Friedelargparser.add_argument("-c", "--crtc", type=int, default=0, 10d94feb65407b07bd2c17ffd94cfcace57f9d0650Ilja H. Friedel help="CRTC id (default first screen)") 11d94feb65407b07bd2c17ffd94cfcace57f9d0650Ilja H. Friedelargparser.add_argument("path", help="output image location") 126d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpson 13f2a4b9f6e2697ef7f6c82b7f276e9e102b61e62bCorbin Simpsonargs = argparser.parse_args() 146d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpson 156d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpson# Do some evil. 166d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpsonsys.path.insert(0, "/usr/local/autotest") 176d0e27ba33ae5e1cc4404a3abf5d39e5fef8a2cbCorbin Simpson 18f2a4b9f6e2697ef7f6c82b7f276e9e102b61e62bCorbin Simpson# This import can't be moved to before the sys.path alteration. 19557946d18d625f8c14e0a1bf6a23d91b17d70f1dJoe Knissfrom cros.graphics.gbm import crtcScreenshot 20f2a4b9f6e2697ef7f6c82b7f276e9e102b61e62bCorbin Simpsonimage = crtcScreenshot(args.crtc) 21f2a4b9f6e2697ef7f6c82b7f276e9e102b61e62bCorbin Simpsonimage.save(args.path) 22