Index: tools/telemetry/telemetry/page_set.py |
diff --git a/tools/telemetry/telemetry/page_set.py b/tools/telemetry/telemetry/page_set.py |
deleted file mode 100644 |
index 45ab8a83e33a5d280e44009f5342dd4436ad47ef..0000000000000000000000000000000000000000 |
--- a/tools/telemetry/telemetry/page_set.py |
+++ /dev/null |
@@ -1,96 +0,0 @@ |
-# Copyright (c) 2012 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. |
-import csv |
-import json |
-import os |
-import urlparse |
- |
-from telemetry import page as page_module |
-from telemetry import page_set_archive_info |
- |
-class PageSet(object): |
- def __init__(self, file_path='', attributes=None): |
- self.description = '' |
- self.archive_data_file = '' |
- self.file_path = file_path |
- self.credentials_path = None |
- self.user_agent_type = None |
- |
- if attributes: |
- for k, v in attributes.iteritems(): |
- setattr(self, k, v) |
- |
- self.pages = [] |
- |
- if self.archive_data_file: |
- base_dir = os.path.dirname(file_path) |
- self.wpr_archive_info = page_set_archive_info.PageSetArchiveInfo.FromFile( |
- os.path.join(base_dir, self.archive_data_file), file_path) |
- else: |
- self.wpr_archive_info = None |
- |
- @classmethod |
- def FromFile(cls, file_path): |
- with open(file_path, 'r') as f: |
- contents = f.read() |
- data = json.loads(contents) |
- return cls.FromDict(data, file_path) |
- |
- @classmethod |
- def FromDict(cls, data, file_path=''): |
- page_set = cls(file_path, data) |
- for page_attributes in data['pages']: |
- url = page_attributes.pop('url') |
- page = page_module.Page(url, page_set, attributes=page_attributes, |
- base_dir=os.path.dirname(file_path)) |
- page_set.pages.append(page) |
- return page_set |
- |
- def ContainsOnlyFileURLs(self): |
- for page in self.pages: |
- parsed_url = urlparse.urlparse(page.url) |
- if parsed_url.scheme != 'file': |
- return False |
- return True |
- |
- def ReorderPageSet(self, results_file): |
- """Reorders this page set based on the results of a past run.""" |
- page_set_dict = {} |
- for page in self.pages: |
- page_set_dict[page.url] = page |
- |
- pages = [] |
- with open(results_file, 'rb') as csv_file: |
- csv_reader = csv.reader(csv_file) |
- csv_header = csv_reader.next() |
- |
- if 'url' not in csv_header: |
- raise Exception('Unusable results_file.') |
- |
- url_index = csv_header.index('url') |
- |
- for csv_row in csv_reader: |
- if csv_row[url_index] in page_set_dict: |
- pages.append(page_set_dict[csv_row[url_index]]) |
- else: |
- raise Exception('Unusable results_file.') |
- |
- return pages |
- |
- def WprFilePathForPage(self, page): |
- if not self.wpr_archive_info: |
- return None |
- return self.wpr_archive_info.WprFilePathForPage(page) |
- |
- def __iter__(self): |
- return self.pages.__iter__() |
- |
- def __len__(self): |
- return len(self.pages) |
- |
- def __getitem__(self, key): |
- return self.pages[key] |
- |
- def __setitem__(self, key, value): |
- self.pages[key] = value |