blob: 25cec88b1c5063b7d1b201c960b5f800ec10f223 [file] [log] [blame]
Armin Ronacher07bc6842008-03-31 14:18:49 +02001# -*- coding: utf-8 -*-
2"""
Armin Ronacher82b3f3d2008-03-31 20:01:08 +02003 jinja2.environment
4 ~~~~~~~~~~~~~~~~~~
Armin Ronacher07bc6842008-03-31 14:18:49 +02005
6 Provides a class that holds runtime and parsing time options.
7
Armin Ronacher55494e42010-01-22 09:41:48 +01008 :copyright: (c) 2010 by the Jinja Team.
Armin Ronacher07bc6842008-03-31 14:18:49 +02009 :license: BSD, see LICENSE for more details.
10"""
Armin Ronacher12a316b2010-03-12 17:59:51 +010011import os
Armin Ronacherba3757b2008-04-16 19:43:16 +020012import sys
Armin Ronacherba6e25a2008-11-02 15:58:14 +010013from jinja2 import nodes
Armin Ronacher7259c762008-04-30 13:03:59 +020014from jinja2.defaults import *
Armin Ronacher9a0078d2008-08-13 18:24:17 +020015from jinja2.lexer import get_lexer, TokenStream
Armin Ronacher05530932008-04-20 13:27:49 +020016from jinja2.parser import Parser
Armin Ronacherbcb7c532008-04-11 16:30:34 +020017from jinja2.optimizer import optimize
18from jinja2.compiler import generate
Armin Ronacher74a0cd92009-02-19 15:56:53 +010019from jinja2.runtime import Undefined, new_context
Armin Ronacher31bbd9e2010-01-14 00:41:30 +010020from jinja2.exceptions import TemplateSyntaxError, TemplateNotFound, \
21 TemplatesNotFound
Armin Ronacherba6e25a2008-11-02 15:58:14 +010022from jinja2.utils import import_string, LRUCache, Markup, missing, \
Armin Ronacher0d242be2010-02-10 01:35:13 +010023 concat, consume, internalcode, _encode_filename
Armin Ronacher07bc6842008-03-31 14:18:49 +020024
25
Armin Ronacher203bfcb2008-04-24 21:54:44 +020026# for direct template usage we have up to ten living environments
27_spontaneous_environments = LRUCache(10)
28
Armin Ronachera18872d2009-03-05 23:47:00 +010029# the function to create jinja traceback objects. This is dynamically
30# imported on the first exception in the exception handler.
31_make_traceback = None
32
Armin Ronacher203bfcb2008-04-24 21:54:44 +020033
Armin Ronacherb5124e62008-04-25 00:36:14 +020034def get_spontaneous_environment(*args):
Georg Brandl3e497b72008-09-19 09:55:17 +000035 """Return a new spontaneous environment. A spontaneous environment is an
36 unnamed and unaccessible (in theory) environment that is used for
37 templates generated from a string and not from the file system.
Armin Ronacher203bfcb2008-04-24 21:54:44 +020038 """
39 try:
40 env = _spontaneous_environments.get(args)
41 except TypeError:
42 return Environment(*args)
43 if env is not None:
44 return env
45 _spontaneous_environments[args] = env = Environment(*args)
Armin Ronacherc9705c22008-04-27 21:28:03 +020046 env.shared = True
Armin Ronacher203bfcb2008-04-24 21:54:44 +020047 return env
48
49
Armin Ronacher7259c762008-04-30 13:03:59 +020050def create_cache(size):
51 """Return the cache class for the given size."""
52 if size == 0:
53 return None
54 if size < 0:
55 return {}
56 return LRUCache(size)
57
58
Armin Ronacherccae0552008-10-05 23:08:58 +020059def copy_cache(cache):
60 """Create an empty copy of the given cache."""
61 if cache is None:
Armin Ronacher2bc1ef72008-12-08 15:21:26 +010062 return None
Armin Ronacherccae0552008-10-05 23:08:58 +020063 elif type(cache) is dict:
64 return {}
65 return LRUCache(cache.capacity)
66
67
Armin Ronacher7259c762008-04-30 13:03:59 +020068def load_extensions(environment, extensions):
69 """Load the extensions from the list and bind it to the environment.
Armin Ronacher023b5e92008-05-08 11:03:10 +020070 Returns a dict of instanciated environments.
Armin Ronacher203bfcb2008-04-24 21:54:44 +020071 """
Armin Ronacher023b5e92008-05-08 11:03:10 +020072 result = {}
Armin Ronacher7259c762008-04-30 13:03:59 +020073 for extension in extensions:
74 if isinstance(extension, basestring):
75 extension = import_string(extension)
Armin Ronacher023b5e92008-05-08 11:03:10 +020076 result[extension.identifier] = extension(environment)
Armin Ronacher7259c762008-04-30 13:03:59 +020077 return result
Armin Ronacher203bfcb2008-04-24 21:54:44 +020078
Armin Ronacher203bfcb2008-04-24 21:54:44 +020079
Armin Ronacher7259c762008-04-30 13:03:59 +020080def _environment_sanity_check(environment):
81 """Perform a sanity check on the environment."""
82 assert issubclass(environment.undefined, Undefined), 'undefined must ' \
83 'be a subclass of undefined because filters depend on it.'
84 assert environment.block_start_string != \
85 environment.variable_start_string != \
86 environment.comment_start_string, 'block, variable and comment ' \
87 'start strings must be different'
Armin Ronacherf3c35c42008-05-23 23:18:14 +020088 assert environment.newline_sequence in ('\r', '\r\n', '\n'), \
89 'newline_sequence set to unknown line ending string.'
Armin Ronacher19cf9c22008-05-01 12:49:53 +020090 return environment
Armin Ronacher203bfcb2008-04-24 21:54:44 +020091
92
Armin Ronacher07bc6842008-03-31 14:18:49 +020093class Environment(object):
Armin Ronacherf3c35c42008-05-23 23:18:14 +020094 r"""The core component of Jinja is the `Environment`. It contains
Armin Ronacher07bc6842008-03-31 14:18:49 +020095 important shared variables like configuration, filters, tests,
Armin Ronacherd1342312008-04-28 12:20:12 +020096 globals and others. Instances of this class may be modified if
97 they are not shared and if no template was loaded so far.
98 Modifications on environments after the first template was loaded
99 will lead to surprising effects and undefined behavior.
100
101 Here the possible initialization parameters:
102
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200103 `block_start_string`
104 The string marking the begin of a block. Defaults to ``'{%'``.
Armin Ronacherd1342312008-04-28 12:20:12 +0200105
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200106 `block_end_string`
107 The string marking the end of a block. Defaults to ``'%}'``.
Armin Ronacherd1342312008-04-28 12:20:12 +0200108
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200109 `variable_start_string`
110 The string marking the begin of a print statement.
111 Defaults to ``'{{'``.
Armin Ronacher115de2e2008-05-01 22:20:05 +0200112
Armin Ronacher63fd7982008-06-20 18:47:56 +0200113 `variable_end_string`
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200114 The string marking the end of a print statement. Defaults to
115 ``'}}'``.
Armin Ronacherd1342312008-04-28 12:20:12 +0200116
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200117 `comment_start_string`
118 The string marking the begin of a comment. Defaults to ``'{#'``.
Armin Ronacherd1342312008-04-28 12:20:12 +0200119
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200120 `comment_end_string`
121 The string marking the end of a comment. Defaults to ``'#}'``.
Armin Ronacherd1342312008-04-28 12:20:12 +0200122
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200123 `line_statement_prefix`
124 If given and a string, this will be used as prefix for line based
125 statements. See also :ref:`line-statements`.
Armin Ronacherd1342312008-04-28 12:20:12 +0200126
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200127 `line_comment_prefix`
128 If given and a string, this will be used as prefix for line based
129 based comments. See also :ref:`line-statements`.
130
131 .. versionadded:: 2.2
132
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200133 `trim_blocks`
134 If this is set to ``True`` the first newline after a block is
135 removed (block, not variable tag!). Defaults to `False`.
Armin Ronacherd1342312008-04-28 12:20:12 +0200136
Armin Ronacherf3c35c42008-05-23 23:18:14 +0200137 `newline_sequence`
138 The sequence that starts a newline. Must be one of ``'\r'``,
139 ``'\n'`` or ``'\r\n'``. The default is ``'\n'`` which is a
140 useful default for Linux and OS X systems as well as web
141 applications.
142
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200143 `extensions`
144 List of Jinja extensions to use. This can either be import paths
Armin Ronachered98cac2008-05-07 08:42:11 +0200145 as strings or extension classes. For more information have a
146 look at :ref:`the extensions documentation <jinja-extensions>`.
Armin Ronacherd1342312008-04-28 12:20:12 +0200147
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200148 `optimized`
149 should the optimizer be enabled? Default is `True`.
Armin Ronacherd1342312008-04-28 12:20:12 +0200150
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200151 `undefined`
152 :class:`Undefined` or a subclass of it that is used to represent
153 undefined values in the template.
Armin Ronacherd1342312008-04-28 12:20:12 +0200154
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200155 `finalize`
Armin Ronacherd9ea26e2010-01-24 14:29:26 +0100156 A callable that can be used to process the result of a variable
157 expression before it is output. For example one can convert
158 `None` implicitly into an empty string here.
Armin Ronacherd1342312008-04-28 12:20:12 +0200159
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200160 `autoescape`
Armin Ronacher8346bd72010-03-14 19:43:47 +0100161 If set to true the XML/HTML autoescaping feature is enabled by
162 default. For more details about auto escaping see
Armin Ronacher1da23d12010-04-05 18:11:18 +0200163 :class:`~jinja2.utils.Markup`. As of Jinja 2.4 this can also
164 be a callable that is passed the template name and has to
165 return `True` or `False` depending on autoescape should be
166 enabled by default.
167
168 .. versionchanged:: 2.4
169 `autoescape` can now be a function
Armin Ronacherd1342312008-04-28 12:20:12 +0200170
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200171 `loader`
172 The template loader for this environment.
Armin Ronacher7259c762008-04-30 13:03:59 +0200173
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200174 `cache_size`
175 The size of the cache. Per default this is ``50`` which means
176 that if more than 50 templates are loaded the loader will clean
177 out the least recently used template. If the cache size is set to
178 ``0`` templates are recompiled all the time, if the cache size is
179 ``-1`` the cache will not be cleaned.
Armin Ronacher7259c762008-04-30 13:03:59 +0200180
Armin Ronacher7b5680c2008-05-06 16:54:22 +0200181 `auto_reload`
182 Some loaders load templates from locations where the template
183 sources may change (ie: file system or database). If
184 `auto_reload` is set to `True` (default) every time a template is
185 requested the loader checks if the source changed and if yes, it
186 will reload the template. For higher performance it's possible to
187 disable that.
Armin Ronacher4d5bdff2008-09-17 16:19:46 +0200188
189 `bytecode_cache`
190 If set to a bytecode cache object, this object will provide a
191 cache for the internal Jinja bytecode so that templates don't
192 have to be parsed if they were not changed.
Armin Ronachera816bf42008-09-17 21:28:01 +0200193
194 See :ref:`bytecode-cache` for more information.
Armin Ronacher07bc6842008-03-31 14:18:49 +0200195 """
196
Armin Ronacherc63243e2008-04-14 22:53:58 +0200197 #: if this environment is sandboxed. Modifying this variable won't make
198 #: the environment sandboxed though. For a real sandboxed environment
199 #: have a look at jinja2.sandbox
200 sandboxed = False
201
Armin Ronacher7259c762008-04-30 13:03:59 +0200202 #: True if the environment is just an overlay
Armin Ronacher619eeed2009-07-09 21:55:29 +0200203 overlayed = False
Armin Ronacher7259c762008-04-30 13:03:59 +0200204
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200205 #: the environment this environment is linked to if it is an overlay
206 linked_to = None
207
Armin Ronacherc9705c22008-04-27 21:28:03 +0200208 #: shared environments have this set to `True`. A shared environment
209 #: must not be modified
210 shared = False
211
Armin Ronacher32ed6c92009-04-02 14:04:41 +0200212 #: these are currently EXPERIMENTAL undocumented features.
Armin Ronachera18872d2009-03-05 23:47:00 +0100213 exception_handler = None
214 exception_formatter = None
215
Armin Ronacher07bc6842008-03-31 14:18:49 +0200216 def __init__(self,
Armin Ronacher7259c762008-04-30 13:03:59 +0200217 block_start_string=BLOCK_START_STRING,
218 block_end_string=BLOCK_END_STRING,
219 variable_start_string=VARIABLE_START_STRING,
220 variable_end_string=VARIABLE_END_STRING,
221 comment_start_string=COMMENT_START_STRING,
222 comment_end_string=COMMENT_END_STRING,
223 line_statement_prefix=LINE_STATEMENT_PREFIX,
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200224 line_comment_prefix=LINE_COMMENT_PREFIX,
Armin Ronacher4f5008f2008-05-23 23:36:07 +0200225 trim_blocks=TRIM_BLOCKS,
226 newline_sequence=NEWLINE_SEQUENCE,
Armin Ronacherb5124e62008-04-25 00:36:14 +0200227 extensions=(),
Armin Ronacherfed44b52008-04-13 19:42:53 +0200228 optimized=True,
Armin Ronacherc63243e2008-04-14 22:53:58 +0200229 undefined=Undefined,
Armin Ronacherd1342312008-04-28 12:20:12 +0200230 finalize=None,
231 autoescape=False,
Armin Ronacher7259c762008-04-30 13:03:59 +0200232 loader=None,
233 cache_size=50,
Armin Ronacher4d5bdff2008-09-17 16:19:46 +0200234 auto_reload=True,
235 bytecode_cache=None):
Armin Ronacherb5124e62008-04-25 00:36:14 +0200236 # !!Important notice!!
237 # The constructor accepts quite a few arguments that should be
238 # passed by keyword rather than position. However it's important to
239 # not change the order of arguments because it's used at least
240 # internally in those cases:
241 # - spontaneus environments (i18n extension and Template)
242 # - unittests
243 # If parameter changes are required only add parameters at the end
244 # and don't change the arguments (or the defaults!) of the arguments
Armin Ronacher7259c762008-04-30 13:03:59 +0200245 # existing already.
Armin Ronacher07bc6842008-03-31 14:18:49 +0200246
247 # lexer / parser information
248 self.block_start_string = block_start_string
249 self.block_end_string = block_end_string
250 self.variable_start_string = variable_start_string
251 self.variable_end_string = variable_end_string
252 self.comment_start_string = comment_start_string
253 self.comment_end_string = comment_end_string
Armin Ronacherbf7c4ad2008-04-12 12:02:36 +0200254 self.line_statement_prefix = line_statement_prefix
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200255 self.line_comment_prefix = line_comment_prefix
Armin Ronacher07bc6842008-03-31 14:18:49 +0200256 self.trim_blocks = trim_blocks
Armin Ronacherf3c35c42008-05-23 23:18:14 +0200257 self.newline_sequence = newline_sequence
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200258
Armin Ronacherf59bac22008-04-20 13:11:43 +0200259 # runtime information
Armin Ronacherc63243e2008-04-14 22:53:58 +0200260 self.undefined = undefined
Armin Ronacherfed44b52008-04-13 19:42:53 +0200261 self.optimized = optimized
Armin Ronacher18c6ca02008-04-17 10:03:29 +0200262 self.finalize = finalize
Armin Ronacherd1342312008-04-28 12:20:12 +0200263 self.autoescape = autoescape
Armin Ronacher07bc6842008-03-31 14:18:49 +0200264
265 # defaults
266 self.filters = DEFAULT_FILTERS.copy()
267 self.tests = DEFAULT_TESTS.copy()
268 self.globals = DEFAULT_NAMESPACE.copy()
269
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200270 # set the loader provided
271 self.loader = loader
Armin Ronacher4d5bdff2008-09-17 16:19:46 +0200272 self.bytecode_cache = None
Armin Ronacher7259c762008-04-30 13:03:59 +0200273 self.cache = create_cache(cache_size)
Armin Ronacher4d5bdff2008-09-17 16:19:46 +0200274 self.bytecode_cache = bytecode_cache
Armin Ronacher7259c762008-04-30 13:03:59 +0200275 self.auto_reload = auto_reload
Armin Ronacher07bc6842008-03-31 14:18:49 +0200276
Armin Ronacherb5124e62008-04-25 00:36:14 +0200277 # load extensions
Armin Ronacher7259c762008-04-30 13:03:59 +0200278 self.extensions = load_extensions(self, extensions)
279
280 _environment_sanity_check(self)
281
Armin Ronacher762079c2008-05-08 23:57:56 +0200282 def extend(self, **attributes):
283 """Add the items to the instance of the environment if they do not exist
284 yet. This is used by :ref:`extensions <writing-extensions>` to register
285 callbacks and configuration values without breaking inheritance.
286 """
287 for key, value in attributes.iteritems():
288 if not hasattr(self, key):
289 setattr(self, key, value)
290
Armin Ronacher7259c762008-04-30 13:03:59 +0200291 def overlay(self, block_start_string=missing, block_end_string=missing,
292 variable_start_string=missing, variable_end_string=missing,
293 comment_start_string=missing, comment_end_string=missing,
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200294 line_statement_prefix=missing, line_comment_prefix=missing,
295 trim_blocks=missing, extensions=missing, optimized=missing,
296 undefined=missing, finalize=missing, autoescape=missing,
297 loader=missing, cache_size=missing, auto_reload=missing,
Armin Ronacher4d5bdff2008-09-17 16:19:46 +0200298 bytecode_cache=missing):
Armin Ronacher7259c762008-04-30 13:03:59 +0200299 """Create a new overlay environment that shares all the data with the
Georg Brandl95632c42009-11-22 18:35:18 +0100300 current environment except of cache and the overridden attributes.
301 Extensions cannot be removed for an overlayed environment. An overlayed
Armin Ronacher7259c762008-04-30 13:03:59 +0200302 environment automatically gets all the extensions of the environment it
303 is linked to plus optional extra extensions.
304
305 Creating overlays should happen after the initial environment was set
306 up completely. Not all attributes are truly linked, some are just
307 copied over so modifications on the original environment may not shine
308 through.
309 """
310 args = dict(locals())
311 del args['self'], args['cache_size'], args['extensions']
312
313 rv = object.__new__(self.__class__)
314 rv.__dict__.update(self.__dict__)
Armin Ronacher619eeed2009-07-09 21:55:29 +0200315 rv.overlayed = True
Armin Ronacher7259c762008-04-30 13:03:59 +0200316 rv.linked_to = self
317
318 for key, value in args.iteritems():
319 if value is not missing:
320 setattr(rv, key, value)
321
322 if cache_size is not missing:
323 rv.cache = create_cache(cache_size)
Armin Ronacherccae0552008-10-05 23:08:58 +0200324 else:
325 rv.cache = copy_cache(self.cache)
Armin Ronacher7259c762008-04-30 13:03:59 +0200326
Armin Ronacher023b5e92008-05-08 11:03:10 +0200327 rv.extensions = {}
328 for key, value in self.extensions.iteritems():
329 rv.extensions[key] = value.bind(rv)
Armin Ronacher7259c762008-04-30 13:03:59 +0200330 if extensions is not missing:
Armin Ronacher023b5e92008-05-08 11:03:10 +0200331 rv.extensions.update(load_extensions(extensions))
Armin Ronacher7259c762008-04-30 13:03:59 +0200332
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200333 return _environment_sanity_check(rv)
Armin Ronacher7259c762008-04-30 13:03:59 +0200334
Armin Ronacher9a0078d2008-08-13 18:24:17 +0200335 lexer = property(get_lexer, doc="The lexer for this environment.")
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200336
Armin Ronacher6dc6f292008-06-12 08:50:07 +0200337 def getitem(self, obj, argument):
338 """Get an item or attribute of an object but prefer the item."""
Armin Ronacher08a6a3b2008-05-13 15:35:47 +0200339 try:
340 return obj[argument]
341 except (TypeError, LookupError):
Armin Ronacherf15f5f72008-05-26 12:21:45 +0200342 if isinstance(argument, basestring):
343 try:
344 attr = str(argument)
345 except:
346 pass
347 else:
348 try:
349 return getattr(obj, attr)
350 except AttributeError:
351 pass
Armin Ronacher08a6a3b2008-05-13 15:35:47 +0200352 return self.undefined(obj=obj, name=argument)
Armin Ronacherc63243e2008-04-14 22:53:58 +0200353
Armin Ronacher6dc6f292008-06-12 08:50:07 +0200354 def getattr(self, obj, attribute):
355 """Get an item or attribute of an object but prefer the attribute.
356 Unlike :meth:`getitem` the attribute *must* be a bytestring.
357 """
358 try:
359 return getattr(obj, attribute)
360 except AttributeError:
361 pass
362 try:
363 return obj[attribute]
Christopher Grebsf1c940f2008-07-10 11:52:17 +0200364 except (TypeError, LookupError, AttributeError):
Armin Ronacher6dc6f292008-06-12 08:50:07 +0200365 return self.undefined(obj=obj, name=attribute)
366
Armin Ronacherd416a972009-02-24 22:58:00 +0100367 @internalcode
Armin Ronacher7f15ef82008-05-16 09:11:39 +0200368 def parse(self, source, name=None, filename=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200369 """Parse the sourcecode and return the abstract syntax tree. This
370 tree of nodes is used by the compiler to convert the template into
371 executable source- or bytecode. This is useful for debugging or to
372 extract information from templates.
Armin Ronachered98cac2008-05-07 08:42:11 +0200373
374 If you are :ref:`developing Jinja2 extensions <writing-extensions>`
375 this gives you a good overview of the node tree generated.
Armin Ronacher07bc6842008-03-31 14:18:49 +0200376 """
Armin Ronacheraaf010d2008-05-01 13:14:30 +0200377 try:
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700378 return self._parse(source, name, filename)
Armin Ronacher2a791922009-04-16 23:15:22 +0200379 except TemplateSyntaxError:
Armin Ronacherbd357722009-08-05 20:25:06 +0200380 exc_info = sys.exc_info()
381 self.handle_exception(exc_info, source_hint=source)
Armin Ronacher07bc6842008-03-31 14:18:49 +0200382
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700383 def _parse(self, source, name, filename):
384 """Internal parsing function used by `parse` and `compile`."""
Armin Ronacher0d242be2010-02-10 01:35:13 +0100385 return Parser(self, source, name, _encode_filename(filename)).parse()
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700386
Armin Ronacher7f15ef82008-05-16 09:11:39 +0200387 def lex(self, source, name=None, filename=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200388 """Lex the given sourcecode and return a generator that yields
389 tokens as tuples in the form ``(lineno, token_type, value)``.
Armin Ronacher5cdc1ac2008-05-07 12:17:18 +0200390 This can be useful for :ref:`extension development <writing-extensions>`
391 and debugging templates.
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200392
393 This does not perform preprocessing. If you want the preprocessing
394 of the extensions to be applied you have to filter source through
395 the :meth:`preprocess` method.
Armin Ronacher07bc6842008-03-31 14:18:49 +0200396 """
Armin Ronacherccae0552008-10-05 23:08:58 +0200397 source = unicode(source)
398 try:
399 return self.lexer.tokeniter(source, name, filename)
Armin Ronacher2a791922009-04-16 23:15:22 +0200400 except TemplateSyntaxError:
Armin Ronacherbd357722009-08-05 20:25:06 +0200401 exc_info = sys.exc_info()
402 self.handle_exception(exc_info, source_hint=source)
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200403
404 def preprocess(self, source, name=None, filename=None):
405 """Preprocesses the source with all extensions. This is automatically
406 called for all parsing and compiling methods but *not* for :meth:`lex`
407 because there you usually only want the actual source tokenized.
408 """
409 return reduce(lambda s, e: e.preprocess(s, name, filename),
410 self.extensions.itervalues(), unicode(source))
411
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100412 def _tokenize(self, source, name, filename=None, state=None):
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200413 """Called by the parser to do the preprocessing and filtering
414 for all the extensions. Returns a :class:`~jinja2.lexer.TokenStream`.
415 """
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200416 source = self.preprocess(source, name, filename)
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100417 stream = self.lexer.tokenize(source, name, filename, state)
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200418 for ext in self.extensions.itervalues():
Armin Ronacher3e3a9be2008-06-14 12:44:15 +0200419 stream = ext.filter_stream(stream)
420 if not isinstance(stream, TokenStream):
421 stream = TokenStream(stream, name, filename)
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200422 return stream
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200423
Armin Ronacherd416a972009-02-24 22:58:00 +0100424 @internalcode
Armin Ronacher64b08a02010-03-12 03:17:41 +0100425 def compile(self, source, name=None, filename=None, raw=False,
426 defer_init=False):
Armin Ronacherd1342312008-04-28 12:20:12 +0200427 """Compile a node or template source code. The `name` parameter is
428 the load name of the template after it was joined using
429 :meth:`join_path` if necessary, not the filename on the file system.
430 the `filename` parameter is the estimated filename of the template on
431 the file system. If the template came from a database or memory this
Armin Ronacher981cbf62008-05-13 09:12:27 +0200432 can be omitted.
Armin Ronacherd1342312008-04-28 12:20:12 +0200433
434 The return value of this method is a python code object. If the `raw`
435 parameter is `True` the return value will be a string with python
436 code equivalent to the bytecode returned otherwise. This method is
437 mainly used internally.
Armin Ronacher64b08a02010-03-12 03:17:41 +0100438
439 `defer_init` is use internally to aid the module code generator. This
440 causes the generated code to be able to import without the global
441 environment variable to be set.
442
443 .. versionadded:: 2.4
444 `defer_init` parameter added.
Armin Ronacher68f77672008-04-17 11:50:39 +0200445 """
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700446 source_hint = None
447 try:
448 if isinstance(source, basestring):
449 source_hint = source
450 source = self._parse(source, name, filename)
451 if self.optimized:
452 source = optimize(source, self)
Armin Ronacher64b08a02010-03-12 03:17:41 +0100453 source = generate(source, self, name, filename,
454 defer_init=defer_init)
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700455 if raw:
456 return source
457 if filename is None:
458 filename = '<template>'
Armin Ronacher0d242be2010-02-10 01:35:13 +0100459 else:
460 filename = _encode_filename(filename)
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700461 return compile(source, filename, 'exec')
462 except TemplateSyntaxError:
463 exc_info = sys.exc_info()
464 self.handle_exception(exc_info, source_hint=source)
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200465
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100466 def compile_expression(self, source, undefined_to_none=True):
467 """A handy helper method that returns a callable that accepts keyword
468 arguments that appear as variables in the expression. If called it
469 returns the result of the expression.
470
471 This is useful if applications want to use the same rules as Jinja
472 in template "configuration files" or similar situations.
473
474 Example usage:
475
476 >>> env = Environment()
477 >>> expr = env.compile_expression('foo == 42')
478 >>> expr(foo=23)
479 False
480 >>> expr(foo=42)
481 True
482
483 Per default the return value is converted to `None` if the
484 expression returns an undefined value. This can be changed
485 by setting `undefined_to_none` to `False`.
486
487 >>> env.compile_expression('var')() is None
488 True
489 >>> env.compile_expression('var', undefined_to_none=False)()
490 Undefined
491
Armin Ronacher0319c662010-02-09 02:09:10 +0100492 .. versionadded:: 2.1
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100493 """
494 parser = Parser(self, source, state='variable')
Armin Ronacherbd357722009-08-05 20:25:06 +0200495 exc_info = None
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100496 try:
497 expr = parser.parse_expression()
498 if not parser.stream.eos:
499 raise TemplateSyntaxError('chunk after expression',
500 parser.stream.current.lineno,
501 None, None)
Armin Ronacher8346bd72010-03-14 19:43:47 +0100502 expr.set_environment(self)
Armin Ronacher2a791922009-04-16 23:15:22 +0200503 except TemplateSyntaxError:
Armin Ronacherbd357722009-08-05 20:25:06 +0200504 exc_info = sys.exc_info()
505 if exc_info is not None:
506 self.handle_exception(exc_info, source_hint=source)
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100507 body = [nodes.Assign(nodes.Name('result', 'store'), expr, lineno=1)]
508 template = self.from_string(nodes.Template(body, lineno=1))
509 return TemplateExpression(template, undefined_to_none)
510
Armin Ronacher64b08a02010-03-12 03:17:41 +0100511 def compile_templates(self, target, extensions=None, filter_func=None,
Armin Ronacher12a316b2010-03-12 17:59:51 +0100512 zip='deflated', log_function=None,
513 ignore_errors=True, py_compile=False):
Armin Ronacher64b08a02010-03-12 03:17:41 +0100514 """Compiles all the templates the loader can find, compiles them
Armin Ronacher12a316b2010-03-12 17:59:51 +0100515 and stores them in `target`. If `zip` is `None`, instead of in a
516 zipfile, the templates will be will be stored in a directory.
517 By default a deflate zip algorithm is used, to switch to
518 the stored algorithm, `zip` can be set to ``'stored'``.
Armin Ronacher64b08a02010-03-12 03:17:41 +0100519
520 `extensions` and `filter_func` are passed to :meth:`list_templates`.
521 Each template returned will be compiled to the target folder or
522 zipfile.
523
Armin Ronacher12a316b2010-03-12 17:59:51 +0100524 By default template compilation errors are ignored. In case a
525 log function is provided, errors are logged. If you want template
526 syntax errors to abort the compilation you can set `ignore_errors`
527 to `False` and you will get an exception on syntax errors.
528
529 If `py_compile` is set to `True` .pyc files will be written to the
530 target instead of standard .py files.
531
Armin Ronacher64b08a02010-03-12 03:17:41 +0100532 .. versionadded:: 2.4
533 """
534 from jinja2.loaders import ModuleLoader
Armin Ronacher12a316b2010-03-12 17:59:51 +0100535
Armin Ronacher64b08a02010-03-12 03:17:41 +0100536 if log_function is None:
537 log_function = lambda x: None
538
Armin Ronacher12a316b2010-03-12 17:59:51 +0100539 if py_compile:
540 import imp, struct, marshal
Armin Ronacherc57959d2010-03-15 00:54:01 +0100541 py_header = imp.get_magic() + \
542 u'\xff\xff\xff\xff'.encode('iso-8859-15')
Armin Ronacher12a316b2010-03-12 17:59:51 +0100543
Armin Ronacherc57959d2010-03-15 00:54:01 +0100544 def write_file(filename, data, mode):
Armin Ronacher12a316b2010-03-12 17:59:51 +0100545 if zip:
546 info = ZipInfo(filename)
547 info.external_attr = 0755 << 16L
548 zip_file.writestr(info, data)
549 else:
Armin Ronacherc57959d2010-03-15 00:54:01 +0100550 f = open(os.path.join(target, filename), mode)
Armin Ronacher12a316b2010-03-12 17:59:51 +0100551 try:
552 f.write(data)
553 finally:
554 f.close()
555
556 if zip is not None:
557 from zipfile import ZipFile, ZipInfo, ZIP_DEFLATED, ZIP_STORED
558 zip_file = ZipFile(target, 'w', dict(deflated=ZIP_DEFLATED,
559 stored=ZIP_STORED)[zip])
Armin Ronacher64b08a02010-03-12 03:17:41 +0100560 log_function('Compiling into Zip archive "%s"' % target)
561 else:
562 if not os.path.isdir(target):
563 os.makedirs(target)
564 log_function('Compiling into folder "%s"' % target)
565
566 try:
567 for name in self.list_templates(extensions, filter_func):
568 source, filename, _ = self.loader.get_source(self, name)
569 try:
570 code = self.compile(source, name, filename, True, True)
571 except TemplateSyntaxError, e:
Armin Ronacher12a316b2010-03-12 17:59:51 +0100572 if not ignore_errors:
573 raise
Armin Ronacher64b08a02010-03-12 03:17:41 +0100574 log_function('Could not compile "%s": %s' % (name, e))
575 continue
Armin Ronacher12a316b2010-03-12 17:59:51 +0100576
577 filename = ModuleLoader.get_module_filename(name)
578
579 if py_compile:
580 c = compile(code, _encode_filename(filename), 'exec')
Armin Ronacherc57959d2010-03-15 00:54:01 +0100581 write_file(filename + 'c', py_header +
582 marshal.dumps(c), 'wb')
Armin Ronacher12a316b2010-03-12 17:59:51 +0100583 log_function('Byte-compiled "%s" as %s' %
584 (name, filename + 'c'))
Armin Ronacher64b08a02010-03-12 03:17:41 +0100585 else:
Armin Ronacherc57959d2010-03-15 00:54:01 +0100586 write_file(filename, code, 'w')
Armin Ronacher12a316b2010-03-12 17:59:51 +0100587 log_function('Compiled "%s" as %s' % (name, filename))
Armin Ronacher64b08a02010-03-12 03:17:41 +0100588 finally:
589 if zip:
Armin Ronacher12a316b2010-03-12 17:59:51 +0100590 zip_file.close()
Armin Ronacher64b08a02010-03-12 03:17:41 +0100591
592 log_function('Finished compiling templates')
593
594 def list_templates(self, extensions=None, filter_func=None):
595 """Returns a list of templates for this environment. This requires
596 that the loader supports the loader's
597 :meth:`~BaseLoader.list_templates` method.
598
599 If there are other files in the template folder besides the
600 actual templates, the returned list can be filtered. There are two
601 ways: either `extensions` is set to a list of file extensions for
602 templates, or a `filter_func` can be provided which is a callable that
603 is passed a template name and should return `True` if it should end up
604 in the result list.
605
606 If the loader does not support that, a :exc:`TypeError` is raised.
607 """
608 x = self.loader.list_templates()
609 if extensions is not None:
610 if filter_func is not None:
611 raise TypeError('either extensions or filter_func '
612 'can be passed, but not both')
613 filter_func = lambda x: '.' in x and \
614 x.rsplit('.', 1)[1] in extensions
615 if filter_func is not None:
616 x = filter(filter_func, x)
617 return x
618
Armin Ronachera18872d2009-03-05 23:47:00 +0100619 def handle_exception(self, exc_info=None, rendered=False, source_hint=None):
620 """Exception handling helper. This is used internally to either raise
621 rewritten exceptions or return a rendered traceback for the template.
622 """
623 global _make_traceback
624 if exc_info is None:
625 exc_info = sys.exc_info()
Armin Ronacher32ed6c92009-04-02 14:04:41 +0200626
627 # the debugging module is imported when it's used for the first time.
628 # we're doing a lot of stuff there and for applications that do not
629 # get any exceptions in template rendering there is no need to load
630 # all of that.
Armin Ronachera18872d2009-03-05 23:47:00 +0100631 if _make_traceback is None:
632 from jinja2.debug import make_traceback as _make_traceback
633 traceback = _make_traceback(exc_info, source_hint)
634 if rendered and self.exception_formatter is not None:
635 return self.exception_formatter(traceback)
636 if self.exception_handler is not None:
637 self.exception_handler(traceback)
638 exc_type, exc_value, tb = traceback.standard_exc_info
639 raise exc_type, exc_value, tb
640
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200641 def join_path(self, template, parent):
642 """Join a template with the parent. By default all the lookups are
Armin Ronacherd1342312008-04-28 12:20:12 +0200643 relative to the loader root so this method returns the `template`
644 parameter unchanged, but if the paths should be relative to the
645 parent template, this function can be used to calculate the real
646 template name.
647
648 Subclasses may override this method and implement template path
649 joining here.
650 """
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200651 return template
652
Armin Ronacherd416a972009-02-24 22:58:00 +0100653 @internalcode
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100654 def _load_template(self, name, globals):
655 if self.loader is None:
656 raise TypeError('no loader for this environment specified')
657 if self.cache is not None:
658 template = self.cache.get(name)
659 if template is not None and (not self.auto_reload or \
660 template.is_up_to_date):
661 return template
662 template = self.loader.load(self, name, globals)
663 if self.cache is not None:
664 self.cache[name] = template
665 return template
666
667 @internalcode
Armin Ronacherfed44b52008-04-13 19:42:53 +0200668 def get_template(self, name, parent=None, globals=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200669 """Load a template from the loader. If a loader is configured this
670 method ask the loader for the template and returns a :class:`Template`.
671 If the `parent` parameter is not `None`, :meth:`join_path` is called
672 to get the real template name before loading.
673
Armin Ronacher7a519ee2008-09-08 23:10:47 +0200674 The `globals` parameter can be used to provide template wide globals.
Armin Ronacher981cbf62008-05-13 09:12:27 +0200675 These variables are available in the context at render time.
Armin Ronacherd1342312008-04-28 12:20:12 +0200676
677 If the template does not exist a :exc:`TemplateNotFound` exception is
678 raised.
Armin Ronacherc2c63512010-02-16 17:37:17 +0100679
680 .. versionchanged:: 2.4
681 If `name` is a :class:`Template` object it is returned from the
682 function unchanged.
Armin Ronacherd1342312008-04-28 12:20:12 +0200683 """
Armin Ronacher9165d3e2010-02-16 17:35:59 +0100684 if isinstance(name, Template):
685 return name
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200686 if parent is not None:
687 name = self.join_path(name, parent)
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100688 return self._load_template(name, self.make_globals(globals))
Armin Ronacher7259c762008-04-30 13:03:59 +0200689
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100690 @internalcode
691 def select_template(self, names, parent=None, globals=None):
692 """Works like :meth:`get_template` but tries a number of templates
693 before it fails. If it cannot find any of the templates, it will
694 raise a :exc:`TemplatesNotFound` exception.
Armin Ronacher7259c762008-04-30 13:03:59 +0200695
Armin Ronacher0319c662010-02-09 02:09:10 +0100696 .. versionadded:: 2.3
Armin Ronacherc2c63512010-02-16 17:37:17 +0100697
698 .. versionchanged:: 2.4
699 If `names` contains a :class:`Template` object it is returned
700 from the function unchanged.
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100701 """
702 if not names:
703 raise TemplatesNotFound(message=u'Tried to select from an empty list '
704 u'of templates.')
705 globals = self.make_globals(globals)
706 for name in names:
Armin Ronacher9165d3e2010-02-16 17:35:59 +0100707 if isinstance(name, Template):
708 return name
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100709 if parent is not None:
710 name = self.join_path(name, parent)
711 try:
712 return self._load_template(name, globals)
713 except TemplateNotFound:
714 pass
715 raise TemplatesNotFound(names)
716
717 @internalcode
718 def get_or_select_template(self, template_name_or_list,
719 parent=None, globals=None):
Armin Ronacher04306792010-02-17 00:16:07 +0100720 """Does a typecheck and dispatches to :meth:`select_template`
721 if an iterable of template names is given, otherwise to
722 :meth:`get_template`.
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100723
Armin Ronacher0319c662010-02-09 02:09:10 +0100724 .. versionadded:: 2.3
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100725 """
726 if isinstance(template_name_or_list, basestring):
727 return self.get_template(template_name_or_list, parent, globals)
Armin Ronacher9165d3e2010-02-16 17:35:59 +0100728 elif isinstance(template_name_or_list, Template):
729 return template_name_or_list
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100730 return self.select_template(template_name_or_list, parent, globals)
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200731
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200732 def from_string(self, source, globals=None, template_class=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200733 """Load a template from a string. This parses the source given and
734 returns a :class:`Template` object.
735 """
Armin Ronacherfed44b52008-04-13 19:42:53 +0200736 globals = self.make_globals(globals)
Armin Ronacher7259c762008-04-30 13:03:59 +0200737 cls = template_class or self.template_class
Armin Ronacher981cbf62008-05-13 09:12:27 +0200738 return cls.from_code(self, self.compile(source), globals, None)
Armin Ronacherfed44b52008-04-13 19:42:53 +0200739
740 def make_globals(self, d):
741 """Return a dict for the globals."""
Armin Ronacher5411ce72008-05-25 11:36:22 +0200742 if not d:
Armin Ronacherfed44b52008-04-13 19:42:53 +0200743 return self.globals
744 return dict(self.globals, **d)
Armin Ronacher46f5f982008-04-11 16:40:09 +0200745
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200746
747class Template(object):
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200748 """The central template object. This class represents a compiled template
749 and is used to evaluate it.
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200750
Armin Ronacherd1342312008-04-28 12:20:12 +0200751 Normally the template object is generated from an :class:`Environment` but
752 it also has a constructor that makes it possible to create a template
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200753 instance directly using the constructor. It takes the same arguments as
754 the environment constructor but it's not possible to specify a loader.
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200755
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200756 Every template object has a few methods and members that are guaranteed
757 to exist. However it's important that a template object should be
758 considered immutable. Modifications on the object are not supported.
759
760 Template objects created from the constructor rather than an environment
761 do have an `environment` attribute that points to a temporary environment
762 that is probably shared with other templates created with the constructor
763 and compatible settings.
764
765 >>> template = Template('Hello {{ name }}!')
766 >>> template.render(name='John Doe')
767 u'Hello John Doe!'
768
769 >>> stream = template.stream(name='John Doe')
770 >>> stream.next()
771 u'Hello John Doe!'
772 >>> stream.next()
773 Traceback (most recent call last):
774 ...
775 StopIteration
776 """
777
778 def __new__(cls, source,
Armin Ronacher4f5008f2008-05-23 23:36:07 +0200779 block_start_string=BLOCK_START_STRING,
780 block_end_string=BLOCK_END_STRING,
781 variable_start_string=VARIABLE_START_STRING,
782 variable_end_string=VARIABLE_END_STRING,
783 comment_start_string=COMMENT_START_STRING,
784 comment_end_string=COMMENT_END_STRING,
785 line_statement_prefix=LINE_STATEMENT_PREFIX,
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200786 line_comment_prefix=LINE_COMMENT_PREFIX,
Armin Ronacher4f5008f2008-05-23 23:36:07 +0200787 trim_blocks=TRIM_BLOCKS,
788 newline_sequence=NEWLINE_SEQUENCE,
Armin Ronacherb5124e62008-04-25 00:36:14 +0200789 extensions=(),
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200790 optimized=True,
791 undefined=Undefined,
Armin Ronacherd1342312008-04-28 12:20:12 +0200792 finalize=None,
793 autoescape=False):
Armin Ronacherb5124e62008-04-25 00:36:14 +0200794 env = get_spontaneous_environment(
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200795 block_start_string, block_end_string, variable_start_string,
796 variable_end_string, comment_start_string, comment_end_string,
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200797 line_statement_prefix, line_comment_prefix, trim_blocks,
798 newline_sequence, frozenset(extensions), optimized, undefined,
799 finalize, autoescape, None, 0, False, None)
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200800 return env.from_string(source, template_class=cls)
Armin Ronacherba3757b2008-04-16 19:43:16 +0200801
Armin Ronacher7259c762008-04-30 13:03:59 +0200802 @classmethod
803 def from_code(cls, environment, code, globals, uptodate=None):
804 """Creates a template object from compiled code and the globals. This
805 is used by the loaders and environment to create a template object.
806 """
Armin Ronacher7259c762008-04-30 13:03:59 +0200807 namespace = {
Armin Ronacher64b08a02010-03-12 03:17:41 +0100808 'environment': environment,
809 '__file__': code.co_filename
Armin Ronacher7259c762008-04-30 13:03:59 +0200810 }
811 exec code in namespace
Armin Ronacher64b08a02010-03-12 03:17:41 +0100812 rv = cls._from_namespace(environment, namespace, globals)
813 rv._uptodate = uptodate
814 return rv
815
816 @classmethod
817 def from_module_dict(cls, environment, module_dict, globals):
818 """Creates a template object from a module. This is used by the
819 module loader to create a template object.
820
821 .. versionadded:: 2.4
822 """
823 return cls._from_namespace(environment, module_dict, globals)
824
825 @classmethod
826 def _from_namespace(cls, environment, namespace, globals):
827 t = object.__new__(cls)
Armin Ronacher7259c762008-04-30 13:03:59 +0200828 t.environment = environment
Armin Ronacher771c7502008-05-18 23:14:14 +0200829 t.globals = globals
Armin Ronacher7259c762008-04-30 13:03:59 +0200830 t.name = namespace['name']
Armin Ronacher64b08a02010-03-12 03:17:41 +0100831 t.filename = namespace['__file__']
Armin Ronacher7259c762008-04-30 13:03:59 +0200832 t.blocks = namespace['blocks']
Armin Ronacher771c7502008-05-18 23:14:14 +0200833
Georg Brandl3e497b72008-09-19 09:55:17 +0000834 # render function and module
Armin Ronacher5411ce72008-05-25 11:36:22 +0200835 t.root_render_func = namespace['root']
Armin Ronacher771c7502008-05-18 23:14:14 +0200836 t._module = None
Armin Ronacher7259c762008-04-30 13:03:59 +0200837
838 # debug and loader helpers
839 t._debug_info = namespace['debug_info']
Armin Ronacher64b08a02010-03-12 03:17:41 +0100840 t._uptodate = None
841
842 # store the reference
843 namespace['environment'] = environment
844 namespace['__jinja_template__'] = t
Armin Ronacher7259c762008-04-30 13:03:59 +0200845
846 return t
847
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200848 def render(self, *args, **kwargs):
Armin Ronacherd1342312008-04-28 12:20:12 +0200849 """This method accepts the same arguments as the `dict` constructor:
850 A dict, a dict subclass or some keyword arguments. If no arguments
851 are given the context will be empty. These two calls do the same::
852
853 template.render(knights='that say nih')
854 template.render({'knights': 'that say nih'})
855
856 This will return the rendered template as unicode string.
857 """
Armin Ronacher771c7502008-05-18 23:14:14 +0200858 vars = dict(*args, **kwargs)
Armin Ronacherf41d1392008-04-18 16:41:52 +0200859 try:
Armin Ronacher5411ce72008-05-25 11:36:22 +0200860 return concat(self.root_render_func(self.new_context(vars)))
Armin Ronacherf41d1392008-04-18 16:41:52 +0200861 except:
Armin Ronacherbd357722009-08-05 20:25:06 +0200862 exc_info = sys.exc_info()
863 return self.environment.handle_exception(exc_info, True)
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200864
865 def stream(self, *args, **kwargs):
Armin Ronacherd1342312008-04-28 12:20:12 +0200866 """Works exactly like :meth:`generate` but returns a
867 :class:`TemplateStream`.
868 """
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200869 return TemplateStream(self.generate(*args, **kwargs))
Armin Ronacherfed44b52008-04-13 19:42:53 +0200870
871 def generate(self, *args, **kwargs):
Armin Ronacherd1342312008-04-28 12:20:12 +0200872 """For very large templates it can be useful to not render the whole
873 template at once but evaluate each statement after another and yield
874 piece for piece. This method basically does exactly that and returns
875 a generator that yields one item after another as unicode strings.
876
877 It accepts the same arguments as :meth:`render`.
878 """
Armin Ronacher771c7502008-05-18 23:14:14 +0200879 vars = dict(*args, **kwargs)
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200880 try:
Armin Ronacher5411ce72008-05-25 11:36:22 +0200881 for event in self.root_render_func(self.new_context(vars)):
Armin Ronacher771c7502008-05-18 23:14:14 +0200882 yield event
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200883 except:
Armin Ronacherbd357722009-08-05 20:25:06 +0200884 exc_info = sys.exc_info()
885 else:
886 return
887 yield self.environment.handle_exception(exc_info, True)
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200888
Armin Ronacher673aa882008-10-04 18:06:57 +0200889 def new_context(self, vars=None, shared=False, locals=None):
Armin Ronacher5411ce72008-05-25 11:36:22 +0200890 """Create a new :class:`Context` for this template. The vars
Armin Ronacherc9705c22008-04-27 21:28:03 +0200891 provided will be passed to the template. Per default the globals
Armin Ronacher673aa882008-10-04 18:06:57 +0200892 are added to the context. If shared is set to `True` the data
893 is passed as it to the context without adding the globals.
894
895 `locals` can be a dict of local variables for internal usage.
Armin Ronacherc9705c22008-04-27 21:28:03 +0200896 """
Armin Ronacher74a0cd92009-02-19 15:56:53 +0100897 return new_context(self.environment, self.name, self.blocks,
898 vars, shared, self.globals, locals)
Armin Ronacherba3757b2008-04-16 19:43:16 +0200899
Armin Ronacher673aa882008-10-04 18:06:57 +0200900 def make_module(self, vars=None, shared=False, locals=None):
Armin Ronacher7ceced52008-05-03 10:15:31 +0200901 """This method works like the :attr:`module` attribute when called
Armin Ronacher0aa0f582009-03-18 01:01:36 +0100902 without arguments but it will evaluate the template on every call
903 rather than caching it. It's also possible to provide
Armin Ronacher7ceced52008-05-03 10:15:31 +0200904 a dict which is then used as context. The arguments are the same
Armin Ronacherf3c35c42008-05-23 23:18:14 +0200905 as for the :meth:`new_context` method.
Armin Ronacherea847c52008-05-02 20:04:32 +0200906 """
Armin Ronacher673aa882008-10-04 18:06:57 +0200907 return TemplateModule(self, self.new_context(vars, shared, locals))
Armin Ronacherea847c52008-05-02 20:04:32 +0200908
Armin Ronacherd84ec462008-04-29 13:43:16 +0200909 @property
910 def module(self):
911 """The template as module. This is used for imports in the
912 template runtime but is also useful if one wants to access
913 exported template variables from the Python layer:
Armin Ronacherd1342312008-04-28 12:20:12 +0200914
Armin Ronacherd84ec462008-04-29 13:43:16 +0200915 >>> t = Template('{% macro foo() %}42{% endmacro %}23')
916 >>> unicode(t.module)
917 u'23'
918 >>> t.module.foo()
Armin Ronacherd1342312008-04-28 12:20:12 +0200919 u'42'
Armin Ronacher6ce170c2008-04-25 12:32:36 +0200920 """
Armin Ronacher771c7502008-05-18 23:14:14 +0200921 if self._module is not None:
Armin Ronacherd84ec462008-04-29 13:43:16 +0200922 return self._module
Armin Ronacherea847c52008-05-02 20:04:32 +0200923 self._module = rv = self.make_module()
Armin Ronacherd84ec462008-04-29 13:43:16 +0200924 return rv
Armin Ronacher963f97d2008-04-25 11:44:59 +0200925
Armin Ronacherba3757b2008-04-16 19:43:16 +0200926 def get_corresponding_lineno(self, lineno):
927 """Return the source line number of a line number in the
928 generated bytecode as they are not in sync.
929 """
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200930 for template_line, code_line in reversed(self.debug_info):
Armin Ronacherba3757b2008-04-16 19:43:16 +0200931 if code_line <= lineno:
932 return template_line
933 return 1
Armin Ronacherc63243e2008-04-14 22:53:58 +0200934
Armin Ronacher9a822052008-04-17 18:44:07 +0200935 @property
Armin Ronacher814f6c22008-04-17 15:52:23 +0200936 def is_up_to_date(self):
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200937 """If this variable is `False` there is a newer version available."""
Armin Ronacher814f6c22008-04-17 15:52:23 +0200938 if self._uptodate is None:
939 return True
940 return self._uptodate()
941
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200942 @property
943 def debug_info(self):
944 """The debug info mapping."""
945 return [tuple(map(int, x.split('='))) for x in
946 self._debug_info.split('&')]
947
Armin Ronacherc63243e2008-04-14 22:53:58 +0200948 def __repr__(self):
Armin Ronacher53042292008-04-26 18:30:19 +0200949 if self.name is None:
950 name = 'memory:%x' % id(self)
951 else:
952 name = repr(self.name)
953 return '<%s %s>' % (self.__class__.__name__, name)
Armin Ronacherc63243e2008-04-14 22:53:58 +0200954
955
Armin Ronacherd84ec462008-04-29 13:43:16 +0200956class TemplateModule(object):
957 """Represents an imported template. All the exported names of the
Armin Ronacher53042292008-04-26 18:30:19 +0200958 template are available as attributes on this object. Additionally
959 converting it into an unicode- or bytestrings renders the contents.
960 """
Armin Ronacher963f97d2008-04-25 11:44:59 +0200961
962 def __init__(self, template, context):
Armin Ronacher5411ce72008-05-25 11:36:22 +0200963 self._body_stream = list(template.root_render_func(context))
Armin Ronacher6ce170c2008-04-25 12:32:36 +0200964 self.__dict__.update(context.get_exported())
Armin Ronacher2feed1d2008-04-26 16:26:52 +0200965 self.__name__ = template.name
Armin Ronacher963f97d2008-04-25 11:44:59 +0200966
Armin Ronacher0faa8612010-02-09 15:04:51 +0100967 def __html__(self):
968 return Markup(concat(self._body_stream))
Armin Ronacher6ce170c2008-04-25 12:32:36 +0200969
970 def __str__(self):
Armin Ronacher2feed1d2008-04-26 16:26:52 +0200971 return unicode(self).encode('utf-8')
Armin Ronacher963f97d2008-04-25 11:44:59 +0200972
Armin Ronacheracbd4082010-02-10 00:07:43 +0100973 # unicode goes after __str__ because we configured 2to3 to rename
974 # __unicode__ to __str__. because the 2to3 tree is not designed to
975 # remove nodes from it, we leave the above __str__ around and let
976 # it override at runtime.
Armin Ronacher790b8a82010-02-10 00:05:46 +0100977 def __unicode__(self):
978 return concat(self._body_stream)
979
Armin Ronacher963f97d2008-04-25 11:44:59 +0200980 def __repr__(self):
Armin Ronacher53042292008-04-26 18:30:19 +0200981 if self.__name__ is None:
982 name = 'memory:%x' % id(self)
983 else:
Armin Ronacherdc02b642008-05-15 22:47:27 +0200984 name = repr(self.__name__)
Armin Ronacher53042292008-04-26 18:30:19 +0200985 return '<%s %s>' % (self.__class__.__name__, name)
Armin Ronacher963f97d2008-04-25 11:44:59 +0200986
987
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100988class TemplateExpression(object):
989 """The :meth:`jinja2.Environment.compile_expression` method returns an
990 instance of this object. It encapsulates the expression-like access
991 to the template with an expression it wraps.
992 """
993
994 def __init__(self, template, undefined_to_none):
995 self._template = template
996 self._undefined_to_none = undefined_to_none
997
998 def __call__(self, *args, **kwargs):
999 context = self._template.new_context(dict(*args, **kwargs))
1000 consume(self._template.root_render_func(context))
1001 rv = context.vars['result']
1002 if self._undefined_to_none and isinstance(rv, Undefined):
1003 rv = None
1004 return rv
1005
1006
Armin Ronacherc63243e2008-04-14 22:53:58 +02001007class TemplateStream(object):
Armin Ronacherd1342312008-04-28 12:20:12 +02001008 """A template stream works pretty much like an ordinary python generator
1009 but it can buffer multiple items to reduce the number of total iterations.
1010 Per default the output is unbuffered which means that for every unbuffered
1011 instruction in the template one unicode string is yielded.
1012
1013 If buffering is enabled with a buffer size of 5, five items are combined
1014 into a new unicode string. This is mainly useful if you are streaming
1015 big templates to a client via WSGI which flushes after each iteration.
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001016 """
Armin Ronacherc63243e2008-04-14 22:53:58 +02001017
1018 def __init__(self, gen):
1019 self._gen = gen
Armin Ronacher9cf95912008-05-24 19:54:43 +02001020 self.disable_buffering()
Armin Ronacherc63243e2008-04-14 22:53:58 +02001021
Armin Ronacher74b51062008-06-17 11:28:59 +02001022 def dump(self, fp, encoding=None, errors='strict'):
1023 """Dump the complete stream into a file or file-like object.
1024 Per default unicode strings are written, if you want to encode
1025 before writing specifiy an `encoding`.
1026
1027 Example usage::
1028
1029 Template('Hello {{ name }}!').stream(name='foo').dump('hello.html')
1030 """
1031 close = False
1032 if isinstance(fp, basestring):
1033 fp = file(fp, 'w')
1034 close = True
1035 try:
1036 if encoding is not None:
1037 iterable = (x.encode(encoding, errors) for x in self)
1038 else:
1039 iterable = self
1040 if hasattr(fp, 'writelines'):
1041 fp.writelines(iterable)
1042 else:
1043 for item in iterable:
1044 fp.write(item)
1045 finally:
1046 if close:
1047 fp.close()
1048
Armin Ronacherc63243e2008-04-14 22:53:58 +02001049 def disable_buffering(self):
1050 """Disable the output buffering."""
1051 self._next = self._gen.next
1052 self.buffered = False
1053
1054 def enable_buffering(self, size=5):
Armin Ronacherd1342312008-04-28 12:20:12 +02001055 """Enable buffering. Buffer `size` items before yielding them."""
Armin Ronacherc63243e2008-04-14 22:53:58 +02001056 if size <= 1:
1057 raise ValueError('buffer size too small')
Armin Ronacherc63243e2008-04-14 22:53:58 +02001058
Armin Ronacher5dfbfc12008-05-25 18:10:12 +02001059 def generator(next):
Armin Ronacherc63243e2008-04-14 22:53:58 +02001060 buf = []
1061 c_size = 0
1062 push = buf.append
Armin Ronacherc63243e2008-04-14 22:53:58 +02001063
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001064 while 1:
1065 try:
Armin Ronacherb5124e62008-04-25 00:36:14 +02001066 while c_size < size:
Armin Ronacher981cbf62008-05-13 09:12:27 +02001067 c = next()
1068 push(c)
1069 if c:
1070 c_size += 1
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001071 except StopIteration:
1072 if not c_size:
Armin Ronacherd84ec462008-04-29 13:43:16 +02001073 return
Armin Ronacherde6bf712008-04-26 01:44:14 +02001074 yield concat(buf)
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001075 del buf[:]
1076 c_size = 0
Armin Ronacherc63243e2008-04-14 22:53:58 +02001077
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001078 self.buffered = True
Armin Ronacher5dfbfc12008-05-25 18:10:12 +02001079 self._next = generator(self._gen.next).next
Armin Ronacherc63243e2008-04-14 22:53:58 +02001080
1081 def __iter__(self):
1082 return self
1083
1084 def next(self):
1085 return self._next()
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001086
1087
1088# hook in default template class. if anyone reads this comment: ignore that
1089# it's possible to use custom templates ;-)
1090Environment.template_class = Template