python - 有没有办法在本地覆盖 __eq__ 函数/运算符并在之后恢复旧的？

Question

我需要在受限范围内对对象进行自定义比较，有没有办法做到这一点并且不会污染操作员，例如之后恢复以前的eq？

class Test():
    def __eq__(self, i):
        print "eq lvl 1"

def test(a , b):
    def _itest(i):
        print "eq lvl 2"

>>> a = Test()
>>> b = Test()
>>> a == b
eq lvl 1
>>> test(a, b)
>>> a == b
Traceback (most recent call last):
File "<console>", line 1, in <module>
TypeError: _itest() takes exactly 1 argument (2 given)

我这样做是因为在某些条件下我需要降低eq运算符。

注意：我想重写__eq__以使用该in语句。

Answer 1

您的样本太小，所以我会尝试推断...

def somewhere():
  old_eq = obj.__eq__
  def new_eq(a, b):
    return False
  obj.__eq__ = new_eq
  # ^^^ Will fail here because you can't assign new __eq__ to the object
  if not obj == obj:
    print('Well, ok...')
  obj.__eq__ = old_eq

您仍然可以尝试使用可配置（通过钩子）__eq__方法创建自己的对象，并用这个方法替换您的对象，例如：

class Patched(YourClass):
  def __eq__(self, i):
    if self.original__eq:
      return YourClass.__eq__(self, i):
    else:
      pass # your code here