piecrust2: piecrust/templating/jinja/extensions.py annotate

author	Ludovic Chabant <ludovic@chabant.com>
date	Sat, 29 Apr 2017 21:42:22 -0700
parents
children	448710d84121

rev	line source
851 2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	1 from jinja2.ext import Extension, Markup
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	2 from jinja2.lexer import Token, describe_token
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	3 from jinja2.nodes import CallBlock, Const
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 from compressinja.html import HtmlCompressor, StreamProcessContext
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	5 from pygments import highlight
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	6 from pygments.formatters import HtmlFormatter
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 from pygments.lexers import get_lexer_by_name, guess_lexer
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 from piecrust.rendering import format_text
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	10
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	11 class PieCrustFormatExtension(Extension):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	12 tags = set(['pcformat'])
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	13
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	14 def __init__(self, environment):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	15 super(PieCrustFormatExtension, self).__init__(environment)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	16
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17 def parse(self, parser):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	18 lineno = next(parser.stream).lineno
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19 args = [parser.parse_expression()]
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20 body = parser.parse_statements(['name:endpcformat'], drop_needle=True)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 return CallBlock(self.call_method('_format', args),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22 [], [], body).set_lineno(lineno)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24 def _format(self, format_name, caller=None):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	25 body = caller()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	26 text = format_text(self.environment.app,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	27 format_name,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	28 Markup(body.rstrip()).unescape(),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	29 exact_format=True)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	30 return text
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	31
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	32
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	33 class PieCrustHighlightExtension(Extension):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	34 tags = set(['highlight', 'geshi'])
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	35
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 def __init__(self, environment):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	37 super(PieCrustHighlightExtension, self).__init__(environment)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	38
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	39 def parse(self, parser):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	40 lineno = next(parser.stream).lineno
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	41
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	42 # Extract the language name.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	43 args = [parser.parse_expression()]
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	44
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	45 # Extract optional arguments.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	46 kwarg_names = {'line_numbers': 0, 'use_classes': 0, 'class': 1,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	47 'id': 1}
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	48 kwargs = {}
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	49 while not parser.stream.current.test('block_end'):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50 name = parser.stream.expect('name')
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	51 if name.value not in kwarg_names:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52 raise Exception("'%s' is not a valid argument for the code "
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	53 "highlighting tag." % name.value)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54 if kwarg_names[name.value] == 0:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	55 kwargs[name.value] = Const(True)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	56 elif parser.stream.skip_if('assign'):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	57 kwargs[name.value] = parser.parse_expression()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	58
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	59 # body of the block
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	60 body = parser.parse_statements(['name:endhighlight', 'name:endgeshi'],
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	61 drop_needle=True)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	62
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	63 return CallBlock(self.call_method('_highlight', args, kwargs),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	64 [], [], body).set_lineno(lineno)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	65
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	66 def _highlight(self, lang, line_numbers=False, use_classes=False,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	67 css_class=None, css_id=None, caller=None):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	68 # Try to be mostly compatible with Jinja2-highlight's settings.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	69 body = caller()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	70
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	71 if lang is None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	72 lexer = guess_lexer(body)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	73 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	74 lexer = get_lexer_by_name(lang, stripall=False)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	75
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	76 if css_class is None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77 try:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	78 css_class = self.environment.jinja2_highlight_cssclass
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	79 except AttributeError:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	80 pass
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	81
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	82 if css_class is not None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	83 formatter = HtmlFormatter(cssclass=css_class,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	84 linenos=line_numbers)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	85 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	86 formatter = HtmlFormatter(linenos=line_numbers)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	87
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	88 code = highlight(Markup(body.rstrip()).unescape(), lexer, formatter)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	89 return code
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	90
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	91
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	92 def get_highlight_css(style_name='default', class_name='.highlight'):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	93 return HtmlFormatter(style=style_name).get_style_defs(class_name)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	94
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	95
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	96 class PieCrustCacheExtension(Extension):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	97 tags = set(['pccache', 'cache'])
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	98
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	99 def __init__(self, environment):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	100 super(PieCrustCacheExtension, self).__init__(environment)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	101 environment.extend(
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	102 piecrust_cache_prefix='',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 piecrust_cache={}
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	104 )
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	105
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	106 def parse(self, parser):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	107 # the first token is the token that started the tag. In our case
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108 # we only listen to ``'pccache'`` so this will be a name token with
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	109 # `pccache` as value. We get the line number so that we can give
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	110 # that line number to the nodes we create by hand.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	111 lineno = next(parser.stream).lineno
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	112
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	113 # now we parse a single expression that is used as cache key.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	114 args = [parser.parse_expression()]
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	115
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	116 # now we parse the body of the cache block up to `endpccache` and
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	117 # drop the needle (which would always be `endpccache` in that case)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	118 body = parser.parse_statements(['name:endpccache', 'name:endcache'],
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	119 drop_needle=True)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	120
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	121 # now return a `CallBlock` node that calls our _cache_support
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	122 # helper method on this extension.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	123 return CallBlock(self.call_method('_cache_support', args),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	124 [], [], body).set_lineno(lineno)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	125
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	126 def _cache_support(self, name, caller):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	127 key = self.environment.piecrust_cache_prefix + name
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	128
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	129 exc_stack = self.environment.app.env.exec_info_stack
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	130 render_ctx = exc_stack.current_page_info.render_ctx
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	131 rdr_pass = render_ctx.current_pass_info
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	132
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	133 # try to load the block from the cache
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	134 # if there is no fragment in the cache, render it and store
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	135 # it in the cache.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	136 pair = self.environment.piecrust_cache.get(key)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	137 if pair is not None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	138 rdr_pass.used_source_names.update(pair[1])
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139 return pair[0]
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	140
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	141 pair = self.environment.piecrust_cache.get(key)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	142 if pair is not None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	143 rdr_pass.used_source_names.update(pair[1])
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	144 return pair[0]
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	145
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146 prev_used = rdr_pass.used_source_names.copy()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	147 rv = caller()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	148 after_used = rdr_pass.used_source_names.copy()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	149 used_delta = after_used.difference(prev_used)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	150 self.environment.piecrust_cache[key] = (rv, used_delta)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	151 return rv
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	152
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	153
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154 class PieCrustSpacelessExtension(HtmlCompressor):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	155 """ A re-implementation of `SelectiveHtmlCompressor` so that we can
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	156 both use `strip` or `spaceless` in templates.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	157 """
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158 def filter_stream(self, stream):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	159 ctx = StreamProcessContext(stream)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	160 strip_depth = 0
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	161 while 1:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	162 if stream.current.type == 'block_begin':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	163 for tk in ['strip', 'spaceless']:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	164 change = self._processToken(ctx, stream, tk)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	165 if change != 0:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	166 strip_depth += change
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	167 if strip_depth < 0:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	168 ctx.fail('Unexpected tag end%s' % tk)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	169 break
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	170 if strip_depth > 0 and stream.current.type == 'data':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	171 ctx.token = stream.current
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	172 value = self.normalize(ctx)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	173 yield Token(stream.current.lineno, 'data', value)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	174 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	175 yield stream.current
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	176 next(stream)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	178 def _processToken(self, ctx, stream, test_token):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	179 change = 0
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	180 if (stream.look().test('name:%s' % test_token) or
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	181 stream.look().test('name:end%s' % test_token)):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	182 stream.skip()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183 if stream.current.value == test_token:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	184 change = 1
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	185 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	186 change = -1
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	187 stream.skip()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	188 if stream.current.type != 'block_end':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	189 ctx.fail('expected end of block, got %s' %
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	190 describe_token(stream.current))
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	191 stream.skip()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	192 return change

851

2c7e57d80bba optimize: Don't load Jinja unless we need to.