如何让Jinja2将所有UndefinedError异常转换为空字符串?

时间:2013-02-03 13:29:57

标签: python flask jinja2

我的代码中的很多模型都具有可以为None的关系。很多时候,这需要在访问数据之前检查None

{% if foo.bar %}
   {{ foo.bar.baz }}
{% endif %}

如果我不添加该检查,则该页面会完全破坏UndefinedError。从Flask应用程序运行时,有什么办法可以让我在Jinja2中为UndefinedErrors默默地失败吗?

1 个答案:

答案 0 :(得分:8)

只需创建jinja2.Undefined的子类,该子类在属性访问时返回自身,并通过覆盖create_jinja_environment方法将其设置为您环境的未定义类型:

from flask import Flask
from jinja2 import Undefined
from werkzeug.datastructures import ImmutableDict

class MalleableUndefined(Undefined):
    def __getitem__(self, key):
        return self

    def __getattr__(self, key):
        return self


class CustomFlask(Flask):
    def create_jinja_environment(self):
        self.jinja_options = ImmutableDict(undefined=MalleableUndefined, **self.jinja_options)
        return super(CustomFlask, self).create_jinja_environment()