Chromium Code Reviews| 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 json | 5 import json |
| 6 import logging | 6 import logging |
| 7 import os | 7 import os |
| 8 | 8 |
| 9 from handlebar_dict_generator import HandlebarDictGenerator | 9 from handlebar_dict_generator import HandlebarDictGenerator |
| 10 import third_party.json_schema_compiler.json_comment_eater as json_comment_eater | 10 import third_party.json_schema_compiler.json_comment_eater as json_comment_eater |
| 11 import third_party.json_schema_compiler.model as model | 11 import third_party.json_schema_compiler.model as model |
| 12 import third_party.json_schema_compiler.idl_schema as idl_schema | 12 import third_party.json_schema_compiler.idl_schema as idl_schema |
| 13 import third_party.json_schema_compiler.idl_parser as idl_parser | 13 import third_party.json_schema_compiler.idl_parser as idl_parser |
| 14 | 14 |
| 15 class APIDataSource(object): | 15 class APIDataSource(object): |
| 16 """This class fetches and loads JSON APIs from the FileSystem passed in with | 16 """This class fetches and loads JSON APIs from the FileSystem passed in with |
| 17 |cache_builder|, so the APIs can be plugged into templates. | 17 |cache_builder|, so the APIs can be plugged into templates. |
| 18 """ | 18 """ |
| 19 def __init__(self, cache_builder, base_path): | 19 def __init__(self, cache_builder, base_path): |
|
cduvall
2012/07/26 22:04:24
See comment in TDS about factory so this can take
chebert
2012/07/26 22:30:52
Done.
| |
| 20 self._json_cache = cache_builder.build(self._LoadJsonAPI) | 20 self._json_cache = cache_builder.build(self._LoadJsonAPI) |
| 21 self._idl_cache = cache_builder.build(self._LoadIdlAPI) | 21 self._idl_cache = cache_builder.build(self._LoadIdlAPI) |
| 22 self._base_path = base_path | 22 self._base_path = base_path |
| 23 | 23 |
| 24 def _LoadJsonAPI(self, api): | 24 def _LoadJsonAPI(self, api): |
| 25 generator = HandlebarDictGenerator( | 25 generator = HandlebarDictGenerator( |
| 26 json.loads(json_comment_eater.Nom(api))[0]) | 26 json.loads(json_comment_eater.Nom(api))[0], |
| 27 self.samples) | |
| 27 return generator.Generate() | 28 return generator.Generate() |
| 28 | 29 |
| 29 def _LoadIdlAPI(self, api): | 30 def _LoadIdlAPI(self, api): |
| 30 idl = idl_parser.IDLParser().ParseData(api) | 31 idl = idl_parser.IDLParser().ParseData(api) |
| 31 generator = HandlebarDictGenerator(idl_schema.IDLSchema(idl).process()[0]) | 32 generator = HandlebarDictGenerator(idl_schema.IDLSchema(idl).process()[0], |
| 33 self.samples) | |
| 32 return generator.Generate() | 34 return generator.Generate() |
| 33 | 35 |
| 34 def __getitem__(self, key): | 36 def __getitem__(self, key): |
| 35 return self.get(key) | 37 return self.get(key) |
| 36 | 38 |
| 37 def get(self, key): | 39 def get(self, key): |
| 38 path, ext = os.path.splitext(key) | 40 path, ext = os.path.splitext(key) |
| 39 unix_name = model.UnixName(path) | 41 unix_name = model.UnixName(path) |
| 40 json_path = unix_name + '.json' | 42 json_path = unix_name + '.json' |
| 41 idl_path = unix_name + '.idl' | 43 idl_path = unix_name + '.idl' |
| 42 try: | 44 try: |
| 43 return self._json_cache.GetFromFile(self._base_path + '/' + json_path) | 45 return self._json_cache.GetFromFile(self._base_path + '/' + json_path) |
| 44 except Exception: | 46 except Exception: |
| 45 try: | 47 try: |
| 46 return self._idl_cache.GetFromFile(self._base_path + '/' + idl_path) | 48 return self._idl_cache.GetFromFile(self._base_path + '/' + idl_path) |
| 47 except Exception as e: | 49 except Exception as e: |
| 48 logging.warn(e) | 50 logging.warn(e) |
| 49 return None | 51 return None |
| OLD | NEW |