python - 覆盖实例下一个函数

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