我应该关闭在上下文管理器 __exit__() 函数中传递给我的对象的流（类文件对象）吗？

Question

我有一个对象，我希望能够在其上使用 with 关键字。我对实现上下文管理器的实用性感到满意，但我正在努力解决最佳实践类型的问题。

对象是文件的包装器。我计划用字符串（文件的路径）或可以直接处理的类似文件来初始化我的对象（文件中有文件的可能性 - 所以我预见到一个明确的用例这与 BytesIO 等...）

所以__init__看起来像这样：

def __init__(self, file_to_process):
    if isinstance(file_to_process, str):
        self._underlying_stream = open(file_to_process, "rb") # it's the path to a file
    elif isinstance(file_to_process, io.IOBase):
        self._underlying_stream = file_to_process # its the file itself
    else:
         raise TypeError()

所以我的问题是，在我的 __exit__() 函数中关闭 _underlying_stream 是最佳做法/可接受/明智吗？当它是一条路径时，这完全是有道理的，但如果它是一条流进来的，我觉得关闭self._underlying_stream 充其量是不礼貌的，最坏的情况是危险的 - 我这样想是否正确，如果是这样，有没有解决这个问题的好方法？

（注意：我考虑用io.BufferedReader 包装进来的流，但结果证明关闭也会关闭底层流...）

Answer 1

我不会关闭底层流。传入一个已经打开的文件对象意味着调用者已经对该对象负责并且在__exit__上关闭该对象充其量是非常烦人的。

PIL 做了类似的事情，尽管不在上下文管理器中。传入文件名时，它会在完成读取图像数据后关闭文件对象。它为此设置了一个布尔标志。而是传入一个文件对象，它会读取但不会关闭。

我也会在这里做同样的事情：

class Foo(object):
    _close_on_exit = False

    def __init__(self, file_to_process):
        if isinstance(file_to_process, str):
            self._underlying_stream = open(file_to_process, "rb") # it's the path to a file
            self._close_on_exit = True
        elif isinstance(file_to_process, io.IOBase):
            self._underlying_stream = file_to_process # its the file itself
        else:
             raise TypeError()

    def __exit__(self, exc_type, exc_value, traceback):
        if self._close_on_exit:
            self._underlying_stream.close()