WIP to improve Telemetry logging

telemetry/telemetry/benchmark_runner.py

 # Copyright 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 """Parses the command line, discovers the appropriate benchmarks, and runs them.
 
 Handles benchmark configuration, but all the logic for
 actually running the benchmark is in Benchmark and PageRunner."""
 
 import argparse
 import logging
 import os
 import sys
 
 from telemetry import benchmark
 from telemetry import decorators
 from telemetry.internal.browser import browser_finder
 from telemetry.internal.browser import browser_options
 from telemetry.internal.util import binary_manager
 from telemetry.internal.util import command_line
 from telemetry.internal.util import ps_util
+from telemetry.internal.util import log_stream
 from telemetry.util import matching
 
 from py_utils import discover
 
 # Right now, we only have one of each of our power perf bots. This means that
 # all eligible Telemetry benchmarks are run unsharded, which results in very
 # long (12h) cycle times. We'd like to reduce the number of tests that we run
 # on each bot drastically until we get more of the same hardware to shard tests
 # with, but we can't do so until we've verified that the hardware configuration
 # is a viable one for Chrome Telemetry tests. This is done by seeing at least
@@ skipping 248 matching lines @@
     return None
   return matched[0]
 
 
 def main(environment, extra_commands=None, **log_config_kwargs):
   # The log level is set in browser_options.
   # Clear the log handlers to ensure we can set up logging properly here.
   logging.getLogger().handlers = []
   log_config_kwargs.pop('level', None)
   log_config_kwargs.setdefault('format', DEFAULT_LOG_FORMAT)
+  stream = log_stream.InitiateLogStream()
+  log_config_kwargs.setdefault('stream', stream)
   logging.basicConfig(**log_config_kwargs)
 
   ps_util.EnableListingStrayProcessesUponExitHook()
 
   # Get the command name from the command line.
   if len(sys.argv) > 1 and sys.argv[1] == '--help':
     sys.argv[1] = 'help'
 
   command_name = 'run'
   for arg in sys.argv[1:]:
@@ skipping 46 matching lines @@
     command.ProcessCommandLineArgs(parser, options, environment)
 
   if command == Help:
     command_instance = command(all_commands)
   else:
     command_instance = command()
   if isinstance(command_instance, command_line.OptparseCommand):
     return command_instance.Run(options)
   else:
     return command_instance.Run(options, args)