问题:想定义一个生成器函数,但是他还涉及一些额外的状态,我们希望能以某种形式将这些状态暴露给用户
解决方案:实现一个类,将生成器函数代码放到__iter__()
方法中即可
import collections
class linehistory:
def __init__(self, lines, histlen=3):
self.lines = lines
self.history = collections.deque(histlen)
def __iter__(self):
""" 将生成器函数定义在__iter__()方法中并没有对算法做任何改变 """
for lineno, line in enumerate(self.lines, 1):
self.history.append((lineno, line))
yield line
def clear(self):
self.history.clear()
# 要使用这个类,可以将其看做是一个普通的生成器函数。但是,由于他会创建一个类实例,所有可以访问内部属性
with open('somefile.txt') as f:
lines = linehistory(f) # 一个生成器,可迭代
for line in lines:
if 'python' in line:
for lineno, hline in line.history:
print('{}:{}'.format(lineno, hline), end='')