jinja2如何扩展列表

from jinja2 import Environment, BaseLoader, Template

class ListExtension(jinja2.ext.Extension):
    tags = set(['list'])

    def __init__(self, environment):
        super(ListExtension, self).__init__(environment)

    def parse(self, parser):
        # 解析模板中的自定义标签
        token = parser.stream.next()
        lineno = token.lineno

        # 获取列表名称
        list_name = parser.parse_expression()

        # 返回一个新的节点，表示对列表的操作
        return jinja2.nodes.CallBlock(
            self.call_method('_render_list', [list_name]),
            [], [], []
        ).set_lineno(lineno)

    def _render_list(self, list_name, caller):
        # 在这里执行对列表的操作，并返回结果
        # 可以使用caller()来渲染块内容
        # 例如：result = [item.upper() for item in list_name]
        result = [item.upper() for item in list_name]
        return caller(result)

# 创建Jinja2环境并加载自定义扩展
env = Environment(loader=BaseLoader())
env.add_extension(ListExtension)

# 渲染模板
template = env.from_string("""
{% list my_list %}
    {{ item }}
{% endlist %}
""")
output = template.render(my_list=['apple', 'banana', 'orange'])

print(output)