Jinja2中的测试器Test和过滤器非常相似,区别是测试器总是返回一个布尔值,它可以用来测试一个变量或者表达式,你需要使用”is”关键字来进行测试。测试器一般都是跟着if控制语句一起使用的。下面我们就来深入了解下这个测试器。

系列文章

  1. Flask中Jinja2模板引擎详解(一)-控制语句和表达式
  2. Flask中Jinja2模板引擎详解(二)-上下文环境
  3. Flask中Jinja2模板引擎详解(三)-过滤器
  4. Flask中Jinja2模板引擎详解(四)-测试器
  5. Flask中Jinja2模板引擎详解(五)-全局函数
  6. Flask中Jinja2模板引擎详解(六)-块和宏
  7. Flask中Jinja2模板引擎详解(七)-本地化
  8. Flask中Jinja2模板引擎详解(八)-自定义扩展

测试器使用

再次取回第一篇开篇的例子,我们在模板中对变量name作如下判断:

{% if name is lower %}
  <h2>"{{ name }}" are all lower case.</h2>
{% endif %}

name变量中的字母都是小写时,这段文字就会显示。这就是测试器,在if语句中,变量或表达式的后面加上is关键字,再加上测试器名称,就可以对该变量或表达式作测试,并根据其测试结果的真或假,来决定是否进入if语句块。测试器也可以有参数,用括号括起。当其只有一个参数时,可以省去括号。

{% if 6 is divisibleby 3 %}
  <h2>"divisibleby" test pass</h2>
{% endif %}

上例中,测试器divisibleby可以判断其所接收的变量是否可以被其参数整除。因为它只有一个参数,我们就可以用空格来分隔测试器和其参数。上面的调用同divisibleby(3)效果一致。测试器也可以配合not关键字一起使用:

{% if 6 is not divisibleby(4) %}
  <h2>"not divisibleby" test pass</h2>
{% endif %}

显然测试器本质上也是一个函数,它的第一个参数就是待测试的变量,在模板中使用时可以省略去。如果它有第二个参数,模板中就必须传进去。测试器函数返回的必须是一个布尔值,这样才可以用来给if语句作判断。

内置测试器 Builtin Tests

同过滤器一样,Jinja2模板引擎提供了丰富的内置测试器。这里介绍几个常用的。

{# 检查变量是否被定义,也可以用undefined检查是否未被定义 #}
{% if name is defined %}
    <p>Name is: {{ name }}</p>
{% endif %}

{# 检查是否所有字符都是大写 #}
{% if name is upper %}
  <h2>"{{ name }}" are all upper case.</h2>
{% endif %}

{# 检查变量是否为空 #}
{% if name is none %}
  <h2>Variable is none.</h2>
{% endif %}

{# 检查变量是否为字符串,也可以用number检查是否为数值 #}
{% if name is string %}
  <h2>{{ name }} is a string.</h2>
{% endif %}

{# 检查数值是否是偶数,也可以用odd检查是否为奇数 #}
{% if 2 is even %}
  <h2>Variable is an even number.</h2>
{% endif %}

{# 检查变量是否可被迭代循环,也可以用sequence检查是否是序列 #}
{% if [1,2,3] is iterable %}
  <h2>Variable is iterable.</h2>
{% endif %}

{# 检查变量是否是字典 #}
{% if {'name':'test'} is mapping %}
  <h2>Variable is dict.</h2>
{% endif %}

更全的内置测试器介绍可以从Jinja2的官方文档中找到。

自定义测试器

如果内置测试器不满足需求,我们就来自己写一个。写法很类似于过滤器,先在Flask应用代码中定义测试器函数,然后通过add_template_test将其添加为模板测试器:

import re
def has_number(str):
    return re.match(r'.*\d+', str)
app.add_template_test(has_number,'contain_number')

我们定义了一个has_number函数,用正则来判断输入参数是否包含数字。然后调用app.add_template_test方法,第一个参数是测试器函数,第二个是测试器名称。之后,我们就可以在模板中使用contain_number测试器了:

{% if name is contain_number %}
  <h2>"{{ name }}" contains number.</h2>
{% endif %}

同过滤器一样,Flask提供了添加测试器的装饰器template_test。下面的代码就添加了一个判断字符串是否以某一子串结尾的测试器。装饰器的参数定义了该测试器的名称end_with

@app.template_test('end_with')
def end_with(str, suffix):
    return str.lower().endswith(suffix.lower())

我们在模板中可以这样使用它:

{% if name is end_with "me" %}
  <h2>"{{ name }}" ends with "me".</h2>
{% endif %}

Flask添加测试器的方法是封装了对Jinja2环境变量的操作。上述添加end_with测试器的方法,等同于下面的代码。

app.jinja_env.tests['end_with'] = end_with

我们在Flask应用中,不建议直接访问Jinja2的环境变量。如果离开Flask环境直接使用Jinja2的话,就可以通过jinja2.Environment来获取环境变量,并添加测试器。

本文中的示例代码可以在这里下载