Chromium Code Reviews| Index: chrome/common/extensions/docs/server2/chrome_version_data_source.py |
| diff --git a/chrome/common/extensions/docs/server2/chrome_version_data_source.py b/chrome/common/extensions/docs/server2/chrome_version_data_source.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..86a804bf38811bc5750f9340df290038b4a1aac4 |
| --- /dev/null |
| +++ b/chrome/common/extensions/docs/server2/chrome_version_data_source.py |
| @@ -0,0 +1,53 @@ |
| +# 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 json |
| +import logging |
| +import object_store |
| + |
| +class ChromeVersionDataSource: |
| + def __init__(self, base_path, fetcher, object_store): |
| + self._base_path = base_path |
| + self._fetcher = fetcher |
| + self._object_store = object_store |
| + |
| + def _GetBranchNumberForVersion(self, version_number): |
| + """Returns the most recent branch number for a given chrome version number |
| + using data stored on omahaproxy (see url_constants) |
| + """ |
| + branch = self._object_store.Get( |
| + version_number, |
| + object_store.CHROME_VERSION_DATA_SOURCE).Get() |
| + |
| + if branch is not None: |
| + return branch |
| + |
| + try: |
| + version_json = json.loads(self._fetcher.Fetch(self._base_path).content) |
| + except Exception as e: |
| + #if omahaproxy is having problems |
| + logging.error("Could not fetch data at Omaha Proxy.\n%s" % (e)) |
|
cduvall
2013/03/21 18:43:53
better return None here
epeterson
2013/03/25 19:35:11
Done? Is there a better way to handle this case so
|
| + |
| + # entry['title'] looks like: 'title - version#.#.branch#.#' |
| + for entry in version_json['events']: |
| + version_title = entry['title'].split(' - ')[1].split('.') |
| + if version_title[0] == version_number: |
| + self._object_store.Set(version_number, |
| + version_title[2], |
| + object_store.CHROME_VERSION_DATA_SOURCE) |
| + return version_title[2] |
| + |
| + def GetDataSourceForVersion(self, version_number): |
| + """Returns an api data source for the most recent branch number |
| + corresponding to a given version number |
| + """ |
| + branch_number = self._GetBranchNumberForVersion(version_number) |
| + branch_api_data_source = self._CreateMemcacheForBranch(branch_number) |
| + return branch_api_data_source |
| + |
| + def SetCreateMemcacheForBranch(self, create_memcache_for_branch): |
| + """Sets a function, initially created in handler, that returns a memcache |
| + tied to a specific branch number |
| + """ |
| + self._CreateMemcacheForBranch = create_memcache_for_branch |