| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 # Copyright (c) 2011, the Dart project authors. Please see the AUTHORS file | 2 # Copyright (c) 2011, the Dart project authors. Please see the AUTHORS file |
| 3 # for details. All rights reserved. Use of this source code is governed by a | 3 # for details. All rights reserved. Use of this source code is governed by a |
| 4 # BSD-style license that can be found in the LICENSE file. | 4 # BSD-style license that can be found in the LICENSE file. |
| 5 | 5 |
| 6 # Template loader and preprocessor. | 6 # Template loader and preprocessor. |
| 7 # | 7 # |
| 8 # Preprocessor language: | 8 # Preprocessor language: |
| 9 # | 9 # |
| 10 # //$ Comment line removed by preprocessor | 10 # //$ Comment line removed by preprocessor |
| (...skipping 15 matching lines...) Expand all Loading... |
| 26 root - a string, the directory under which the templates are stored. | 26 root - a string, the directory under which the templates are stored. |
| 27 subpaths - a list of strings, subpaths of root in search order. | 27 subpaths - a list of strings, subpaths of root in search order. |
| 28 conditions - a dictionay from strings to booleans. Any conditional | 28 conditions - a dictionay from strings to booleans. Any conditional |
| 29 expression must be a key in the map. | 29 expression must be a key in the map. |
| 30 """ | 30 """ |
| 31 self._root = root | 31 self._root = root |
| 32 self._subpaths = subpaths | 32 self._subpaths = subpaths |
| 33 self._conditions = conditions | 33 self._conditions = conditions |
| 34 self._cache = {} | 34 self._cache = {} |
| 35 | 35 |
| 36 def TryLoad(self, name): | 36 def TryLoad(self, name, more_conditions={}): |
| 37 """Returns content of template file as a string, or None of not found.""" | 37 """Returns content of template file as a string, or None of not found.""" |
| 38 if name in self._cache: | 38 conditions = dict(self._conditions, **more_conditions) |
| 39 return self._cache[name] | 39 cache_key = (name, tuple(sorted(conditions.items()))) |
| 40 if cache_key in self._cache: |
| 41 return self._cache[cache_key] |
| 40 | 42 |
| 41 for subpath in self._subpaths: | 43 for subpath in self._subpaths: |
| 42 template_file = os.path.join(self._root, subpath, name) | 44 template_file = os.path.join(self._root, subpath, name) |
| 43 if os.path.exists(template_file): | 45 if os.path.exists(template_file): |
| 44 template = ''.join(open(template_file).readlines()) | 46 template = ''.join(open(template_file).readlines()) |
| 45 template = self._Preprocess(template, template_file) | 47 template = self._Preprocess(template, template_file, conditions) |
| 46 self._cache[name] = template | 48 self._cache[cache_key] = template |
| 47 return template | 49 return template |
| 48 | 50 |
| 49 return None | 51 return None |
| 50 | 52 |
| 51 def Load(self, name): | 53 def Load(self, name, more_conditions={}): |
| 52 """Returns contents of template file as a string, or raises an exception.""" | 54 """Returns contents of template file as a string, or raises an exception.""" |
| 53 template = self.TryLoad(name) | 55 template = self.TryLoad(name, more_conditions) |
| 54 if template is not None: # Can be empty string | 56 if template is not None: # Can be empty string |
| 55 return template | 57 return template |
| 56 raise Exception("Could not find template '%s' on %s / %s" % ( | 58 raise Exception("Could not find template '%s' on %s / %s" % ( |
| 57 name, self._root, self._subpaths)) | 59 name, self._root, self._subpaths)) |
| 58 | 60 |
| 59 def _Preprocess(self, template, filename): | 61 def _Preprocess(self, template, filename, conditions): |
| 60 def error(lineno, message): | 62 def error(lineno, message): |
| 61 raise Exception('%s:%s: %s' % (filename, lineno, message)) | 63 raise Exception('%s:%s: %s' % (filename, lineno, message)) |
| 62 | 64 |
| 63 lines = template.splitlines(True) | 65 lines = template.splitlines(True) |
| 64 out = [] | 66 out = [] |
| 65 | 67 |
| 66 condition_stack = [] | 68 condition_stack = [] |
| 67 active = True | 69 active = True |
| 68 seen_else = False | 70 seen_else = False |
| 69 | 71 |
| 70 for (lineno, full_line) in enumerate(lines): | 72 for (lineno, full_line) in enumerate(lines): |
| 71 line = full_line.strip() | 73 line = full_line.strip() |
| 72 | 74 |
| 73 if line.startswith('$'): | 75 if line.startswith('$'): |
| 74 words = line.split() | 76 words = line.split() |
| 75 directive = words[0] | 77 directive = words[0] |
| 76 | 78 |
| 77 if directive == '$if': | 79 if directive == '$if': |
| 78 if len(words) != 2: | 80 if len(words) != 2: |
| 79 error(lineno, '$if does not have single variable') | 81 error(lineno, '$if does not have single variable') |
| 80 variable = words[1] | 82 variable = words[1] |
| 81 if variable in self._conditions: | 83 if variable in conditions: |
| 82 condition_stack.append((active, seen_else)) | 84 condition_stack.append((active, seen_else)) |
| 83 active = self._conditions[variable] | 85 active = conditions[variable] |
| 84 seen_else = False | 86 seen_else = False |
| 85 else: | 87 else: |
| 86 error(lineno, "Unknown $if variable '%s'" % variable) | 88 error(lineno, "Unknown $if variable '%s'" % variable) |
| 87 | 89 |
| 88 elif directive == '$else': | 90 elif directive == '$else': |
| 89 if not condition_stack: | 91 if not condition_stack: |
| 90 error(lineno, '$else without $if') | 92 error(lineno, '$else without $if') |
| 91 if seen_else: | 93 if seen_else: |
| 92 raise error(lineno, 'Double $else') | 94 raise error(lineno, 'Double $else') |
| 93 seen_else = True | 95 seen_else = True |
| (...skipping 13 matching lines...) Expand all Loading... |
| 107 | 109 |
| 108 else: | 110 else: |
| 109 if active: | 111 if active: |
| 110 out.append(full_line) | 112 out.append(full_line) |
| 111 continue | 113 continue |
| 112 | 114 |
| 113 if condition_stack: | 115 if condition_stack: |
| 114 error(len(lines), 'Unterminated $if') | 116 error(len(lines), 'Unterminated $if') |
| 115 | 117 |
| 116 return ''.join(out) | 118 return ''.join(out) |
| OLD | NEW |