视图装饰器
Python 有一个非常有趣的功能：函数装饰器。这个功能可以使网络应用干净整洁。
Flask 中的每个视图都是一个装饰器，它可以被注入额外的功能。你可能已经用过了
route() 装饰器。但是，你有可能需要使用你自己的装饰器。
假设有一个视图，只有已经登录的用户才能使用。如果用户访问时没有登录，则会被
重定向到登录页面。这种情况下就是使用装饰器的绝佳机会。
检查登录装饰器¶
让我们来实现这个装饰器。装饰器是一个包装并替换另一个函数的函数。既然源函数
已经被替代，就需要记住：要复制源函数的信息到新函数中。可以用
functools.wraps() 处理这个事情。
下面是检查登录装饰器的例子。假设登录页面为 'login' ，当前用户被储存在
g.user 中，如果还没有登录，其值为 None:
from functools import wraps
from flask import g, request, redirect, url_for
def login_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if g.user is None:
return redirect(url_for('login', next=request.url))
return f(*args, **kwargs)
return decorated_function
为了使用这个装饰器呢，需要把这个装饰器放在最靠近函数的地方。当使用更进一步
的装饰器时，请记住要把 route() 装饰器放在最外面:
@app.route('/secret_page')
@login_required
def secret_page():
pass
Note
登录页面在一个``GET`` 请求之后 next 值会存在于 request.args
之中。当从登录表单发送 POST 请求时必须一起传递它。可以使用一个
隐藏标记来做到这点，然后当用户登录时，从 request.form 获取它。
<input type="hidden" value="{{ request.args.get('next', '') }}"/>
缓存装饰器¶
假设有一个视图函数需要消耗昂贵的计算成本，因此你需要在一定时间内缓存这个视 图的计算结果。这种情况下装饰器是一个好的选择。我们假设你像 缓存 方案中一样设置了缓存。
下面是一个示例缓存函数。它根据一个特定的前缀（实际上是一个格式字符串）和请 求的当前路径生成缓存键。注意，我们先使用了一个函数来创建装饰器，这个装饰器 用于装饰函数。听起来拗口吧，确实有一点复杂，但是下面的示例代码还是很容易读 懂的。
被装饰代码按如下步骤工作
基于基础路径获得当前请求的唯一缓存键。
从缓存中获取键值。如果获取成功则返回获取到的值。
否则调用原来的函数，并把返回值存放在缓存中，直至过期（缺省值为五分钟）。
代码:
from functools import wraps
from flask import request
def cached(timeout=5 * 60, key='view/%s'):
def decorator(f):
@wraps(f)
def decorated_function(*args, **kwargs):
cache_key = key % request.path
rv = cache.get(cache_key)
if rv is not None:
return rv
rv = f(*args, **kwargs)
cache.set(cache_key, rv, timeout=timeout)
return rv
return decorated_function
return decorator
注意，以上代码假设存在一个可用的实例化的 cache 对象，更多信息参见 缓存 方案。
模板装饰器¶
不久前， TurboGear 的人发明了模板装饰器这个通用模式。其工作原理是返回一个 字典，这个字典包含从视图传递给模板的值，模板自动被渲染。以下三个例子的功能 是相同的:
@app.route('/')
def index():
return render_template('index.html', value=42)
@app.route('/')
@templated('index.html')
def index():
return dict(value=42)
@app.route('/')
@templated()
def index():
return dict(value=42)
正如你所见，如果没有提供模板名称，那么就会使用 URL 映射的端点（把点转换为
斜杠）加上 '.html' 。如果提供了，那么就会使用所提供的模板名称。当装饰
器函数返回时，返回的字典就被传送到模板渲染函数。如果返回的是 None ，就
会使用空字典。如果返回的不是字典，那么就会直接传递原封不动的返回值。这样就
可以仍然使用重定向函数或返回简单的字符串。
以下是装饰器的代码:
from functools import wraps
from flask import request, render_template
def templated(template=None):
def decorator(f):
@wraps(f)
def decorated_function(*args, **kwargs):
template_name = template
if template_name is None:
template_name = request.endpoint \
.replace('.', '/') + '.html'
ctx = f(*args, **kwargs)
if ctx is None:
ctx = {}
elif not isinstance(ctx, dict):
return ctx
return render_template(template_name, **ctx)
return decorated_function
return decorator
