烧瓶从关闭重定向

Question

def check_login(func):
    """Check if user is logged in."""
    def decorator(*args, **kwargs):
        if not login_session_test():
            print ("Not logged in - redirect to /login")
            flash ("Well that was wrong. Chicken winner. No more dinner.")
            return redirect(url_for('login'))
        print ("Logged in, do what needs to be done.")
        return func(*args, **kwargs)
    return decorator

@check_login
@app.route("/sacred/secret/stuff", methods=['GET'])
def funfunfun():
    return "Super fun"

它永远不会重定向到/login，但会给出一些垃圾页面。

交换@/closure 订单产生：

AssertionError: View function mapping is overwriting an existing endpoint function: decorator

我还没有完全 Python 化。

Answer 1

您的装饰器顺序不正确，并且您没有将函数名称复制到包装函数。

使用这个顺序：

@app.route("/sacred/secret/stuff", methods=['GET'])
@check_login
def funfunfun():
    return "Super fun"

否则为视图注册未修饰的函数。

使用@functools.wraps() 将各种元数据从原始包装函数复制到替换它的包装器：

from functools import wraps

def check_login(func):
    """Check if user is logged in."""
    @wraps(func)
    def decorator(*args, **kwargs):
        if not login_session_test():
            print ("Not logged in - redirect to /login")
            flash ("Well that was wrong. Chicken winner. No more dinner.")
            return redirect(url_for('login'))
        print ("Logged in, do what needs to be done.")
        return func(*args, **kwargs)
    return decorator

路由需要一个端点名称，如果你没有明确指定，Flask 使用函数的名称（来自functionobj.__name__）。但是您的装饰器包装对象的名称为decorator，因此如果您多次使用装饰器，Flask 会抱怨它已经使用了该端点名称。

@functools.wraps() 复制 __name__ 属性，所以现在你的装饰器包装器也称为funfunfun，而另一个装饰路由函数也可以保留它的名字。