有没有办法将变量传递给装饰器？

Question

我已经基于OpenID提供程序登录系统，并且想要创建类似@login_required的装饰器。 但是我有一个问题-我需要检查每个对象的权限。 我将id传递给我的视图，所以我想知道有什么方法可以将其传递给装饰器，否则这是不可能的，我需要在视图中检查用户权限。

Answer 1

def outer(var1, var2):
    def inner(func)
        func.foo = var1
        func.bar = var2
        return func
    return inner

@outer(myvar1, myvar2)
def myfunc():
    # whatever
    pass

虽然您不能将变量直接传递给装饰函数的函数，但是可以创建另一个函数 （ outer ），该函数在调用时返回装饰函数 （ inner ）。 您传递outer函数变量，它们对于inner函数可用。 这称为关闭。

Answer 2

根据何时需要使用id进行操作，您可能需要两层包装，在其中将print(permission_id)替换为调用方法之前需要执行的操作：

def login_required(permission_id):
    def decorator(func):
        def method(self, *args, **kw):
            print(permission_id)
            return func(self, *args, **kw)
        return method
    return decorator

class foo(object):

    @login_required('foo')
    def bar(self):
        print('bar')

foo = foo()
print('instantiated')
foo.bar()

输出：

instantiated
foo
bar