你想做的是有意义的,但Python里面还有一些邪恶的东西.
class foo(object):
c = 0
def __init__(self):
self.next = self.next2
def __iter__(self):
return self
def next(self):
if self.c == 5: raise StopIteration
self.c += 1
return 1
def next2(self):
if self.c == 5: raise StopIteration
self.c += 1
return 2
it = iter(foo())
# Outputs: >
print it.next
# 2
print it.next()
# 1?!
for x in it:
print x
foo()是一个迭代器,它可以随时修改其下一个方法 – 在Python中任何地方完全合法.我们创建的迭代器,它有我们期望的方法:it.next是next2.当我们直接使用迭代器时,通过调用next(),我们得到2.然而,当我们在for循环中使用它时,我们得到了原来的下一个,我们已经被明确地覆盖了.
我不熟悉Python内部部件,但是似乎一个对象的“next”方法正在tp_iternext(http://docs.python.org/c-api/typeobj.html#tp_iternext)中缓存,然后当该类更改时,它不会被更新.
这绝对是一个Python的bug.也许这是在生成器PEP中描述的,但它不在核心的Python文档中,并且与普通的Python行为完全不一致.
您可以通过保留原来的下一个功能来解决这个问题,并明确地包装它:
class IteratorWrapper2(object):
def __init__(self, otheriter):
self.wrapped_iter_next = otheriter.next
def __iter__(self):
return self
def next(self):
return self.wrapped_iter_next()
for j in IteratorWrapper2(iter([1, 2, 3])):
print j
…但这显然效率不高,你不应该这样做.