如何在 Python 的类装饰器中传递参数？

Question

我在 Python 中使用此代码将装饰器应用于类的所有方法：

import inspect

def decallmethods(decorator, prefix='test_'):
  def dectheclass(cls):
    for name, m in inspect.getmembers(cls, inspect.ismethod):
      if name.startswith(prefix):
        setattr(cls, name, decorator(m))
    return cls
  return dectheclass


@decallmethods(login_testuser)
class TestCase(object):
    def setUp(self):
        pass

    def test_1(self):
        print "test_1()"

    def test_2(self):
        print "test_2()"

这很好用。但是，我想将一些kwargs 传递给我想申请的装饰器。

所以，我想要类似的东西：

@decallmethods(login_testuser, data="TEST") on class

它应用于方法的装饰器变为：

@login_testuser(data="TEST")

Answer 1

您构造它的方式意味着您的decorator 参数decallmethods 成为一个装饰器工厂（构造器），就像decallmethods 返回一个类装饰器一样。如果你明白了，你只需要改变调用方式

setattr(cls, name, decorator(m))

到

setattr(cls, name, decorator(data="TEST")(m))

对于静态实现。要使这种动态化，您还需要重新定义 decallmethods 以接受 data 参数

def decallmethods(decorator, prefix='test_', data='TEST'):
...
        setattr(cls, name, decorator(data=data)(m))

（另外，回想一下the time，当我向您指出装饰器只是接受一个函数返回一个新函数的函数时，在您的情况下，调用是test_method = login_testuser(data="TEST")(test_method)，当它以这种方式放置时，繁殖是微不足道的。）

当然，这假定函数 decorator 将能够使用 data 参数，并且如果您传入不是装饰器工厂的装饰器或不期望此参数的函数，则会发生异常。