如何优雅处理Path与StringIO的文件写入，避免AttributeError？

你可以通过封装上下文管理器的方式，统一Path对象和类文件对象的处理逻辑，既符合Python的EAFP（优先请求原谅而非许可）风格，又不破坏with语句的资源管理结构。

解决方案1：自定义上下文管理器 #

写一个通用的上下文管理器，自动判断输入对象的特征并返回可写入句柄：

import io
import pathlib
import typing
from contextlib import contextmanager

@contextmanager
def get_writable_handle(obj: typing.Union[pathlib.Path, typing.IO]):
    # 判断对象是否具备open方法（Path对象的核心特征）
    if hasattr(obj, 'open'):
        with obj.open('w') as handle:
            yield handle
    else:
        # 类文件对象直接返回，由with语句统一管理（StringIO无需手动关闭）
        yield obj

def foobar(file_like_obj: typing.Union[pathlib.Path, typing.IO]):
    with get_writable_handle(file_like_obj) as handle:
        handle.write('foobar')

解决方案2：try-except简化版 #

如果不想额外编写上下文管理器，也可以直接在函数内用try-except处理，同样遵循EAFP原则：

import io
import pathlib
import typing

def foobar(file_like_obj: typing.Union[pathlib.Path, typing.IO]):
    try:
        # 优先尝试适配Path对象的open逻辑
        with file_like_obj.open('w') as handle:
            handle.write('foobar')
    except AttributeError:
        # 无open方法则直接当作类文件对象写入
        file_like_obj.write('foobar')

测试验证 #

两种方案都能正常处理两种输入类型：

if __name__ == '__main__':
    # 测试Path对象
    p = pathlib.Path.home() / 'my.txt'
    foobar(p)
    
    # 测试StringIO对象
    sio = io.StringIO()
    foobar(sio)
    print(sio.getvalue())  # 输出: foobar

为什么这是Python风格的做法？ #

• 避免了显式的isinstance类型检查，通过行为判断对象能力（是否有open方法），契合Python的"鸭子类型"理念。
• 保留了with语句的资源自动管理特性，确保文件句柄能被正确关闭（针对Path对象）。

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