tools/android/loading: Archive tracks in LoadingTrace.

tools/android/loading/trace_recorder.py

 #! /usr/bin/python
 # Copyright 2016 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.
 
 """Loading trace recorder."""
 
+import argparse
+import datetime
+import json
+import logging
 import os
 import sys
+import time
 
 _SRC_DIR = os.path.abspath(os.path.join(
     os.path.dirname(__file__), '..', '..', '..'))
 
 sys.path.append(os.path.join(_SRC_DIR, 'third_party', 'catapult', 'devil'))
 from devil.android import device_utils
 
 sys.path.append(os.path.join(_SRC_DIR, 'build', 'android'))
 import devil_chromium
 
 import device_setup
 import devtools_monitor
+import loading_trace
 import page_track
-
-class AndroidTraceRecorder(object):
-  """Records a loading trace."""
-  def __init__(self, url):
-    self.url = url
-    self.devtools_connection = None
-    self.page_track = None
-
-  def Go(self, connection):
-    self.devtools_connection = connection
-    self.page_track = page_track.PageTrack(self.devtools_connection)
-    self.devtools_connection.SetUpMonitoring()
-    self.devtools_connection.SendAndIgnoreResponse(
-        'Page.navigate', {'url': self.url})
-    self.devtools_connection.StartMonitoring()
-    print self.page_track.GetEvents()
+import request_track
+import tracing
 
 
-def DoIt(url):
-  devil_chromium.Initialize()
-  devices = device_utils.DeviceUtils.HealthyDevices()
-  device = devices[0]
-  trace_recorder = AndroidTraceRecorder(url)
-  device_setup.SetUpAndExecute(device, 'chrome', trace_recorder.Go)
+def RecordAndDumpTrace(device, url, output_filename):
+  with file(output_filename, 'w') as output,\
+        device_setup.DeviceConnection(device) as connection:
+    page = page_track.PageTrack(connection)
+    request = request_track.RequestTrack(connection)
+    tracing_track = tracing.TracingTrack(connection)
+    connection.SetUpMonitoring()
+    connection.SendAndIgnoreResponse('Network.clearBrowserCache', {})
+    connection.SendAndIgnoreResponse('Page.navigate', {'url': url})
+    connection.StartMonitoring()
+    metadata = {'date': datetime.datetime.utcnow().isoformat(),
+                'seconds_since_epoch': time.time()}
+    trace = loading_trace.LoadingTrace(url, metadata, page, request,
+                                       tracing_track)
+    json.dump(trace.ToJsonDict(), output)
 
 
+def main():
+  logging.basicConfig(level=logging.INFO)
+  devil_chromium.Initialize()
+
+  parser = argparse.ArgumentParser()
+  parser.add_argument('--url', required=True)
+  parser.add_argument('--output', required=True)
+  args = parser.parse_args()
+  url = args.url
+  if not url.startswith('http'):
+    url = 'http://' + url
+  device = device_utils.DeviceUtils.HealthyDevices()[0]
+  RecordAndDumpTrace(device, url, args.output)
+

[mattcary, 2016/01/19 15:24:47]

two blank lines?

[Benoit L, 2016/01/19 15:54:50]

Done.

 if __name__ == '__main__':
-  DoIt(sys.argv[1])
+  main()