Index: build/android/screenshot.py |
diff --git a/build/android/screenshot.py b/build/android/screenshot.py |
index e1dfdb3e159223ff11a73a2555979499c607f932..86607cc0649349b5991cac456cdeb3e7e5e3e8b8 100755 |
--- a/build/android/screenshot.py |
+++ b/build/android/screenshot.py |
@@ -6,7 +6,7 @@ |
"""Takes and saves a screenshot from an Android device. |
-Usage: screenshot.py [-s SERIAL] [-f FILE] |
+Usage: screenshot.py [-s SERIAL] [[-f] FILE] |
Options: |
-s SERIAL connect to device with specified SERIAL |
@@ -20,27 +20,30 @@ import sys |
from pylib import android_commands |
-def main(argv): |
+def main(): |
# Parse options. |
- parser = OptionParser() |
+ parser = OptionParser(usage='screenshot.py [-s SERIAL] [[-f] FILE]') |
parser.add_option('-s', '--serial', dest='serial', |
help='connect to device with specified SERIAL', |
metavar='SERIAL', default=None) |
parser.add_option('-f', '--file', dest='filename', |
help='write screenshot to FILE (default: %default)', |
metavar='FILE', default='Screenshot.png') |
- (options, args) = parser.parse_args(argv) |
+ (options, args) = parser.parse_args() |
if not options.serial and len(android_commands.GetAttachedDevices()) > 1: |
parser.error('Multiple devices are attached. ' |
'Please specify SERIAL with -s.') |
+ if len(args) > 1: |
+ parser.error('Too many positional arguments.') |
+ filename = os.path.abspath(args[0] if args else options.filename) |
+ |
# Grab screenshot and write to disk. |
- filename = os.path.abspath(options.filename) |
ac = android_commands.AndroidCommands(options.serial) |
ac.TakeScreenshot(filename) |
return 0 |
if __name__ == '__main__': |
- sys.exit(main(sys.argv)) |
+ sys.exit(main()) |