OLD | NEW |
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 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 import os | 4 import os |
5 import sys | 5 import sys |
6 | 6 |
| 7 from telemetry.core import discover |
7 from telemetry.story import story_set | 8 from telemetry.story import story_set |
8 from telemetry.util import classes_util | |
9 | 9 |
10 | 10 |
11 # Import all submodules' StorySet classes. | 11 # Import all submodules' StorySet classes. |
12 start_dir = os.path.dirname(os.path.abspath(__file__)) | 12 start_dir = os.path.dirname(os.path.abspath(__file__)) |
13 top_level_dir = os.path.dirname(start_dir) | 13 top_level_dir = os.path.dirname(start_dir) |
14 base_class = story_set.StorySet | 14 base_class = story_set.StorySet |
15 for cls in classes_util.DiscoverClasses(start_dir, top_level_dir, base_class): | 15 for cls in discover.DiscoverClasses( |
| 16 start_dir, top_level_dir, base_class).values(): |
16 setattr(sys.modules[__name__], cls.__name__, cls) | 17 setattr(sys.modules[__name__], cls.__name__, cls) |
OLD | NEW |