Python 装饰器和类

Question

我正在使用 Flask 和 Flask-Login 开发 Web 服务器。它一直在工作，直到我需要将它移到课堂上。由于装饰器而损坏：在我有代码之前

login_manager = LoginManager(app)
...
@login_required
def control_panel():
    return render_template(...)

但现在变成了：

class webserver:
    def __init__(self):
        self.login_manager = LoginManager(app)

    ...
    @login_required
    def control_panel(self):
        return self.render_template(...)

问题是我真的需要写，

@self.login_required

但它不是有效的语法。我该如何克服这个问题？

谢谢！

Answer 1

我不确定你的 LoginManager 是如何创建装饰器的，但你可以让它像这样工作：

class LoginManager:
    def __init__(self, app):
        pass

    @staticmethod
    def login_required(func):
        def inner(*args, **kwargs):
            print "login_required: before"
            func(*args, **kwargs)
            print "login_required: after"
        return inner


class Webserver:
    def __init__(self):
        self.login_manager = LoginManager(app)

    @LoginManager.login_required
    def control_panel(self):
        print "control_panel: called"


app = {}
ws = Webserver()
ws.control_panel()

输出：

login_required: before
control_panel: called
login_required: after