[telemetry] Add --output option to output results to a file

R=tonyg
NOTRY=True

Review URL: https://chromiumcodereview.appspot.com/11875008

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@176566 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 13 insertions, 2 deletions

diff --git a/tools/telemetry/telemetry/multi_page_benchmark_runner.py b/tools/telemetry/telemetry/multi_page_benchmark_runner.py
index 1acb558..dc0cb8d 100755
--- a/tools/telemetry/telemetry/multi_page_benchmark_runner.py
+++ b/tools/telemetry/telemetry/multi_page_benchmark_runner.py
@@ -41,6 +41,9 @@ def Main(benchmark_dir):
                     default='csv',
                     help='Output format. Can be "csv" or "block". '
                     'Defaults to "%default".')
+  parser.add_option('-o', '--output',
+                    dest='output_file',
+                    help='Redirects output to a file. Defaults to stdout.')
 
   benchmark = None
   if benchmark_name is not None:
@@ -68,12 +71,20 @@ def Main(benchmark_dir):
 Use --browser=list to figure out which are available.\n"""
     sys.exit(1)
 
+  if not options.output_file:
+    output_file = sys.stdout
+  elif options.output_file == '-':
+    output_file = sys.stdout
+  else:
+    output_file = open(options.output_file, 'w')
+
   if options.output_format == 'csv':
     results = csv_page_benchmark_results.CsvPageBenchmarkResults(
-      csv.writer(sys.stdout),
+      csv.writer(output_file),
       benchmark.results_are_the_same_on_every_page)
   elif options.output_format in ('block', 'terminal-block'):
-    results = block_page_benchmark_results.BlockPageBenchmarkResults(sys.stdout)
+    results = block_page_benchmark_results.BlockPageBenchmarkResults(
+      output_file)
   else:
     raise Exception('Invalid --output-format value: "%s". Valid values are '
                     '"csv" and "block".'