python - 通过装饰器注入(inject)类属性

Question

我有一大堆 Django 测试用例，它们使用 setUp 方法来初始化在众多测试中使用的一些属性，它们的构造方式和相互依赖的方式是我想要从测试用例中移出的逻辑，重用

def setUp(self):
    self.property_1 = ##some logic
    ...

我想将它们重写为一些方便的包装器，可以通过简单的继承或装饰器注入(inject)到类中，例如

@with_property_1(x=1, y=2)
def setUp(self):
    ...


def with_property_1(**model_kwargs):
    def wrapper(f):
        def wrapped(*args, **kwargs):
            self = args[0]
            self.property_1 = ## logic
            f(*args, **kwargs)
        return wrapped
    return wrapper

但问题是 PyCharm 无法识别这些实例属性的存在，因为 TestCase 类内部没有任何内容设置它们。是否有另一种方法可以很好地实现这一目标，或者有一种方法可以诱使 PyCharm 认识到这些属性在装饰器存在的情况下是合法的？

Answer 1

免责声明:未经测试。

这里的问题是 Python 不能神奇地将 self 放入装饰器的上下文中(这在任何 IDE 中都不起作用)。您可能忘记的是，self 是调用每个类方法时传递给它的参数之一。因此，它存在于您的 *args 中，您可以对其进行操作。

这是我的试用代码:

def with_property_1(**model_kwargs):
  def wrapper(f):
    def wrapped(*args, **kwargs):
      for key, value in model_kwargs.items():
        setattr(args[0], key, value)
      f(*args, **kwargs)
    return wrapped
  return wrapper

说明:

1. 迭代 **model_kwargs 中的每个键/值。

2. 使用 setattr 使用提供的 kwargs 修改 args[0]，它应该是 self 变量。

3. 使用更新后的 self 变量正常调用该函数。