编写“代码” Jinja2 中的扩展

Question

我正在尝试为 Jinja2 编写一个扩展，以便我可以在模板中运行任意 python 代码（我有一个沙箱来安全地运行此代码）。我知道我可以使用 Mako 之类的东西，它允许嵌入 python 代码，但我更喜欢使用 Jinja2，因为我喜欢它的一般语法。

我打算在模板中执行的操作的一个示例：

{{ x }}

{% code %}
x = x + 1
{% endcode %}

{{ x }}

如果我使用 x = 1 渲染上面的模板，我希望输出是，

1
2

我根据文档上的示例编写了一个扩展。我可以使用 contextfunction 装饰器访问上下文，但它是不可变的。

在扩展程序的回调中，我想做这样的事情，

exec body in context

一些 Jinja2 大师可以帮助我吗？