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 a6771fd6adf762e163f7824b369c7310d9b19601..ed0bee425268bc7f64eb146cebac21f8a144d21d 100644 |
--- a/tools/telemetry/telemetry/user_story/user_story_set.py |
+++ b/tools/telemetry/telemetry/user_story/user_story_set.py |
@@ -16,7 +16,8 @@ class UserStorySet(object): |
AddUserStory for each UserStory.. |
""" |
- def __init__(self, archive_data_file='', cloud_storage_bucket=None): |
+ def __init__(self, archive_data_file='', cloud_storage_bucket=None, |
+ serving_dirs=None): |
"""Creates a new UserStorySet. |
Args: |
@@ -33,6 +34,9 @@ class UserStorySet(object): |
page_set_archive_info.AssertValidCloudStorageBucket(cloud_storage_bucket) |
self._cloud_storage_bucket = cloud_storage_bucket |
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 base_dir(self): |
@@ -43,6 +47,10 @@ class UserStorySet(object): |
return self._base_dir |
@property |
+ def serving_dirs(self): |
+ return self._serving_dirs |
+ |
+ @property |
def archive_data_file(self): |
return self._archive_data_file |
@@ -88,6 +96,15 @@ class UserStorySet(object): |
pass |
def WprFilePathForUserStory(self, story): |
+ """Convenient function to retrive WPR archive file path. |
+ |
+ Args: |
+ user_story: The UserStory to lookup. |
+ |
+ Returns: |
+ The WPR archive file path for the given UserStory, if found. |
+ Otherwise, return None. |
+ """ |
if not self.wpr_archive_info: |
return None |
return self.wpr_archive_info.WprFilePathForUserStory(story) |