OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2013 The Chromium Authors. All rights reserved. | 2 # Copyright 2013 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 import os | 6 import os |
7 import sys | 7 import sys |
8 import tempfile | 8 import tempfile |
9 | 9 |
10 sys.path.append(os.path.join( | 10 sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir, 'telemetry')) |
11 os.path.dirname(os.path.realpath(__file__)), os.pardir, 'telemetry')) | |
12 | 11 |
13 from telemetry.core import browser_finder | 12 from telemetry.core import browser_finder |
14 from telemetry.core import browser_options | 13 from telemetry.core import browser_options |
15 | 14 |
16 | 15 |
17 def _RunPrebuilt(options): | 16 def _RunPrebuilt(options): |
18 browser_to_create = browser_finder.FindBrowser(options) | 17 browser_to_create = browser_finder.FindBrowser(options) |
19 with browser_to_create.Create() as browser: | 18 with browser_to_create.Create() as browser: |
20 browser.Start() | 19 browser.Start() |
21 output_file = os.path.join(tempfile.mkdtemp(), options.profiler) | 20 output_file = os.path.join(tempfile.mkdtemp(), options.profiler) |
22 raw_input('Press enter to start profiling...') | 21 raw_input('Press enter to start profiling...') |
23 print '>> Starting profiler', options.profiler | 22 print '>> Starting profiler', options.profiler |
24 browser.StartProfiling(options.profiler, output_file) | 23 browser.StartProfiling(options.profiler, output_file) |
25 print 'Press enter or CTRL+C to stop' | 24 print 'Press enter or CTRL+C to stop' |
26 try: | 25 try: |
27 raw_input() | 26 raw_input() |
28 except KeyboardInterrupt: | 27 except KeyboardInterrupt: |
29 pass | 28 pass |
30 finally: | 29 finally: |
31 browser.StopProfiling() | 30 browser.StopProfiling() |
32 print '<< Stopped profiler ', options.profiler | 31 print '<< Stopped profiler ', options.profiler |
33 | 32 |
34 | 33 |
35 if __name__ == '__main__': | 34 if __name__ == '__main__': |
36 browser_finder_options = browser_options.BrowserFinderOptions() | 35 browser_finder_options = browser_options.BrowserFinderOptions() |
37 parser = browser_finder_options.CreateParser('') | 36 parser = browser_finder_options.CreateParser('') |
38 profiler_options, _ = parser.parse_args() | 37 profiler_options, _ = parser.parse_args() |
39 sys.exit(_RunPrebuilt(profiler_options)) | 38 sys.exit(_RunPrebuilt(profiler_options)) |
OLD | NEW |