Python

将生成器函数定义到类中

2018-10-03  本文已影响1人  cook__

问题:想定义一个生成器函数,但是他还涉及一些额外的状态,我们希望能以某种形式将这些状态暴露给用户

解决方案:实现一个类,将生成器函数代码放到__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='')
上一篇 下一篇

猜你喜欢

热点阅读