用枚举定义 python 类

Question

最近我定义了一个 Python 类，如下所示。

from datetime import datetime, date, time
import enums


class ExampleClass:

    defaults = (-1, "", "", datetime.today(), "", -1, [], "", -1, "", "", [], "")

    def __init__(self, **kwargs):
        count = 0
        for ex in enums.ExampleEnums:
            setattr(self, ex.name, kwargs.get(ex.value, ExampleClass.defaults[count]))
            count += 1

    def __str__(self):
        return_string = "Example Object with "
        count = 0
        for val in enums.ExampleEnums:
            if (getattr(self, val.name) != ExampleClass.defaults[count]):
                return_string += str("%s: %s, " % (val.name, getattr(self, val.name)))
            count += 1
        return return_string[:-2]

    def __repr__(self):
        return_string = ""
        count = 0
        for val in enums.ExampleEnums:
            if (getattr(self, val.name) != ExampleClass.defaults[count]):
                return_string += str("%s=%s, " % (val.value, getattr(self, val.name)))
            count += 1
        return return_string[:-2]

    def __eq__(self, other):
        for val in enums.ExampleEnums:
            if (getattr(self, val.name) != getattr(other, val.name)):
                return False
        return True

    def __ne__(self, other):
        for val in enums.ExampleEnums:
            if (getattr(self, val.name) == getattr(other, val.name)):
                return False
        return True

无论如何，我想知道：这是为数据类编写类定义的好方法吗？有什么方法可以改进吗？我不需要任何代码，只需要概括性就可以了，因为我只是将其发布为了解如何提高自己在 Python 中的编码能力的一种方式。

谢谢

Answer 1

你可以在函数__ne__中调用self._eq__(other)。

Answer 2

你多次使用这个模式（这里显示的是__init__，它也适用于__str__和__repr__）：

count = 0
for ex in enums.ExampleEnums:
    setattr(self, ex.name, kwargs.get(ex.value, ExampleClass.defaults[count]))
    count += 1

最好直接遍历ExampleClass.defaults 中的项目，而不是手动计算索引。这可以使用zip 来实现：

for ex, default in zip(enums.ExampleEnums, ExampleClass.defaults):
    setattr(self, ex.name, kwargs.get(ex.value, default))

---

__eq__ 方法可以使用all 进行简化：

def __eq__(self, other):
    return all(getattr(self, val.name) == getattr(other, val.name)
               for val in enums.ExampleEnums)

然后，正如其他人已经说过的，您可以将__ne__ 表达为__eq__，甚至使用== 运算符：

def __ne__(self, other):
    return not self == other

Answer 3

编写数据类的最佳方式因用例而异。但是根据您所提供的内容，您不应该重复代码。一旦你定义了__eq__ 操作符，你应该在别处使用它。 （如果您对__eq__ 的定义发生变化怎么办？）另外，您不必定义太阳下的每一种神奇方法……只要定义对您有价值的东西即可。

查看 Python 魔法方法指南：http://www.rafekettler.com/magicmethods.html

另请参阅 __ne__ 与 __eq__ 的答案以及如何定义它们：Python implementing ne operator based on eq。

您还应该研究装饰器（特别是 @property）。