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 from fnmatch import fnmatch | 5 from fnmatch import fnmatch |
6 import mimetypes | 6 import mimetypes |
7 import os | 7 import os |
8 | 8 |
| 9 from file_system import FileNotFoundError |
| 10 |
9 STATIC_DIR_PREFIX = 'docs/server2' | 11 STATIC_DIR_PREFIX = 'docs/server2' |
10 DOCS_PATH = 'docs' | 12 DOCS_PATH = 'docs' |
11 | 13 |
12 class ServerInstance(object): | 14 class ServerInstance(object): |
13 """This class is used to hold a data source and fetcher for an instance of a | 15 """This class is used to hold a data source and fetcher for an instance of a |
14 server. Each new branch will get its own ServerInstance. | 16 server. Each new branch will get its own ServerInstance. |
15 """ | 17 """ |
16 def __init__(self, | 18 def __init__(self, |
17 template_data_source_factory, | 19 template_data_source_factory, |
18 example_zipper, | 20 example_zipper, |
19 cache_builder): | 21 cache_builder): |
20 self._template_data_source_factory = template_data_source_factory | 22 self._template_data_source_factory = template_data_source_factory |
21 self._example_zipper = example_zipper | 23 self._example_zipper = example_zipper |
22 self._cache = cache_builder.build(lambda x: x) | 24 self._cache = cache_builder.build(lambda x: x) |
23 mimetypes.init() | 25 mimetypes.init() |
24 | 26 |
25 def _FetchStaticResource(self, path, response): | 27 def _FetchStaticResource(self, path, response): |
26 """Fetch a resource in the 'static' directory. | 28 """Fetch a resource in the 'static' directory. |
27 """ | 29 """ |
28 try: | 30 try: |
29 result = self._cache.GetFromFile(STATIC_DIR_PREFIX + '/' + path) | 31 result = self._cache.GetFromFile(STATIC_DIR_PREFIX + '/' + path) |
30 base, ext = os.path.splitext(path) | 32 base, ext = os.path.splitext(path) |
31 response.headers['content-type'] = mimetypes.types_map[ext] | 33 response.headers['content-type'] = mimetypes.types_map[ext] |
32 return result | 34 return result |
33 except Exception: | 35 except FileNotFoundError: |
34 return '' | 36 return None |
35 | 37 |
36 def Get(self, path, request, response): | 38 def Get(self, path, request, response): |
37 templates = self._template_data_source_factory.Create(request) | 39 templates = self._template_data_source_factory.Create(request) |
38 | 40 |
39 if fnmatch(path, 'extensions/examples/*.zip'): | 41 if fnmatch(path, 'extensions/examples/*.zip'): |
40 content = self._example_zipper.Create( | 42 content = self._example_zipper.Create( |
41 path[len('extensions/'):-len('.zip')]) | 43 path[len('extensions/'):-len('.zip')]) |
42 response.headers['content-type'] = mimetypes.types_map['.zip'] | 44 response.headers['content-type'] = mimetypes.types_map['.zip'] |
43 elif path.startswith('extensions/examples/'): | 45 elif path.startswith('extensions/examples/'): |
44 content = self._cache.GetFromFile( | 46 content = self._cache.GetFromFile( |
45 DOCS_PATH + '/' + path[len('extensions/'):]) | 47 DOCS_PATH + '/' + path[len('extensions/'):]) |
46 response.headers['content-type'] = 'text/plain' | 48 response.headers['content-type'] = 'text/plain' |
47 elif path.startswith('static/'): | 49 elif path.startswith('static/'): |
48 content = self._FetchStaticResource(path, response) | 50 content = self._FetchStaticResource(path, response) |
49 else: | 51 else: |
50 content = templates.Render(path) | 52 content = templates.Render(path) |
51 | 53 |
52 if content: | 54 if content: |
53 response.out.write(content) | 55 response.out.write(content) |
54 else: | 56 else: |
55 response.set_status(404); | 57 response.set_status(404); |
56 response.out.write(templates.Render('404')) | 58 response.out.write(templates.Render('404')) |
OLD | NEW |