| OLD | NEW |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import os | 5 import os |
| 6 | 6 |
| 7 import third_party.json_schema_compiler.model as model | 7 import third_party.json_schema_compiler.model as model |
| 8 from docs_server_utils import SanitizeAPIName | 8 from docs_server_utils import SanitizeAPIName |
| 9 | 9 |
| 10 class APIListDataSource(object): | 10 class APIListDataSource(object): |
| 11 """ This class creates a list of chrome.* APIs and chrome.experimental.* APIs | 11 """ This class creates a list of chrome.* APIs and chrome.experimental.* APIs |
| 12 that are used in the api_index.html and experimental.html pages. | 12 that are used in the api_index.html and experimental.html pages. |
| 13 """ | 13 """ |
| 14 def __init__(self, cache_builder, file_system, api_path, public_path): | 14 def __init__(self, cache_builder, file_system, api_path, public_path): |
| 15 self._cache = cache_builder.build(self._ListAPIs) | 15 self._cache = cache_builder.build(self._ListAPIs) |
| 16 self._file_system = file_system | 16 self._file_system = file_system |
| 17 self._api_path = api_path + '/' | 17 self._api_path = api_path + '/' |
| 18 self._public_path = public_path + '/' | 18 self._public_path = public_path + '/' |
| 19 | 19 |
| 20 def _ListAPIs(self, apis): | 20 def _ListAPIs(self, apis): |
| 21 api_names = set(SanitizeAPIName(name, self._api_path) for name in apis) | 21 api_names = set(SanitizeAPIName(name, api_path=self._api_path) |
| 22 for name in apis) |
| 22 public_templates = self._file_system.ReadSingle(self._public_path) | 23 public_templates = self._file_system.ReadSingle(self._public_path) |
| 23 template_names = [os.path.splitext(name)[0] for name in public_templates] | 24 template_names = [os.path.splitext(name)[0] for name in public_templates] |
| 24 experimental_apis = [] | 25 experimental_apis = [] |
| 25 chrome_apis = [] | 26 chrome_apis = [] |
| 26 for i, template_name in enumerate(sorted(template_names)): | 27 for i, template_name in enumerate(sorted(template_names)): |
| 27 if model.UnixName(template_name) in api_names: | 28 if model.UnixName(template_name) in api_names: |
| 28 if template_name.startswith('experimental'): | 29 if template_name.startswith('experimental'): |
| 29 experimental_apis.append({ 'name': template_name.replace('_', '.') }) | 30 experimental_apis.append({ 'name': template_name.replace('_', '.') }) |
| 30 else: | 31 else: |
| 31 chrome_apis.append({ 'name': template_name.replace('_', '.') }) | 32 chrome_apis.append({ 'name': template_name.replace('_', '.') }) |
| 32 chrome_apis[-1]['last'] = True | 33 chrome_apis[-1]['last'] = True |
| 33 experimental_apis[-1]['last'] = True | 34 experimental_apis[-1]['last'] = True |
| 34 return { | 35 return { |
| 35 'chrome': chrome_apis, | 36 'chrome': chrome_apis, |
| 36 'experimental': experimental_apis | 37 'experimental': experimental_apis |
| 37 } | 38 } |
| 38 | 39 |
| 39 def __getitem__(self, key): | 40 def __getitem__(self, key): |
| 40 return self.get(key) | 41 return self.get(key) |
| 41 | 42 |
| 42 def get(self, key): | 43 def get(self, key): |
| 43 try: | 44 try: |
| 44 return self._cache.GetFromFileListing(self._api_path)[key] | 45 return self._cache.GetFromFileListing(self._api_path)[key] |
| 45 except Exception as e: | 46 except Exception as e: |
| 46 return None | 47 return None |
| OLD | NEW |