背景: 我想在我没有创建的对象上使用 heapq (以及其他任何东西),这些对象本身没有
__lt__
运算符。我可以吗? (没有包装类)。
班级:
class Node:
def __init__(self, val):
self.val = val
现在,在解释器运行时,我收到了一些对象集合。我想迭代它们,添加一个 dunder 方法(在我的例子中 lt),例如:
n = Node(4)
m = Node(5)
def myLT(self, other):
return self.val < other.val
我尝试过的:
n.__lt__ = types.MethodType(myLT, n)
m.__lt__ = types.MethodType(myLT, m)
还有
n.__lt__ = types.MethodType(myLT, n)
m.__lt__ = types.MethodType(myLT, n)
(如果绑定相同的函子事物会改善问题)
>>> n < m
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: '<' not supported between instances of 'Node' and 'Node'
尽管如此:
>>> n.__lt__(m)
True
我可以使用包装类,这在某些方面是令人讨厌的(额外的内存和遍历代码变得更难看,但至少保持原始对象不变):
class NodeWrapper:
def __init__(self, n):
self.node = n
def __lt__(self):
return self.node.val
我只是想知道我在添加 dunder 方法时是否做错了什么,或者这是否在 python 3.x 中不起作用。如果重要的话我正在使用 3.6.9。
您可以尝试通过更改实例的
__class__
属性来对 dunder 进行猴子修补。正如文档部分特殊方法查找中所解释的:
对于自定义类,特殊方法的隐式调用只是 如果在对象的类型上定义,而不是在 对象的实例字典。
def patch_call(instance, func, memo={}):
if type(instance) not in memo:
class _(type(instance)):
def __lt__(self, *arg, **kwargs):
return func(self, *arg, **kwargs)
memo[type(instance)] = _
instance.__class__ = memo[type(instance)]
patch_call(m, myLT)
patch_call(n, myLT)
n < m
# True
感谢 @juanpa.arrivilaga 建议缓存类以提高性能。