OLD | NEW |
| (Empty) |
1 #!/usr/bin/python | |
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 | |
4 # BSD-style license that can be found in the LICENSE file. | |
5 | |
6 # Template loader and preprocessor. | |
7 # | |
8 # Preprocessor language: | |
9 # | |
10 # $if VAR | |
11 # $else | |
12 # $endif | |
13 # | |
14 # VAR must be defined in the conditions dictionary. | |
15 | |
16 import os | |
17 | |
18 class TemplateLoader(object): | |
19 """Loads template files from a path.""" | |
20 | |
21 def __init__(self, root, subpaths, conditions = {}): | |
22 """Initializes loader. | |
23 | |
24 Args: | |
25 root - a string, the directory under which the templates are stored. | |
26 subpaths - a list of strings, subpaths of root in search order. | |
27 conditions - a dictionay from strings to booleans. Any conditional | |
28 expression must be a key in the map. | |
29 """ | |
30 self._root = root | |
31 self._subpaths = subpaths | |
32 self._conditions = conditions | |
33 self._cache = {} | |
34 | |
35 def TryLoad(self, name): | |
36 """Returns content of template file as a string, or None of not found.""" | |
37 if name in self._cache: | |
38 return self._cache[name] | |
39 | |
40 for subpath in self._subpaths: | |
41 template_file = os.path.join(self._root, subpath, name) | |
42 if os.path.exists(template_file): | |
43 template = ''.join(open(template_file).readlines()) | |
44 template = self._Preprocess(template, template_file) | |
45 self._cache[name] = template | |
46 return template | |
47 | |
48 return None | |
49 | |
50 def Load(self, name): | |
51 """Returns contents of template file as a string, or raises an exception.""" | |
52 template = self.TryLoad(name) | |
53 if template is not None: # Can be empty string | |
54 return template | |
55 raise Exception("Could not find template '%s' on %s / %s" % ( | |
56 name, self._root, self._subpaths)) | |
57 | |
58 def _Preprocess(self, template, filename): | |
59 def error(lineno, message): | |
60 raise Exception('%s:%s: %s' % (filename, lineno, message)) | |
61 | |
62 lines = template.splitlines(True) | |
63 out = [] | |
64 | |
65 condition_stack = [] | |
66 active = True | |
67 seen_else = False | |
68 | |
69 for (lineno, full_line) in enumerate(lines): | |
70 line = full_line.strip() | |
71 | |
72 if line.startswith('$'): | |
73 words = line.split() | |
74 directive = words[0] | |
75 | |
76 if directive == '$if': | |
77 if len(words) != 2: | |
78 error(lineno, '$if does not have single variable') | |
79 variable = words[1] | |
80 if variable in self._conditions: | |
81 condition_stack.append((active, seen_else)) | |
82 active = self._conditions[variable] | |
83 seen_else = False | |
84 else: | |
85 error(lineno, "Unknown $if variable '%s'" % variable) | |
86 | |
87 elif directive == '$else': | |
88 if not condition_stack: | |
89 error(lineno, '$else without $if') | |
90 if seen_else: | |
91 raise error(lineno, 'Double $else') | |
92 seen_else = True | |
93 active = not active | |
94 | |
95 elif directive == '$endif': | |
96 if not condition_stack: | |
97 error(lineno, '$endif without $if') | |
98 (active, seen_else) = condition_stack.pop() | |
99 | |
100 else: | |
101 # Something else, like '$!MEMBERS' | |
102 if active: | |
103 out.append(full_line) | |
104 | |
105 else: | |
106 if active: | |
107 out.append(full_line); | |
108 continue | |
109 | |
110 if condition_stack: | |
111 error(len(lines), 'Unterminated $if') | |
112 | |
113 return ''.join(out) | |
OLD | NEW |