Question

我试图在测试Flask应用程序中使用蓝图，但我遇到了一个奇怪的问题。这是我的代码的相关部分：

from functools import wraps
from flask import flash, redirect, render_template, \
request, session, url_for, Blueprint
from sqlalchemy.exc import IntegrityError
from datetime import datetime
from time import localtime, strftime

from .forms import AddAppointmentForm
from project import db
from project.models import Appointment

appointments_blueprint = Blueprint('appointments', __name__)

def login_required(test):
    @wraps(test)
    def wrap(*args, **kwargs):
        if 'logged_in' in session:
            return test(*args, **kwargs)
        else:
            flash('You need to login first.')
            return redirect(url_for('users.login'))
        return wrap

@appointments_blueprint.route('/appointments/')
@login_required
def appointments():
    # Get current date
    current_datetime = strftime("%Y-%m-%d %H:%M:%S", localtime())
    future_appointments = db.session.query(Appointment)\
        .filter(Appointment.due_date >= current_datetime)\
        .order_by(Appointment.due_date.asc())
    past_appointments = db.session.query(Appointment)\
        .filter(Appointment.due_date < current_datetime)\
        .order_by(Appointment.due_date.asc())
    return render_template('appointments.html',
        form = AddAppointmentForm(request.form),
        future_appointments=future_appointments,
        past_appointments=past_appointments)

现在，当我运行应用程序时，我收到此错误：

  File "/home/mosquito/git/flask-scheduler/project/appointments/views.py", line 72, in <module>
    @login_required
  File "/home/mosquito/python_envs/flask-scheduler/local/lib/python2.7/site-packages/flask/blueprints.py", line 160, in decorator
    endpoint = options.pop("endpoint", f.__name__)
AttributeError: 'NoneType' object has no attribute '__name__'

看看blueprints.py，我看到了：

def route(self, rule, **options):
    """Like :meth:`Flask.route` but for a blueprint.  The endpoint for the
    :func:`url_for` function is prefixed with the name of the blueprint.
    """
    def decorator(f):
        endpoint = options.pop("endpoint", f.__name__)
        self.add_url_rule(rule, endpoint, f, **options)
        return f
    return decorator

这失败了因为＆＃39; f＆＃39;是没有。如果我删除

，现在甚至更有趣

@login_required

装饰器，当它到达那个点时，f是一个函数，所以没问题。

另一方面，如果我删除

@appointments_blueprint.route()

装饰者，它也有效。所以看起来两个装饰器的组合导致f为None ....任何想法在这里发生了什么？

Answer 1

您需要取消return行：

def login_required(test):
    @wraps(test)
    def wrap(*args, **kwargs):
        if 'logged_in' in session:
            return test(*args, **kwargs)
        else:
            flash('You need to login first.')
            return redirect(url_for('users.login'))
    return wrap

你把它作为wrap函数本身的一部分，所以外部装饰器函数什么也没有返回。

删除@appointments_blueprint.route()根本就没有注册路线，因此您永远不会发现您将appointments设置为None。

烧瓶装饰器不能很好地协同工作

1 个答案: