如何从初始化参数中正确设置 Python 类属性

Question

作为一名 Python 程序员，我经常声明类似的类

class Foo:
  def __init__(self, attr1, attr2, attr3, attr4, attr5, attr6, attr7, attr8, attr9):
    self.attr1 = attr1
    self.attr2 = attr2
    ...
    self.attr9 = attr9

    # init your class

问题是我一遍又一遍地这样做，感觉效率很低。在 C++ 中，执行此操作的方法是

class Foo {
  public:
    Foo(int, int, int, int, int);
    
  private:
    int attr1, attr2, attr3, attr4, attr5;
    
};

Foo::Foo(int attr1, int attr2, int attr3, int attr4, int attr5) : 
    attr1(attr1), attr2(attr2), attr3(attr3), attr4(attr4), attr5(attr5) {
    // init your class
}

我觉得它的效率稍微高一些。我的问题是：是否有一种标准方法可以避免必须将每个参数的参数归因于该类？我可以想出一些方法来编程它，例如

def attribute_setter(*arg_names):
  def wrapper(func):
    def setter_init(self, *args, **kwargs):
      for i, name in enumerate(arg_names):
        if i < len(args):
          setattr(self, name, args[i])
        elif name in kwargs:
          setattr(self, name, kwargs[name])
      func(self, *args, **kwargs)
    return setter_init
  return wrapper

class Foo:
  @attribute_setter('attr1', 'attr2', 'attr3', 'attr4', 'attr5', 'attr6', 'attr7', 'attr8', 'attr9')
  def __init__(self, *args, **kwargs):
    # init your class
    pass

哪个产量

>>> foo = Foo(1, 2, 3, 4, 5, attr6=6, attr7=7, attr8=8, attr9=9)
>>> foo.attr1
1
>>> foo.attr9
9

通过更多的错误处理，这个装饰器可能会变得可以安全使用，但我关心的是代码的可读性并确保我没有违反任何良好的实践原则。

Answer 1

这正是该dataclasses模块的用途。你像这样使用它：

from dataclasses import dataclass


@dataclass
class Foo:
  attr1: str
  attr2: int
  attr3: bool
  # ...etc...

这为您提供了一个带有隐式__init__方法的类，该方法可以为您处理所有参数设置。您还可以获得诸如有用__repr__且__str__预定义的内容：

>>> myvar = Foo(attr1="somevalue", attr2=42, attr3=False)
>>> myvar
Foo(attr1='somevalue', attr2=42, attr3=False)
>>> print(myvar)
Foo(attr1='somevalue', attr2=42, attr3=False)