OLD | NEW |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import os | 5 import os |
6 | 6 |
7 from telemetry.core import discover | |
8 from telemetry.internal.platform import profiler | 7 from telemetry.internal.platform import profiler |
9 from telemetry.core import util | 8 from telemetry.core import util |
| 9 from telemetry.util import classes_util |
10 | 10 |
11 | 11 |
12 def _DiscoverProfilers(): | 12 def _DiscoverProfilers(): |
13 profiler_dir = os.path.dirname(__file__) | 13 profiler_dir = os.path.dirname(__file__) |
14 return discover.DiscoverClasses(profiler_dir, util.GetTelemetryDir(), | 14 return classes_util.DiscoverClasses( |
15 profiler.Profiler, | 15 profiler_dir, util.GetTelemetryDir(), profiler.Profiler) |
16 index_by_class_name=True).values() | |
17 | 16 |
18 | 17 |
19 def FindProfiler(name): | 18 def FindProfiler(name): |
20 for p in _DiscoverProfilers(): | 19 for p in _DiscoverProfilers(): |
21 if p.name() == name: | 20 if p.name() == name: |
22 return p | 21 return p |
23 return None | 22 return None |
24 | 23 |
25 | 24 |
26 def GetAllAvailableProfilers(): | 25 def GetAllAvailableProfilers(): |
27 return sorted([p.name() for p in _DiscoverProfilers() | 26 return sorted([p.name() for p in _DiscoverProfilers() |
28 if p.is_supported(browser_type='any')]) | 27 if p.is_supported(browser_type='any')]) |
OLD | NEW |