python self next_Python迭代器-如何在新样式类中动态分配self.next？

作为某些WSGI中间件的一部分，我想编写一个python类来包装迭代器，以便在迭代器上实现close方法。

当我用一个旧样式的类尝试时，这个方法很好，但是当我用一个新样式的类尝试时，它会抛出一个TypeError。我需要做些什么才能让这门课和一门新的课一起使用呢？

示例：class IteratorWrapper1:

def __init__(self, otheriter):

self._iterator = otheriter

self.next = otheriter.next

def __iter__(self):

return self

def close(self):

if getattr(self._iterator, 'close', None) is not None:

self._iterator.close()

# other arbitrary resource cleanup code here

class IteratorWrapper2(object):

def __init__(self, otheriter):

self._iterator = otheriter

self.next = otheriter.next

def __iter__(self):

return self

def close(self):

if getattr(self._iterator, 'close', None) is not None:

self._iterator.close()

# other arbitrary resource cleanup code here

if __name__ == "__main__":

for i in IteratorWrapper1(iter([1, 2, 3])):

print i

for j in IteratorWrapper2(iter([1, 2, 3])):

print j

提供以下输出：1

2

3

Traceback (most recent call last):

...

TypeError: iter() returned non-iterator of type 'IteratorWrapper2'