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 b4315ffc9e7799afe216fa37bcf115909d0daace..a160b52602988a20a64a31d57755e49ee848ea90 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. |