如何使用FlaskLogin和自定义Python装饰器来分配用户权限？

Question

我不知道如何让Python装饰程序在Flask视图上工作

这是我的装饰师：

from functools import wraps

def admin_only(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print("It did something")
        if current_user.is_anonymous:
            flash("Please log in to use this site.", "info")
            return redirect(url_for('login'))
        if not current_user.admin:
            abort(403)
        return func(*args, **kwargs)
    return wrapper


@bp.errorhandler(403)
def forbidden_403(exception):
    return 'Admin access only', 403

以下是我的观点：

@bp.route('/add_info', methods=['GET', 'POST'])
@admin_only
def add_info():
    add_info_form = AddInfoForm()
    if add_info_form.validate_on_submit():
        print("Code here")
    else:
        for error in add_info_form.errors:
            flash(error, "info")
    context = {
        'add_info_form': add_info_form,
    }
    return render_template('admin/add_info.html', **context)

装饰器根本不起作用——视图可以工作，但允许任何用户通过。在view函数运行之前应该实现的所有代码均未运行。我尝试过各种不同的格式化技巧，但都不奏效

我使用了这个答案中的代码-Python decorator with Flask：

def decorator(take_a_function):
    def wrapper1(take_a_function):
        def wrapper2(*takes_multiple_arguments):
           # do stuff
           return take_a_function(*takes_multiple_arguments)

        return wrapper2
    return wrapper1

似乎什么都不管用。我错过了什么

提前谢谢

Answer 1

好吧，我不知道为什么会这样，但我要说：

我使用的蓝图文件结构如下：

/app
   /user
      /__init__.py
      /forms.py
      /templates/user
         /user.html
   /lists
      /__init__.py
      /forms.py
      /templates/lists
         /lists.html
   /admin
      /__init__.py
      /forms.py
      /templates/admin
         /add_user.html
         /add_info.html
         /all_lists.html
   __init__.py
   forms.py
   models.py
   /templates
      index.html

我之前在/admin/__init__.py中定义了decorator函数。当我简单地将其移动到app/__init__.py文件时，该函数工作正常

我很高兴，但我也希望有人能提供一些关于为什么会这样的细节。如果没有，我会接受我的回答。也许这与Flask在__init__.py文件中运行函数的顺序有关