diff --git a/CHANGES b/CHANGES index 9550fa46b..eea91dac7 100644 --- a/CHANGES +++ b/CHANGES @@ -10,6 +10,7 @@ Version 2.10 derived context. - Added an `in` test that works like the in operator. This can be used in combination with `reject` and `select`. +- Added filter support to the block ``set`` tag. Version 2.9.6 ------------- diff --git a/docs/templates.rst b/docs/templates.rst index e9e363911..d669a4b20 100644 --- a/docs/templates.rst +++ b/docs/templates.rst @@ -896,6 +896,17 @@ Example:: The `navigation` variable then contains the navigation HTML source. +.. versionchanged:: 2.10 + +Starting with Jinja 2.10, the block assignment supports filters. + +Example:: + + {% set reply | wordwrap %} + You wrote: + {{ message }} + {% endset %} + .. _extends: diff --git a/jinja2/compiler.py b/jinja2/compiler.py index 0a3f7962a..69d1caa25 100644 --- a/jinja2/compiler.py +++ b/jinja2/compiler.py @@ -1378,7 +1378,12 @@ def visit_AssignBlock(self, node, frame): self.newline(node) self.visit(node.target, frame) self.write(' = (Markup if context.eval_ctx.autoescape ' - 'else identity)(concat(%s))' % block_frame.buffer) + 'else identity)(') + if node.filter is not None: + self.visit_Filter(node.filter, block_frame) + else: + self.write('concat(%s)' % block_frame.buffer) + self.write(')') self.pop_assign_tracking(frame) self.leave_frame(block_frame) diff --git a/jinja2/nodes.py b/jinja2/nodes.py index 9903d43e1..458d95c4e 100644 --- a/jinja2/nodes.py +++ b/jinja2/nodes.py @@ -387,7 +387,7 @@ class Assign(Stmt): class AssignBlock(Stmt): """Assigns a block to a target.""" - fields = ('target', 'body') + fields = ('target', 'filter', 'body') class Expr(Node): diff --git a/jinja2/parser.py b/jinja2/parser.py index 0bf74c945..b018415ca 100644 --- a/jinja2/parser.py +++ b/jinja2/parser.py @@ -180,9 +180,10 @@ def parse_set(self): if self.stream.skip_if('assign'): expr = self.parse_tuple() return nodes.Assign(target, expr, lineno=lineno) + filter_node = self.parse_filter(None) body = self.parse_statements(('name:endset',), drop_needle=True) - return nodes.AssignBlock(target, body, lineno=lineno) + return nodes.AssignBlock(target, filter_node, body, lineno=lineno) def parse_for(self): """Parse a for loop.""" diff --git a/tests/test_core_tags.py b/tests/test_core_tags.py index f48d8b448..b0e846a0d 100644 --- a/tests/test_core_tags.py +++ b/tests/test_core_tags.py @@ -364,6 +364,19 @@ def test_block_escaping(self): '{% endset %}foo: {{ foo }}') assert tmpl.render(test='') == 'foo: <unsafe>' + def test_block_escaping_filtered(self): + env = Environment(autoescape=True) + tmpl = env.from_string('{% set foo | trim %}{{ test }} ' + '{% endset %}foo: {{ foo }}') + assert tmpl.render(test='') == 'foo: <unsafe>' + + def test_block_filtered(self, env_trim): + tmpl = env_trim.from_string( + '{% set foo | trim | length | string %} 42 {% endset %}' + '{{ foo }}') + assert tmpl.render() == '2' + assert tmpl.module.foo == u'2' + @pytest.mark.core_tags @pytest.mark.with_