使用Python迭代器__next__读取文本时仅读取首行且遇空行停止的修复方案咨询

我来帮你梳理下代码里的问题，然后给出修复后的版本：

问题分析 #

你的代码有两个核心问题：

• 空行导致提前终止：当遇到空行时，readline().strip().split()会得到一个空列表，执行pop(0)会抛出IndexError，而你用了宽泛的except捕获所有异常，直接触发StopIteration，导致迭代提前停止。
• 未持续读取后续行：虽然逻辑上是current_line空了就读下一行，但空行触发的异常直接终止了整个迭代，后续行根本没机会被读取。

修复后的代码 #

class ReadFile():
    # 构造函数接收文件名
    def __init__(self, filename):
        self.__file = open(filename)
        self.__current_line = []

    def __iter__(self):
        return self

    def __next__(self):
        while True:
            if len(self.__current_line) == 0:
                line = self.__file.readline()
                # 检查是否到达文件末尾
                if not line:
                    self.__file.close()  # 关闭文件
                    raise StopIteration
                # 处理空行：strip后为空的话，跳过当前行
                stripped_line = line.strip()
                if not stripped_line:
                    continue
                self.__current_line = stripped_line.split()
            # 取出当前行的第一个单词
            return self.__current_line.pop(0)

修改说明 #

1. 精准处理文件结束：只有当readline()返回空字符串（表示文件读完）时，才关闭文件并触发StopIteration，避免误判。
2. 跳过空行：读取到行后先strip，如果结果为空（空行），就直接进入下一次循环读取下一行，不处理为空的行。
3. 缩小异常范围：移除了宽泛的except，只在真正到达文件末尾时主动抛出StopIteration，避免捕获无关异常导致的逻辑错误。
4. 规范类名：按照Python命名规范，类名改用大驼峰ReadFile（不是必须，但更符合社区惯例）。

使用示例 #

假设你的文本文件test.txt内容如下：

hello world

foo bar
baz

使用迭代器的方式：

for word in ReadFile("test.txt"):
    print(word)

输出会是：

hello
world
foo
bar
baz

内容的提问来源于stack exchange，提问作者Sumit