| 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 |
| 10 from telemetry.util import classes_util |
| 11 | 11 |
| 12 | 12 |
| 13 # Import all submodules' StorySet classes. | 13 # Import all submodules' StorySet classes. |
| 14 start_dir = os.path.dirname(os.path.abspath(__file__)) | 14 start_dir = os.path.dirname(os.path.abspath(__file__)) |
| 15 top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir)) | 15 top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir)) |
| 16 base_class = story.StorySet | 16 base_class = story.StorySet |
| 17 for cls in discover.DiscoverClasses( | 17 for cls in classes_util.DiscoverClasses(start_dir, top_level_dir, base_class): |
| 18 start_dir, top_level_dir, base_class).values(): | |
| 19 setattr(sys.modules[__name__], cls.__name__, cls) | 18 setattr(sys.modules[__name__], cls.__name__, cls) |
| 20 | 19 |
| OLD | NEW |