WIP to improve Telemetry logging

telemetry/telemetry/internal/util/log_stream.py

+# Copyright 2017 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.
+
+import cStringIO
+
+
+_LOG_STREAM = None
+
+
+def InitiateLogStream():
+  global _LOG_STREAM
+  assert _LOG_STREAM is None, 'Already initialized'
+  _LOG_STREAM = cStringIO.StringIO()
+  return _LOG_STREAM
+
+
+def TruncateLogStream():
+  if not _LOG_STREAM:
+    return
+  _LOG_STREAM.truncate(0)
+
+
+def FlushLogStream(outstream):
+  if not _LOG_STREAM:
+    return
+  print >> outstream, _LOG_STREAM.getvalue()
+  TruncateLogStream()