覆盖实例下一个函数

Question

我正在尝试使用下面的代码 (python 2.7) 覆盖对象的下一个函数。

直接调用对象的next方法时，会调用新函数。然而，当我在我的对象上调用内置的 next() 函数（according to the docs 应该调用实例的 next 方法）时，会调用 ORIGINAL 函数。

有人可以解释这种行为吗？

class Test(object):
    def __iter__(self):
        return self
    def next(self):
        return 1

test = Test()
def new_next(self):
    return 2

test.next = type(test.__class__.next)(new_next, test, test.__class__)
print test.next() # 2
print next(test) # 1

Answer 1

如果我正确阅读this source，似乎在定义类时设置了迭代器。不过我可能读错了。我猜这是为了快速查找next 函数（将其设置为slot），因为它用于循环等。

鉴于此，以下似乎可以满足您的需求：

>>> test.__class__.next = type(test.__class__.next)(new_next, test, test.__class__)
>>> next(test)
2