OLD | NEW |
1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 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 inspect | 5 import inspect |
6 import os | 6 import os |
7 import sys | 7 import sys |
8 | 8 |
9 from telemetry.core import discover | |
10 from telemetry import story | 9 from telemetry import story |
11 import video | 10 import video |
12 | 11 |
| 12 from py_utils import discover |
13 | 13 |
14 # Import all submodules' StorySet classes. | 14 # Import all submodules' StorySet classes. |
15 start_dir = os.path.dirname(os.path.abspath(__file__)) | 15 start_dir = os.path.dirname(os.path.abspath(__file__)) |
16 top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir)) | 16 top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir)) |
17 base_class = story.StorySet | 17 base_class = story.StorySet |
18 for cls in discover.DiscoverClasses( | 18 for cls in discover.DiscoverClasses( |
19 start_dir, top_level_dir, base_class).values(): | 19 start_dir, top_level_dir, base_class).values(): |
20 setattr(sys.modules[__name__], cls.__name__, cls) | 20 setattr(sys.modules[__name__], cls.__name__, cls) |
21 | 21 |
22 # DiscoverClasses makes the assumption that there is exactly one matching | 22 # DiscoverClasses makes the assumption that there is exactly one matching |
23 # class per file, however the following are declared in the same file. | 23 # class per file, however the following are declared in the same file. |
24 for cls in (video.VideoDirectStorySet, | 24 for cls in (video.VideoDirectStorySet, |
25 video.VideoProxiedStorySet, | 25 video.VideoProxiedStorySet, |
26 video.VideoCompareStorySet): | 26 video.VideoCompareStorySet): |
27 setattr(sys.modules[__name__], cls.__name__, cls) | 27 setattr(sys.modules[__name__], cls.__name__, cls) |
OLD | NEW |