我的代码中的很多模型都具有可以为None的关系。很多时候,这需要在访问数据之前检查None
:
{% if foo.bar %}
{{ foo.bar.baz }}
{% endif %}
如果我不添加该检查,则该页面会完全破坏UndefinedError。从Flask应用程序运行时,有什么办法可以让我在Jinja2中为UndefinedErrors默默地失败吗?
答案 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()