Flask基本HTTP身份验证使用登录页面

4
我正在构建一个测试应用程序,并使用此处提供的指南(http://flask.pocoo.org/snippets/8/)设置简单的身份验证。在需要身份验证的页面上,我会得到一个弹出窗口,显示“要求授权”。 相反,我想重定向到一个登录页面,用户可以在表单中输入他们的用户名/密码。
以下是我的代码(与链接中的片段相同):
from functools import wraps
from flask import request, Response


def check_auth(username, password):
    """This function is called to check if a username /
    password combination is valid.
    """
    return username == 'admin' and password == 'secret'

def authenticate():
    """Sends a 401 response that enables basic auth"""
    return Response(
    'Could not verify your access level for that URL.\n'
    'You have to login with proper credentials', 401,
    {'WWW-Authenticate': 'Basic realm="Login Required"'})

def requires_auth(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        auth = request.authorization
        if not auth or not check_auth(auth.username, auth.password):
            return authenticate()
        return f(*args, **kwargs)
    return decorated

看起来我可以使用Flask-Auth,但实际上我只需要上面提供的功能。

谢谢, Ryan

1个回答

4

从这里的文档中:http://flask.pocoo.org/docs/0.10/patterns/viewdecorators/。有一个示例装饰器可以实现这一点:

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

现在您需要做的是返回重定向而不是调用authenticate()方法。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接