将生成器函数定义到类中

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

解决方案:实现一个类,将生成器函数代码放到__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='')
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值