请求前的烧瓶-为特定路由添加异常

Question

在下面的before_request（）函数中，如果用户尚未登录，我想将其重定向到登录。是否有一个特殊的变量可以提供当前URL，该URL将像下面的示例那样工作？

@app.before_request
def before_request():
    # the variable current_url does not exist
    # but i want something that works like it
    if (not 'logged_in' in session) and (current_url != '/login'):
        return redirect(url_for('login'))

我需要检查当前URL是否为/login，因为如果我不这样做，服务器将进入无限循环。

Answer 1

下面是使用flask登录的公认答案的实现：

@app.before_request
def require_authorization():
    from flask import request
    from flask.ext.login import current_user

    if not (current_user.is_authenticated or request.endpoint == 'login'):
        return login_manager.unauthorized()

Answer 2

你可以用一个装饰师。下面的示例显示了如何在特定请求之前检查API密钥：

from functools import wraps

def require_api_key(api_method):
    @wraps(api_method)

    def check_api_key(*args, **kwargs):
        apikey = request.headers.get('ApiKey')
        if apikey and apikey == SECRET_KEY:
            return api_method(*args, **kwargs)
        else:
            abort(401)

    return check_api_key

您可以使用它：

@require_api_key

Answer 3

您可以检查请求对象上的几个属性，并在此处记录，request.path可能是您想要的。不过，我可以建议request.endpoint吗，这样如果您决定将视图路由到另一个url或多个url，您就会被覆盖

@app.before_request
def before_request():
    if 'logged_in' not in session and request.endpoint != 'login':
        return redirect(url_for('login'))