| Index: tools/telemetry/telemetry/user_story/user_story_set.py
|
| diff --git a/tools/telemetry/telemetry/user_story/user_story_set.py b/tools/telemetry/telemetry/user_story/user_story_set.py
|
| index aabd76e325869451acd2aee8ed3d6f2361a6c440..df500df45235b8649941c64eda6caae23725a1dd 100644
|
| --- a/tools/telemetry/telemetry/user_story/user_story_set.py
|
| +++ b/tools/telemetry/telemetry/user_story/user_story_set.py
|
| @@ -17,7 +17,7 @@ class UserStorySet(object):
|
| """
|
|
|
| def __init__(self, archive_data_file='', cloud_storage_bucket=None,
|
| - serving_dirs=None):
|
| + base_dir=None, serving_dirs=None):
|
| """Creates a new UserStorySet.
|
|
|
| Args:
|
| @@ -33,11 +33,21 @@ class UserStorySet(object):
|
| self._wpr_archive_info = None
|
| archive_info.AssertValidCloudStorageBucket(cloud_storage_bucket)
|
| self._cloud_storage_bucket = cloud_storage_bucket
|
| - self._base_dir = os.path.dirname(inspect.getfile(self.__class__))
|
| + if base_dir:
|
| + if not os.path.isdir(base_dir):
|
| + raise ValueError('Must provide valid directory path for base_dir.')
|
| + self._base_dir = base_dir
|
| + else:
|
| + self._base_dir = os.path.dirname(inspect.getfile(self.__class__))
|
| # Convert any relative serving_dirs to absolute paths.
|
| self._serving_dirs = set(os.path.realpath(os.path.join(self.base_dir, d))
|
| for d in serving_dirs or [])
|
|
|
| +
|
| + @property
|
| + def file_path(self):
|
| + return inspect.getfile(self.__class__).replace('.pyc', '.py')
|
| +
|
| @property
|
| def base_dir(self):
|
| """The base directory to resolve archive_data_file.
|
|
|