| # Copyright 2015 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| from profile_creators import profile_safe_url_list |
| from telemetry.page import page as page_module |
| from telemetry.page import shared_page_state |
| from telemetry import story |
| |
| |
| class ProfileSafeUrlPage(page_module.Page): |
| def __init__(self, url, page_set): |
| super(ProfileSafeUrlPage, self).__init__( |
| url=url, |
| page_set = page_set, |
| shared_page_state_class=shared_page_state.SharedDesktopPageState, |
| credentials_path = 'data/credentials.json') |
| self.credentials = 'google' |
| |
| |
| class ProfileSafeUrlsPageSet(story.StorySet): |
| """Safe urls used for profile generation.""" |
| |
| def __init__(self): |
| super(ProfileSafeUrlsPageSet, self).__init__( |
| archive_data_file='data/profile_safe_urls.json', |
| cloud_storage_bucket=story.PARTNER_BUCKET) |
| |
| # Only use the first 500 urls to prevent the .wpr files from getting too |
| # big. |
| safe_urls = profile_safe_url_list.GetShuffledSafeUrls()[0:500] |
| for safe_url in safe_urls: |
| self.AddStory(ProfileSafeUrlPage(safe_url, self)) |