Декораторы представлений

Декоратор обязательной авторизации

Итак, давайте реализуем такой декоратор. Декоратор — это функция, которая оборачивает и заменяет другую функцию. Поскольку исходная функция заменяется, вам необходимо запомнить, чтобы скопировать информацию из исходной функции в новую функцию. Используйте 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

<input type="hidden" value="{{ request.args.get('next', '') }}"/>

Декоратор кэширования

Представьте, что у вас есть функция представления, которая выполняет дорогостоящее вычисление, и поэтому вы хотите кэшировать полученные результаты на определённый период времени. Декоратор будет полезен для этого. Мы предполагаем, что вы настроили кэш, как указано в Кэшировании.

Вот пример функции кэширования. Она генерирует ключ кэша из определенного префикса (фактически, строка форматирования) и текущего пути запроса. Обратите внимание, что мы используем функцию, которая сначала создает декоратор, а затем декорирует функцию. Звучит ужасно? К сожалению, это немного сложнее, но код все равно должен быть легко читаемым.

Декорированная функция будет работать следующим образом:

1. Получить уникальный ключ кэша для текущего запроса на основе текущего пути.
2. Получить значение для этого ключа из кэша. Если кэш вернул что-то, мы вернём это значение.
3. В противном случае вызывается исходная функция, а возвращаемое значение сохраняется в кэше на время ожидания, указанное по умолчанию (5 минут).

Вот код:

from functools import wraps
from flask import request

def cached(timeout=5 * 60, key='view/{}'):
    def decorator(f):
        @wraps(f)
        def decorated_function(*args, **kwargs):
            cache_key = key.format(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 доступен, см. Кэширование.

Декоратор шаблонизации

Распространенный шаблон, придуманный авторами TurboGears некоторое время назад, — это декоратор шаблонизации. Идея этого декоратора в том, что вы возвращаете словарь со значениями, переданными в шаблон из функции представления, а шаблон автоматически рендерится. Благодаря этому следующие три примера делают ровно то же самое:

@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 = f"{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

Декоратор конечной точки

Когда вы хотите использовать систему маршрутизации werkzeug для большей гибкости, вам нужно сопоставить конечную точку, как определено в Rule, с функцией представления. Это возможно с помощью этого декоратора. Например:

from flask import Flask
from werkzeug.routing import Rule

app = Flask(__name__)
app.url_map.add(Rule('/', endpoint='index'))

@app.endpoint('index')
def my_index():
    return "Hello world"