Python中如何自定义继承TextIOWrapper的类并添加扩展方法

你之前的两种方案报错都是因为Python的内置IO类型是C实现的扩展类型，既不允许动态修改实例的__class__属性，也不允许直接给类动态添加成员。

推荐方案：组合包装类（无侵入，兼容性最好） #

核心思路是用Python的鸭子类型特性，写一个包装类持有原始TextIOWrapper实例，通过__getattr__自动转发所有原生IO方法到内部实例，只在包装类中实现自定义方法即可。

import io
from functools import wraps

class ExtendedTextIOWrapper:
    def __init__(self, raw_io: io.TextIOWrapper):
        self._raw_io = raw_io

    # 自定义方法
    def desired_feature(self):
        # 这里写你的自定义逻辑，比如统计行数、内容校验等
        self._raw_io.seek(0)
        return len(self._raw_io.readlines())

    # 自动转发所有未在当前类定义的属性/方法到原始IO实例
    def __getattr__(self, item):
        return getattr(self._raw_io, item)

    # 可选：显式实现上下文管理器协议，兼容with语法
    def __enter__(self):
        self._raw_io.__enter__()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        return self._raw_io.__exit__(exc_type, exc_val, exc_tb)

对应修改你的ReadFile类实现：

class ReadFile:
    def __init__(self):
        self._i = 0

    def do_magic(self, file):
        # 你的原有逻辑
        raw_io = open(file, mode='r', encoding='utf-8')
        # 用包装类包裹后返回
        return ExtendedTextIOWrapper(raw_io)

测试效果完全符合需求：

if __name__ == "__main__":
    import pandas as pd

    rf = ReadFile()
    a = rf.do_magic("./data.csv")
    # pandas读取完全正常，兼容所有原生IO接口
    data = pd.read_csv(a)
    # 自定义方法可正常调用
    print(a.desired_feature())
    # 原生IO方法也可正常调用
    print(a.readable())
    a.close()

    # 也支持with语法自动关闭文件
    with rf.do_magic("./data.csv") as f:
        print(f.desired_feature())

其他可选方案：直接构造TextIOWrapper子类实例 #

如果不想用包装类，也可以直接读取底层二进制流，手动构造子类实例：

import io

class _TextIOWrapper(io.TextIOWrapper):
    def desired_feature(self):
        self.seek(0)
        return len(self.readlines())

class ReadFile:
    def do_magic(self, file):
        # 先以二进制模式打开拿到底层缓冲流
        raw_buf = open(file, mode='rb')
        # 手动构造子类实例，编码等参数和普通open保持一致即可
        return _TextIOWrapper(raw_buf, encoding='utf-8')

该方案的缺点是需要自己处理编码、换行、缓冲区大小等open函数的所有参数，灵活性不如包装类。

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