Added event_mon.set_default_event

infra_libs/event_mon/config.py

 # Copyright 2015 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 logging
 import socket
 
 import infra_libs
 from infra_libs.event_mon.protos.chrome_infra_log_pb2 import (
   ChromeInfraEvent, ServiceEvent)
@@ skipping 147 matching lines @@
       _router = ev_router._TextStreamRouter()
     elif run_type == 'file':
       _router = ev_router._LocalFileRouter(output_file,
                                            dry_run=dry_run)
     else:
       _router = ev_router._HttpRouter(_cache,
                                       ENDPOINTS.get(run_type),
                                       dry_run=dry_run)
 
 
+def set_default_event(event):
+  """Change the default ChromeInfraEvent used to compute all events.
+
+  Args:
+    event (ChromeInfraEvent): default event
+  """
+  # Here we raise an exception because failing to set the default event
+  # could lead to invalid data in the database.
+  if not isinstance(event, ChromeInfraEvent):
+    msg = ('A ChromeInfraEvent is required as the default event. Got %s' %
+           type(event))
+    logging.error(msg)
+    raise TypeError(msg)
+
+  _cache['default_event'] = event
+
+
 def close():
   """Reset the state.
 
   Call this right before exiting the program.
 
   Returns:
     success (bool): False if an error occured
   """
   global _router, _cache
   _router = None
   _cache = {}
   return True