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

Декоратор, требующий авторизации

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