Python：`__eq__()` 在子类时不能按预期工作

Question

首先，我正在使用类似于PEP-0478 的插件系统工作的项目使用PriortyQueue 来确保插件以正确的顺序运行。我将省略它的工作原理：

1. 在 PEP-487 中有详细记录
2. 理解此问题并不直接需要。

   首先，我有以下基本课程：

   
   # abstract.py
   import abc
   
   from dataclasses import data, field
   
   @dataclass(order=True)
   class PluginBase(abc.ABC):
       value: int
   
       def __eq__(self, __o: object) -> bool:
           print("PluginBase __eq__ called")
           return self.value == __o.value
   
   # plugin_a.py
   from abstract import PluginBase
   
   class PluginA(PluginBase):
       pass
   
   # plugin_b.py
   from abstract import PluginBase
   
   class PluginB(PluginBase):
       pass
   
   

   首先，在插件被发现并加载后，它们被包裹在dataclass 中，这是为了确保每个插件的代码量最少。

   问题：__eq__() 永远不会传递给子类。

   我如何设置课程甚至使用funtools.total_ordering 都没有关系。

   我注意到的一件事是，如果你这样做 assert PluginA(1) == PluginB(1) 总是返回 false 并且 assert PluginA(1) < PluginB(2) 总是返回 TypeError: '<' not supported between instances 与当前类。

   这一直是故意的？

   要解决第一个问题，== 不起作用，您需要添加：

       def __eq__(self, __o: object) -> bool:
           print("Plugin __eq__ called")
           return super().__eq__(__o)
   

   向一个或两个子类添加更多我不想要的插件系统的样板代码。

   要修复 '<' 问题，您需要将 PluginBase 更改为：

   @functools.total_ordering
   @dataclass
   class PluginBase(abc.ABC):
       value: int
   
       def __eq__(self, __o: object) -> bool:
           print("PluginBase __eq__ called")
           return self.value == __o.value
   
       def __lt__(self, __o: object) -> bool:
           print("PluginBase __lt__ called")
           return self.value < __o.value
   

   这允许您执行PluginA(1) < PluginB(2)，这将是true。但是，== 在这里仍然不起作用。

   我认为这是由于类实例的差异，Python 强制__eq__() 先检查实例类型。反正有没有让这个工作？

   一种解决方案是将每个插件包装到一个可比较的对象中并将其用于队列。

Answer 1

在 python 中，继承 __eq__ 工作得很好

例子：

class C():
  value = 3
  def __eq__(self, other):
    print('A __eq__ called')
    return self.value == other.value

class A(C):
  value=3
class B(C):
  pass
a, b = A(), B()
print(a == b)

但是，如果您想使用注释 @dataclass 中的 __eq__ ，则不能，因为它会比较类类型。您可以在此处的文档中看到这一点 (https://docs.python.org/3/library/dataclasses.html)：

eq：如果为 true（默认），将生成一个 __eq__() 方法。这个 方法比较类，就好像它是其字段的元组一样，在 命令。比较中的两个实例必须是相同的类型。

简而言之：如果要比较不同的类，则必须手动定义 eq 方法。