如何Pythonic实现无需在构造函数打开文件、逐行返回内容的类
优化实现
以下实现完全满足「不在构造函数中打开文件」的约束,同时修正了原实现的缺陷、符合Python编码规范:
import petl as etl class CSVReader: def __init__(self): # 构造函数仅初始化私有属性,不涉及文件操作 self._data = None self._file_path = None def load(self, file_path: str): """加载CSV文件路径,支持链式调用""" self._file_path = file_path self._data = None return self def rows(self): """逐行yield返回CSV行数据""" # 惰性初始化,首次调用时才真正读取文件 if self._data is None: if self._file_path is None: raise RuntimeError("请先调用load()方法加载CSV文件") self._data = etl.fromcsv(self._file_path) yield from self._data.namedtuples() # 实现迭代器协议,支持直接迭代实例 def __iter__(self): yield from self.rows()
优化说明
- 修正原实现的核心缺陷:原实现将
data、initialized声明为类属性,多实例场景下会出现数据串扰;同时原代码中read_file方法参数名和内部使用的变量名不一致,会触发变量未定义报错 - 去掉冗余状态字段:无需单独的
initialized标记,直接通过_data判断初始化状态,代码更简洁 - 惰性加载逻辑更严谨:首次调用
rows()方法时才真正读取文件,完全满足构造阶段不操作文件的要求 - 更符合Python使用习惯:实现了迭代器协议,支持直接通过
for row in 实例的方式遍历数据,无需显式调用rows()方法 - 支持链式调用和重复加载:多次调用
load()方法会自动重置旧数据,无需手动修改状态标记
使用示例
# 基础用法 reader = CSVReader() reader.load("demo.csv") for row in reader.rows(): print(row) # 更简洁的迭代写法 reader = CSVReader().load("demo.csv") for row in reader: print(row)
内容的提问来源于stack exchange,提问作者Alvaro Aguilar
相关产品推荐
相关产品推荐

