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 |