blob: 529c14c407d0a1b29050cb11e9161730b394b19f [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 Ronacher5b3f4dc2010-04-12 14:04:14 +0200337 def iter_extensions(self):
338 """Iterates over the extensions by priority."""
339 return iter(sorted(self.extensions.values(),
340 key=lambda x: x.priority))
341
Armin Ronacher6dc6f292008-06-12 08:50:07 +0200342 def getitem(self, obj, argument):
343 """Get an item or attribute of an object but prefer the item."""
Armin Ronacher08a6a3b2008-05-13 15:35:47 +0200344 try:
345 return obj[argument]
346 except (TypeError, LookupError):
Armin Ronacherf15f5f72008-05-26 12:21:45 +0200347 if isinstance(argument, basestring):
348 try:
349 attr = str(argument)
350 except:
351 pass
352 else:
353 try:
354 return getattr(obj, attr)
355 except AttributeError:
356 pass
Armin Ronacher08a6a3b2008-05-13 15:35:47 +0200357 return self.undefined(obj=obj, name=argument)
Armin Ronacherc63243e2008-04-14 22:53:58 +0200358
Armin Ronacher6dc6f292008-06-12 08:50:07 +0200359 def getattr(self, obj, attribute):
360 """Get an item or attribute of an object but prefer the attribute.
361 Unlike :meth:`getitem` the attribute *must* be a bytestring.
362 """
363 try:
364 return getattr(obj, attribute)
365 except AttributeError:
366 pass
367 try:
368 return obj[attribute]
Christopher Grebsf1c940f2008-07-10 11:52:17 +0200369 except (TypeError, LookupError, AttributeError):
Armin Ronacher6dc6f292008-06-12 08:50:07 +0200370 return self.undefined(obj=obj, name=attribute)
371
Armin Ronacherd416a972009-02-24 22:58:00 +0100372 @internalcode
Armin Ronacher7f15ef82008-05-16 09:11:39 +0200373 def parse(self, source, name=None, filename=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200374 """Parse the sourcecode and return the abstract syntax tree. This
375 tree of nodes is used by the compiler to convert the template into
376 executable source- or bytecode. This is useful for debugging or to
377 extract information from templates.
Armin Ronachered98cac2008-05-07 08:42:11 +0200378
379 If you are :ref:`developing Jinja2 extensions <writing-extensions>`
380 this gives you a good overview of the node tree generated.
Armin Ronacher07bc6842008-03-31 14:18:49 +0200381 """
Armin Ronacheraaf010d2008-05-01 13:14:30 +0200382 try:
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700383 return self._parse(source, name, filename)
Armin Ronacher2a791922009-04-16 23:15:22 +0200384 except TemplateSyntaxError:
Armin Ronacherbd357722009-08-05 20:25:06 +0200385 exc_info = sys.exc_info()
386 self.handle_exception(exc_info, source_hint=source)
Armin Ronacher07bc6842008-03-31 14:18:49 +0200387
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700388 def _parse(self, source, name, filename):
389 """Internal parsing function used by `parse` and `compile`."""
Armin Ronacher0d242be2010-02-10 01:35:13 +0100390 return Parser(self, source, name, _encode_filename(filename)).parse()
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700391
Armin Ronacher7f15ef82008-05-16 09:11:39 +0200392 def lex(self, source, name=None, filename=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200393 """Lex the given sourcecode and return a generator that yields
394 tokens as tuples in the form ``(lineno, token_type, value)``.
Armin Ronacher5cdc1ac2008-05-07 12:17:18 +0200395 This can be useful for :ref:`extension development <writing-extensions>`
396 and debugging templates.
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200397
398 This does not perform preprocessing. If you want the preprocessing
399 of the extensions to be applied you have to filter source through
400 the :meth:`preprocess` method.
Armin Ronacher07bc6842008-03-31 14:18:49 +0200401 """
Armin Ronacherccae0552008-10-05 23:08:58 +0200402 source = unicode(source)
403 try:
404 return self.lexer.tokeniter(source, name, filename)
Armin Ronacher2a791922009-04-16 23:15:22 +0200405 except TemplateSyntaxError:
Armin Ronacherbd357722009-08-05 20:25:06 +0200406 exc_info = sys.exc_info()
407 self.handle_exception(exc_info, source_hint=source)
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200408
409 def preprocess(self, source, name=None, filename=None):
410 """Preprocesses the source with all extensions. This is automatically
411 called for all parsing and compiling methods but *not* for :meth:`lex`
412 because there you usually only want the actual source tokenized.
413 """
414 return reduce(lambda s, e: e.preprocess(s, name, filename),
Armin Ronacher5b3f4dc2010-04-12 14:04:14 +0200415 self.iter_extensions(), unicode(source))
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200416
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100417 def _tokenize(self, source, name, filename=None, state=None):
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200418 """Called by the parser to do the preprocessing and filtering
419 for all the extensions. Returns a :class:`~jinja2.lexer.TokenStream`.
420 """
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200421 source = self.preprocess(source, name, filename)
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100422 stream = self.lexer.tokenize(source, name, filename, state)
Armin Ronacher5b3f4dc2010-04-12 14:04:14 +0200423 for ext in self.iter_extensions():
Armin Ronacher3e3a9be2008-06-14 12:44:15 +0200424 stream = ext.filter_stream(stream)
425 if not isinstance(stream, TokenStream):
426 stream = TokenStream(stream, name, filename)
Armin Ronacher9ad96e72008-06-13 22:44:01 +0200427 return stream
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200428
Armin Ronacherd416a972009-02-24 22:58:00 +0100429 @internalcode
Armin Ronacher64b08a02010-03-12 03:17:41 +0100430 def compile(self, source, name=None, filename=None, raw=False,
431 defer_init=False):
Armin Ronacherd1342312008-04-28 12:20:12 +0200432 """Compile a node or template source code. The `name` parameter is
433 the load name of the template after it was joined using
434 :meth:`join_path` if necessary, not the filename on the file system.
435 the `filename` parameter is the estimated filename of the template on
436 the file system. If the template came from a database or memory this
Armin Ronacher981cbf62008-05-13 09:12:27 +0200437 can be omitted.
Armin Ronacherd1342312008-04-28 12:20:12 +0200438
439 The return value of this method is a python code object. If the `raw`
440 parameter is `True` the return value will be a string with python
441 code equivalent to the bytecode returned otherwise. This method is
442 mainly used internally.
Armin Ronacher64b08a02010-03-12 03:17:41 +0100443
444 `defer_init` is use internally to aid the module code generator. This
445 causes the generated code to be able to import without the global
446 environment variable to be set.
447
448 .. versionadded:: 2.4
449 `defer_init` parameter added.
Armin Ronacher68f77672008-04-17 11:50:39 +0200450 """
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700451 source_hint = None
452 try:
453 if isinstance(source, basestring):
454 source_hint = source
455 source = self._parse(source, name, filename)
456 if self.optimized:
457 source = optimize(source, self)
Armin Ronacher64b08a02010-03-12 03:17:41 +0100458 source = generate(source, self, name, filename,
459 defer_init=defer_init)
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700460 if raw:
461 return source
462 if filename is None:
463 filename = '<template>'
Armin Ronacher0d242be2010-02-10 01:35:13 +0100464 else:
465 filename = _encode_filename(filename)
Armin Ronacherefcc0e52009-09-13 00:22:50 -0700466 return compile(source, filename, 'exec')
467 except TemplateSyntaxError:
468 exc_info = sys.exc_info()
469 self.handle_exception(exc_info, source_hint=source)
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200470
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100471 def compile_expression(self, source, undefined_to_none=True):
472 """A handy helper method that returns a callable that accepts keyword
473 arguments that appear as variables in the expression. If called it
474 returns the result of the expression.
475
476 This is useful if applications want to use the same rules as Jinja
477 in template "configuration files" or similar situations.
478
479 Example usage:
480
481 >>> env = Environment()
482 >>> expr = env.compile_expression('foo == 42')
483 >>> expr(foo=23)
484 False
485 >>> expr(foo=42)
486 True
487
488 Per default the return value is converted to `None` if the
489 expression returns an undefined value. This can be changed
490 by setting `undefined_to_none` to `False`.
491
492 >>> env.compile_expression('var')() is None
493 True
494 >>> env.compile_expression('var', undefined_to_none=False)()
495 Undefined
496
Armin Ronacher0319c662010-02-09 02:09:10 +0100497 .. versionadded:: 2.1
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100498 """
499 parser = Parser(self, source, state='variable')
Armin Ronacherbd357722009-08-05 20:25:06 +0200500 exc_info = None
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100501 try:
502 expr = parser.parse_expression()
503 if not parser.stream.eos:
504 raise TemplateSyntaxError('chunk after expression',
505 parser.stream.current.lineno,
506 None, None)
Armin Ronacher8346bd72010-03-14 19:43:47 +0100507 expr.set_environment(self)
Armin Ronacher2a791922009-04-16 23:15:22 +0200508 except TemplateSyntaxError:
Armin Ronacherbd357722009-08-05 20:25:06 +0200509 exc_info = sys.exc_info()
510 if exc_info is not None:
511 self.handle_exception(exc_info, source_hint=source)
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100512 body = [nodes.Assign(nodes.Name('result', 'store'), expr, lineno=1)]
513 template = self.from_string(nodes.Template(body, lineno=1))
514 return TemplateExpression(template, undefined_to_none)
515
Armin Ronacher64b08a02010-03-12 03:17:41 +0100516 def compile_templates(self, target, extensions=None, filter_func=None,
Armin Ronacher12a316b2010-03-12 17:59:51 +0100517 zip='deflated', log_function=None,
518 ignore_errors=True, py_compile=False):
Armin Ronacher64b08a02010-03-12 03:17:41 +0100519 """Compiles all the templates the loader can find, compiles them
Armin Ronacher12a316b2010-03-12 17:59:51 +0100520 and stores them in `target`. If `zip` is `None`, instead of in a
521 zipfile, the templates will be will be stored in a directory.
522 By default a deflate zip algorithm is used, to switch to
523 the stored algorithm, `zip` can be set to ``'stored'``.
Armin Ronacher64b08a02010-03-12 03:17:41 +0100524
525 `extensions` and `filter_func` are passed to :meth:`list_templates`.
526 Each template returned will be compiled to the target folder or
527 zipfile.
528
Armin Ronacher12a316b2010-03-12 17:59:51 +0100529 By default template compilation errors are ignored. In case a
530 log function is provided, errors are logged. If you want template
531 syntax errors to abort the compilation you can set `ignore_errors`
532 to `False` and you will get an exception on syntax errors.
533
534 If `py_compile` is set to `True` .pyc files will be written to the
535 target instead of standard .py files.
536
Armin Ronacher64b08a02010-03-12 03:17:41 +0100537 .. versionadded:: 2.4
538 """
539 from jinja2.loaders import ModuleLoader
Armin Ronacher12a316b2010-03-12 17:59:51 +0100540
Armin Ronacher64b08a02010-03-12 03:17:41 +0100541 if log_function is None:
542 log_function = lambda x: None
543
Armin Ronacher12a316b2010-03-12 17:59:51 +0100544 if py_compile:
545 import imp, struct, marshal
Armin Ronacherc57959d2010-03-15 00:54:01 +0100546 py_header = imp.get_magic() + \
547 u'\xff\xff\xff\xff'.encode('iso-8859-15')
Armin Ronacher12a316b2010-03-12 17:59:51 +0100548
Armin Ronacherc57959d2010-03-15 00:54:01 +0100549 def write_file(filename, data, mode):
Armin Ronacher12a316b2010-03-12 17:59:51 +0100550 if zip:
551 info = ZipInfo(filename)
552 info.external_attr = 0755 << 16L
553 zip_file.writestr(info, data)
554 else:
Armin Ronacherc57959d2010-03-15 00:54:01 +0100555 f = open(os.path.join(target, filename), mode)
Armin Ronacher12a316b2010-03-12 17:59:51 +0100556 try:
557 f.write(data)
558 finally:
559 f.close()
560
561 if zip is not None:
562 from zipfile import ZipFile, ZipInfo, ZIP_DEFLATED, ZIP_STORED
563 zip_file = ZipFile(target, 'w', dict(deflated=ZIP_DEFLATED,
564 stored=ZIP_STORED)[zip])
Armin Ronacher64b08a02010-03-12 03:17:41 +0100565 log_function('Compiling into Zip archive "%s"' % target)
566 else:
567 if not os.path.isdir(target):
568 os.makedirs(target)
569 log_function('Compiling into folder "%s"' % target)
570
571 try:
572 for name in self.list_templates(extensions, filter_func):
573 source, filename, _ = self.loader.get_source(self, name)
574 try:
575 code = self.compile(source, name, filename, True, True)
576 except TemplateSyntaxError, e:
Armin Ronacher12a316b2010-03-12 17:59:51 +0100577 if not ignore_errors:
578 raise
Armin Ronacher64b08a02010-03-12 03:17:41 +0100579 log_function('Could not compile "%s": %s' % (name, e))
580 continue
Armin Ronacher12a316b2010-03-12 17:59:51 +0100581
582 filename = ModuleLoader.get_module_filename(name)
583
584 if py_compile:
585 c = compile(code, _encode_filename(filename), 'exec')
Armin Ronacherc57959d2010-03-15 00:54:01 +0100586 write_file(filename + 'c', py_header +
587 marshal.dumps(c), 'wb')
Armin Ronacher12a316b2010-03-12 17:59:51 +0100588 log_function('Byte-compiled "%s" as %s' %
589 (name, filename + 'c'))
Armin Ronacher64b08a02010-03-12 03:17:41 +0100590 else:
Armin Ronacherc57959d2010-03-15 00:54:01 +0100591 write_file(filename, code, 'w')
Armin Ronacher12a316b2010-03-12 17:59:51 +0100592 log_function('Compiled "%s" as %s' % (name, filename))
Armin Ronacher64b08a02010-03-12 03:17:41 +0100593 finally:
594 if zip:
Armin Ronacher12a316b2010-03-12 17:59:51 +0100595 zip_file.close()
Armin Ronacher64b08a02010-03-12 03:17:41 +0100596
597 log_function('Finished compiling templates')
598
599 def list_templates(self, extensions=None, filter_func=None):
600 """Returns a list of templates for this environment. This requires
601 that the loader supports the loader's
602 :meth:`~BaseLoader.list_templates` method.
603
604 If there are other files in the template folder besides the
605 actual templates, the returned list can be filtered. There are two
606 ways: either `extensions` is set to a list of file extensions for
607 templates, or a `filter_func` can be provided which is a callable that
608 is passed a template name and should return `True` if it should end up
609 in the result list.
610
611 If the loader does not support that, a :exc:`TypeError` is raised.
612 """
613 x = self.loader.list_templates()
614 if extensions is not None:
615 if filter_func is not None:
616 raise TypeError('either extensions or filter_func '
617 'can be passed, but not both')
618 filter_func = lambda x: '.' in x and \
619 x.rsplit('.', 1)[1] in extensions
620 if filter_func is not None:
621 x = filter(filter_func, x)
622 return x
623
Armin Ronachera18872d2009-03-05 23:47:00 +0100624 def handle_exception(self, exc_info=None, rendered=False, source_hint=None):
625 """Exception handling helper. This is used internally to either raise
626 rewritten exceptions or return a rendered traceback for the template.
627 """
628 global _make_traceback
629 if exc_info is None:
630 exc_info = sys.exc_info()
Armin Ronacher32ed6c92009-04-02 14:04:41 +0200631
632 # the debugging module is imported when it's used for the first time.
633 # we're doing a lot of stuff there and for applications that do not
634 # get any exceptions in template rendering there is no need to load
635 # all of that.
Armin Ronachera18872d2009-03-05 23:47:00 +0100636 if _make_traceback is None:
637 from jinja2.debug import make_traceback as _make_traceback
638 traceback = _make_traceback(exc_info, source_hint)
639 if rendered and self.exception_formatter is not None:
640 return self.exception_formatter(traceback)
641 if self.exception_handler is not None:
642 self.exception_handler(traceback)
643 exc_type, exc_value, tb = traceback.standard_exc_info
644 raise exc_type, exc_value, tb
645
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200646 def join_path(self, template, parent):
647 """Join a template with the parent. By default all the lookups are
Armin Ronacherd1342312008-04-28 12:20:12 +0200648 relative to the loader root so this method returns the `template`
649 parameter unchanged, but if the paths should be relative to the
650 parent template, this function can be used to calculate the real
651 template name.
652
653 Subclasses may override this method and implement template path
654 joining here.
655 """
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200656 return template
657
Armin Ronacherd416a972009-02-24 22:58:00 +0100658 @internalcode
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100659 def _load_template(self, name, globals):
660 if self.loader is None:
661 raise TypeError('no loader for this environment specified')
662 if self.cache is not None:
663 template = self.cache.get(name)
664 if template is not None and (not self.auto_reload or \
665 template.is_up_to_date):
666 return template
667 template = self.loader.load(self, name, globals)
668 if self.cache is not None:
669 self.cache[name] = template
670 return template
671
672 @internalcode
Armin Ronacherfed44b52008-04-13 19:42:53 +0200673 def get_template(self, name, parent=None, globals=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200674 """Load a template from the loader. If a loader is configured this
675 method ask the loader for the template and returns a :class:`Template`.
676 If the `parent` parameter is not `None`, :meth:`join_path` is called
677 to get the real template name before loading.
678
Armin Ronacher7a519ee2008-09-08 23:10:47 +0200679 The `globals` parameter can be used to provide template wide globals.
Armin Ronacher981cbf62008-05-13 09:12:27 +0200680 These variables are available in the context at render time.
Armin Ronacherd1342312008-04-28 12:20:12 +0200681
682 If the template does not exist a :exc:`TemplateNotFound` exception is
683 raised.
Armin Ronacherc2c63512010-02-16 17:37:17 +0100684
685 .. versionchanged:: 2.4
686 If `name` is a :class:`Template` object it is returned from the
687 function unchanged.
Armin Ronacherd1342312008-04-28 12:20:12 +0200688 """
Armin Ronacher9165d3e2010-02-16 17:35:59 +0100689 if isinstance(name, Template):
690 return name
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200691 if parent is not None:
692 name = self.join_path(name, parent)
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100693 return self._load_template(name, self.make_globals(globals))
Armin Ronacher7259c762008-04-30 13:03:59 +0200694
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100695 @internalcode
696 def select_template(self, names, parent=None, globals=None):
697 """Works like :meth:`get_template` but tries a number of templates
698 before it fails. If it cannot find any of the templates, it will
699 raise a :exc:`TemplatesNotFound` exception.
Armin Ronacher7259c762008-04-30 13:03:59 +0200700
Armin Ronacher0319c662010-02-09 02:09:10 +0100701 .. versionadded:: 2.3
Armin Ronacherc2c63512010-02-16 17:37:17 +0100702
703 .. versionchanged:: 2.4
704 If `names` contains a :class:`Template` object it is returned
705 from the function unchanged.
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100706 """
707 if not names:
708 raise TemplatesNotFound(message=u'Tried to select from an empty list '
709 u'of templates.')
710 globals = self.make_globals(globals)
711 for name in names:
Armin Ronacher9165d3e2010-02-16 17:35:59 +0100712 if isinstance(name, Template):
713 return name
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100714 if parent is not None:
715 name = self.join_path(name, parent)
716 try:
717 return self._load_template(name, globals)
718 except TemplateNotFound:
719 pass
720 raise TemplatesNotFound(names)
721
722 @internalcode
723 def get_or_select_template(self, template_name_or_list,
724 parent=None, globals=None):
Armin Ronacher04306792010-02-17 00:16:07 +0100725 """Does a typecheck and dispatches to :meth:`select_template`
726 if an iterable of template names is given, otherwise to
727 :meth:`get_template`.
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100728
Armin Ronacher0319c662010-02-09 02:09:10 +0100729 .. versionadded:: 2.3
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100730 """
731 if isinstance(template_name_or_list, basestring):
732 return self.get_template(template_name_or_list, parent, globals)
Armin Ronacher9165d3e2010-02-16 17:35:59 +0100733 elif isinstance(template_name_or_list, Template):
734 return template_name_or_list
Armin Ronacher31bbd9e2010-01-14 00:41:30 +0100735 return self.select_template(template_name_or_list, parent, globals)
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200736
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200737 def from_string(self, source, globals=None, template_class=None):
Armin Ronacherd1342312008-04-28 12:20:12 +0200738 """Load a template from a string. This parses the source given and
739 returns a :class:`Template` object.
740 """
Armin Ronacherfed44b52008-04-13 19:42:53 +0200741 globals = self.make_globals(globals)
Armin Ronacher7259c762008-04-30 13:03:59 +0200742 cls = template_class or self.template_class
Armin Ronacher981cbf62008-05-13 09:12:27 +0200743 return cls.from_code(self, self.compile(source), globals, None)
Armin Ronacherfed44b52008-04-13 19:42:53 +0200744
745 def make_globals(self, d):
746 """Return a dict for the globals."""
Armin Ronacher5411ce72008-05-25 11:36:22 +0200747 if not d:
Armin Ronacherfed44b52008-04-13 19:42:53 +0200748 return self.globals
749 return dict(self.globals, **d)
Armin Ronacher46f5f982008-04-11 16:40:09 +0200750
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200751
752class Template(object):
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200753 """The central template object. This class represents a compiled template
754 and is used to evaluate it.
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200755
Armin Ronacherd1342312008-04-28 12:20:12 +0200756 Normally the template object is generated from an :class:`Environment` but
757 it also has a constructor that makes it possible to create a template
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200758 instance directly using the constructor. It takes the same arguments as
759 the environment constructor but it's not possible to specify a loader.
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200760
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200761 Every template object has a few methods and members that are guaranteed
762 to exist. However it's important that a template object should be
763 considered immutable. Modifications on the object are not supported.
764
765 Template objects created from the constructor rather than an environment
766 do have an `environment` attribute that points to a temporary environment
767 that is probably shared with other templates created with the constructor
768 and compatible settings.
769
770 >>> template = Template('Hello {{ name }}!')
771 >>> template.render(name='John Doe')
772 u'Hello John Doe!'
773
774 >>> stream = template.stream(name='John Doe')
775 >>> stream.next()
776 u'Hello John Doe!'
777 >>> stream.next()
778 Traceback (most recent call last):
779 ...
780 StopIteration
781 """
782
783 def __new__(cls, source,
Armin Ronacher4f5008f2008-05-23 23:36:07 +0200784 block_start_string=BLOCK_START_STRING,
785 block_end_string=BLOCK_END_STRING,
786 variable_start_string=VARIABLE_START_STRING,
787 variable_end_string=VARIABLE_END_STRING,
788 comment_start_string=COMMENT_START_STRING,
789 comment_end_string=COMMENT_END_STRING,
790 line_statement_prefix=LINE_STATEMENT_PREFIX,
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200791 line_comment_prefix=LINE_COMMENT_PREFIX,
Armin Ronacher4f5008f2008-05-23 23:36:07 +0200792 trim_blocks=TRIM_BLOCKS,
793 newline_sequence=NEWLINE_SEQUENCE,
Armin Ronacherb5124e62008-04-25 00:36:14 +0200794 extensions=(),
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200795 optimized=True,
796 undefined=Undefined,
Armin Ronacherd1342312008-04-28 12:20:12 +0200797 finalize=None,
798 autoescape=False):
Armin Ronacherb5124e62008-04-25 00:36:14 +0200799 env = get_spontaneous_environment(
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200800 block_start_string, block_end_string, variable_start_string,
801 variable_end_string, comment_start_string, comment_end_string,
Armin Ronacher59b6bd52009-03-30 21:00:16 +0200802 line_statement_prefix, line_comment_prefix, trim_blocks,
803 newline_sequence, frozenset(extensions), optimized, undefined,
804 finalize, autoescape, None, 0, False, None)
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200805 return env.from_string(source, template_class=cls)
Armin Ronacherba3757b2008-04-16 19:43:16 +0200806
Armin Ronacher7259c762008-04-30 13:03:59 +0200807 @classmethod
808 def from_code(cls, environment, code, globals, uptodate=None):
809 """Creates a template object from compiled code and the globals. This
810 is used by the loaders and environment to create a template object.
811 """
Armin Ronacher7259c762008-04-30 13:03:59 +0200812 namespace = {
Armin Ronacher64b08a02010-03-12 03:17:41 +0100813 'environment': environment,
814 '__file__': code.co_filename
Armin Ronacher7259c762008-04-30 13:03:59 +0200815 }
816 exec code in namespace
Armin Ronacher64b08a02010-03-12 03:17:41 +0100817 rv = cls._from_namespace(environment, namespace, globals)
818 rv._uptodate = uptodate
819 return rv
820
821 @classmethod
822 def from_module_dict(cls, environment, module_dict, globals):
823 """Creates a template object from a module. This is used by the
824 module loader to create a template object.
825
826 .. versionadded:: 2.4
827 """
828 return cls._from_namespace(environment, module_dict, globals)
829
830 @classmethod
831 def _from_namespace(cls, environment, namespace, globals):
832 t = object.__new__(cls)
Armin Ronacher7259c762008-04-30 13:03:59 +0200833 t.environment = environment
Armin Ronacher771c7502008-05-18 23:14:14 +0200834 t.globals = globals
Armin Ronacher7259c762008-04-30 13:03:59 +0200835 t.name = namespace['name']
Armin Ronacher64b08a02010-03-12 03:17:41 +0100836 t.filename = namespace['__file__']
Armin Ronacher7259c762008-04-30 13:03:59 +0200837 t.blocks = namespace['blocks']
Armin Ronacher771c7502008-05-18 23:14:14 +0200838
Georg Brandl3e497b72008-09-19 09:55:17 +0000839 # render function and module
Armin Ronacher5411ce72008-05-25 11:36:22 +0200840 t.root_render_func = namespace['root']
Armin Ronacher771c7502008-05-18 23:14:14 +0200841 t._module = None
Armin Ronacher7259c762008-04-30 13:03:59 +0200842
843 # debug and loader helpers
844 t._debug_info = namespace['debug_info']
Armin Ronacher64b08a02010-03-12 03:17:41 +0100845 t._uptodate = None
846
847 # store the reference
848 namespace['environment'] = environment
849 namespace['__jinja_template__'] = t
Armin Ronacher7259c762008-04-30 13:03:59 +0200850
851 return t
852
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200853 def render(self, *args, **kwargs):
Armin Ronacherd1342312008-04-28 12:20:12 +0200854 """This method accepts the same arguments as the `dict` constructor:
855 A dict, a dict subclass or some keyword arguments. If no arguments
856 are given the context will be empty. These two calls do the same::
857
858 template.render(knights='that say nih')
859 template.render({'knights': 'that say nih'})
860
861 This will return the rendered template as unicode string.
862 """
Armin Ronacher771c7502008-05-18 23:14:14 +0200863 vars = dict(*args, **kwargs)
Armin Ronacherf41d1392008-04-18 16:41:52 +0200864 try:
Armin Ronacher5411ce72008-05-25 11:36:22 +0200865 return concat(self.root_render_func(self.new_context(vars)))
Armin Ronacherf41d1392008-04-18 16:41:52 +0200866 except:
Armin Ronacherbd357722009-08-05 20:25:06 +0200867 exc_info = sys.exc_info()
868 return self.environment.handle_exception(exc_info, True)
Armin Ronacherbcb7c532008-04-11 16:30:34 +0200869
870 def stream(self, *args, **kwargs):
Armin Ronacherd1342312008-04-28 12:20:12 +0200871 """Works exactly like :meth:`generate` but returns a
872 :class:`TemplateStream`.
873 """
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200874 return TemplateStream(self.generate(*args, **kwargs))
Armin Ronacherfed44b52008-04-13 19:42:53 +0200875
876 def generate(self, *args, **kwargs):
Armin Ronacherd1342312008-04-28 12:20:12 +0200877 """For very large templates it can be useful to not render the whole
878 template at once but evaluate each statement after another and yield
879 piece for piece. This method basically does exactly that and returns
880 a generator that yields one item after another as unicode strings.
881
882 It accepts the same arguments as :meth:`render`.
883 """
Armin Ronacher771c7502008-05-18 23:14:14 +0200884 vars = dict(*args, **kwargs)
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200885 try:
Armin Ronacher5411ce72008-05-25 11:36:22 +0200886 for event in self.root_render_func(self.new_context(vars)):
Armin Ronacher771c7502008-05-18 23:14:14 +0200887 yield event
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200888 except:
Armin Ronacherbd357722009-08-05 20:25:06 +0200889 exc_info = sys.exc_info()
890 else:
891 return
892 yield self.environment.handle_exception(exc_info, True)
Armin Ronacher19cf9c22008-05-01 12:49:53 +0200893
Armin Ronacher673aa882008-10-04 18:06:57 +0200894 def new_context(self, vars=None, shared=False, locals=None):
Armin Ronacher5411ce72008-05-25 11:36:22 +0200895 """Create a new :class:`Context` for this template. The vars
Armin Ronacherc9705c22008-04-27 21:28:03 +0200896 provided will be passed to the template. Per default the globals
Armin Ronacher673aa882008-10-04 18:06:57 +0200897 are added to the context. If shared is set to `True` the data
898 is passed as it to the context without adding the globals.
899
900 `locals` can be a dict of local variables for internal usage.
Armin Ronacherc9705c22008-04-27 21:28:03 +0200901 """
Armin Ronacher74a0cd92009-02-19 15:56:53 +0100902 return new_context(self.environment, self.name, self.blocks,
903 vars, shared, self.globals, locals)
Armin Ronacherba3757b2008-04-16 19:43:16 +0200904
Armin Ronacher673aa882008-10-04 18:06:57 +0200905 def make_module(self, vars=None, shared=False, locals=None):
Armin Ronacher7ceced52008-05-03 10:15:31 +0200906 """This method works like the :attr:`module` attribute when called
Armin Ronacher0aa0f582009-03-18 01:01:36 +0100907 without arguments but it will evaluate the template on every call
908 rather than caching it. It's also possible to provide
Armin Ronacher7ceced52008-05-03 10:15:31 +0200909 a dict which is then used as context. The arguments are the same
Armin Ronacherf3c35c42008-05-23 23:18:14 +0200910 as for the :meth:`new_context` method.
Armin Ronacherea847c52008-05-02 20:04:32 +0200911 """
Armin Ronacher673aa882008-10-04 18:06:57 +0200912 return TemplateModule(self, self.new_context(vars, shared, locals))
Armin Ronacherea847c52008-05-02 20:04:32 +0200913
Armin Ronacherd84ec462008-04-29 13:43:16 +0200914 @property
915 def module(self):
916 """The template as module. This is used for imports in the
917 template runtime but is also useful if one wants to access
918 exported template variables from the Python layer:
Armin Ronacherd1342312008-04-28 12:20:12 +0200919
Armin Ronacherd84ec462008-04-29 13:43:16 +0200920 >>> t = Template('{% macro foo() %}42{% endmacro %}23')
921 >>> unicode(t.module)
922 u'23'
923 >>> t.module.foo()
Armin Ronacherd1342312008-04-28 12:20:12 +0200924 u'42'
Armin Ronacher6ce170c2008-04-25 12:32:36 +0200925 """
Armin Ronacher771c7502008-05-18 23:14:14 +0200926 if self._module is not None:
Armin Ronacherd84ec462008-04-29 13:43:16 +0200927 return self._module
Armin Ronacherea847c52008-05-02 20:04:32 +0200928 self._module = rv = self.make_module()
Armin Ronacherd84ec462008-04-29 13:43:16 +0200929 return rv
Armin Ronacher963f97d2008-04-25 11:44:59 +0200930
Armin Ronacherba3757b2008-04-16 19:43:16 +0200931 def get_corresponding_lineno(self, lineno):
932 """Return the source line number of a line number in the
933 generated bytecode as they are not in sync.
934 """
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200935 for template_line, code_line in reversed(self.debug_info):
Armin Ronacherba3757b2008-04-16 19:43:16 +0200936 if code_line <= lineno:
937 return template_line
938 return 1
Armin Ronacherc63243e2008-04-14 22:53:58 +0200939
Armin Ronacher9a822052008-04-17 18:44:07 +0200940 @property
Armin Ronacher814f6c22008-04-17 15:52:23 +0200941 def is_up_to_date(self):
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200942 """If this variable is `False` there is a newer version available."""
Armin Ronacher814f6c22008-04-17 15:52:23 +0200943 if self._uptodate is None:
944 return True
945 return self._uptodate()
946
Armin Ronacher203bfcb2008-04-24 21:54:44 +0200947 @property
948 def debug_info(self):
949 """The debug info mapping."""
950 return [tuple(map(int, x.split('='))) for x in
951 self._debug_info.split('&')]
952
Armin Ronacherc63243e2008-04-14 22:53:58 +0200953 def __repr__(self):
Armin Ronacher53042292008-04-26 18:30:19 +0200954 if self.name is None:
955 name = 'memory:%x' % id(self)
956 else:
957 name = repr(self.name)
958 return '<%s %s>' % (self.__class__.__name__, name)
Armin Ronacherc63243e2008-04-14 22:53:58 +0200959
960
Armin Ronacherd84ec462008-04-29 13:43:16 +0200961class TemplateModule(object):
962 """Represents an imported template. All the exported names of the
Armin Ronacher53042292008-04-26 18:30:19 +0200963 template are available as attributes on this object. Additionally
964 converting it into an unicode- or bytestrings renders the contents.
965 """
Armin Ronacher963f97d2008-04-25 11:44:59 +0200966
967 def __init__(self, template, context):
Armin Ronacher5411ce72008-05-25 11:36:22 +0200968 self._body_stream = list(template.root_render_func(context))
Armin Ronacher6ce170c2008-04-25 12:32:36 +0200969 self.__dict__.update(context.get_exported())
Armin Ronacher2feed1d2008-04-26 16:26:52 +0200970 self.__name__ = template.name
Armin Ronacher963f97d2008-04-25 11:44:59 +0200971
Armin Ronacher0faa8612010-02-09 15:04:51 +0100972 def __html__(self):
973 return Markup(concat(self._body_stream))
Armin Ronacher6ce170c2008-04-25 12:32:36 +0200974
975 def __str__(self):
Armin Ronacher2feed1d2008-04-26 16:26:52 +0200976 return unicode(self).encode('utf-8')
Armin Ronacher963f97d2008-04-25 11:44:59 +0200977
Armin Ronacheracbd4082010-02-10 00:07:43 +0100978 # unicode goes after __str__ because we configured 2to3 to rename
979 # __unicode__ to __str__. because the 2to3 tree is not designed to
980 # remove nodes from it, we leave the above __str__ around and let
981 # it override at runtime.
Armin Ronacher790b8a82010-02-10 00:05:46 +0100982 def __unicode__(self):
983 return concat(self._body_stream)
984
Armin Ronacher963f97d2008-04-25 11:44:59 +0200985 def __repr__(self):
Armin Ronacher53042292008-04-26 18:30:19 +0200986 if self.__name__ is None:
987 name = 'memory:%x' % id(self)
988 else:
Armin Ronacherdc02b642008-05-15 22:47:27 +0200989 name = repr(self.__name__)
Armin Ronacher53042292008-04-26 18:30:19 +0200990 return '<%s %s>' % (self.__class__.__name__, name)
Armin Ronacher963f97d2008-04-25 11:44:59 +0200991
992
Armin Ronacherba6e25a2008-11-02 15:58:14 +0100993class TemplateExpression(object):
994 """The :meth:`jinja2.Environment.compile_expression` method returns an
995 instance of this object. It encapsulates the expression-like access
996 to the template with an expression it wraps.
997 """
998
999 def __init__(self, template, undefined_to_none):
1000 self._template = template
1001 self._undefined_to_none = undefined_to_none
1002
1003 def __call__(self, *args, **kwargs):
1004 context = self._template.new_context(dict(*args, **kwargs))
1005 consume(self._template.root_render_func(context))
1006 rv = context.vars['result']
1007 if self._undefined_to_none and isinstance(rv, Undefined):
1008 rv = None
1009 return rv
1010
1011
Armin Ronacherc63243e2008-04-14 22:53:58 +02001012class TemplateStream(object):
Armin Ronacherd1342312008-04-28 12:20:12 +02001013 """A template stream works pretty much like an ordinary python generator
1014 but it can buffer multiple items to reduce the number of total iterations.
1015 Per default the output is unbuffered which means that for every unbuffered
1016 instruction in the template one unicode string is yielded.
1017
1018 If buffering is enabled with a buffer size of 5, five items are combined
1019 into a new unicode string. This is mainly useful if you are streaming
1020 big templates to a client via WSGI which flushes after each iteration.
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001021 """
Armin Ronacherc63243e2008-04-14 22:53:58 +02001022
1023 def __init__(self, gen):
1024 self._gen = gen
Armin Ronacher9cf95912008-05-24 19:54:43 +02001025 self.disable_buffering()
Armin Ronacherc63243e2008-04-14 22:53:58 +02001026
Armin Ronacher74b51062008-06-17 11:28:59 +02001027 def dump(self, fp, encoding=None, errors='strict'):
1028 """Dump the complete stream into a file or file-like object.
1029 Per default unicode strings are written, if you want to encode
1030 before writing specifiy an `encoding`.
1031
1032 Example usage::
1033
1034 Template('Hello {{ name }}!').stream(name='foo').dump('hello.html')
1035 """
1036 close = False
1037 if isinstance(fp, basestring):
1038 fp = file(fp, 'w')
1039 close = True
1040 try:
1041 if encoding is not None:
1042 iterable = (x.encode(encoding, errors) for x in self)
1043 else:
1044 iterable = self
1045 if hasattr(fp, 'writelines'):
1046 fp.writelines(iterable)
1047 else:
1048 for item in iterable:
1049 fp.write(item)
1050 finally:
1051 if close:
1052 fp.close()
1053
Armin Ronacherc63243e2008-04-14 22:53:58 +02001054 def disable_buffering(self):
1055 """Disable the output buffering."""
1056 self._next = self._gen.next
1057 self.buffered = False
1058
1059 def enable_buffering(self, size=5):
Armin Ronacherd1342312008-04-28 12:20:12 +02001060 """Enable buffering. Buffer `size` items before yielding them."""
Armin Ronacherc63243e2008-04-14 22:53:58 +02001061 if size <= 1:
1062 raise ValueError('buffer size too small')
Armin Ronacherc63243e2008-04-14 22:53:58 +02001063
Armin Ronacher5dfbfc12008-05-25 18:10:12 +02001064 def generator(next):
Armin Ronacherc63243e2008-04-14 22:53:58 +02001065 buf = []
1066 c_size = 0
1067 push = buf.append
Armin Ronacherc63243e2008-04-14 22:53:58 +02001068
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001069 while 1:
1070 try:
Armin Ronacherb5124e62008-04-25 00:36:14 +02001071 while c_size < size:
Armin Ronacher981cbf62008-05-13 09:12:27 +02001072 c = next()
1073 push(c)
1074 if c:
1075 c_size += 1
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001076 except StopIteration:
1077 if not c_size:
Armin Ronacherd84ec462008-04-29 13:43:16 +02001078 return
Armin Ronacherde6bf712008-04-26 01:44:14 +02001079 yield concat(buf)
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001080 del buf[:]
1081 c_size = 0
Armin Ronacherc63243e2008-04-14 22:53:58 +02001082
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001083 self.buffered = True
Armin Ronacher5dfbfc12008-05-25 18:10:12 +02001084 self._next = generator(self._gen.next).next
Armin Ronacherc63243e2008-04-14 22:53:58 +02001085
1086 def __iter__(self):
1087 return self
1088
1089 def next(self):
1090 return self._next()
Armin Ronacher203bfcb2008-04-24 21:54:44 +02001091
1092
1093# hook in default template class. if anyone reads this comment: ignore that
1094# it's possible to use custom templates ;-)
1095Environment.template_class = Template