python - 关闭在对象初始化中打开的文件的正确方法是什么？

Question

我希望有一个类在初始化期间传递字符串或已打开的文件。如果它得到一个字符串，它会打开文件。

from typing import IO

class Parser:
  def __init__(self, fin: str|IO[str]) -> None:
    if isinstance(fin, str):
      self.fin = open(fin, 'r')
    else:
      if not fin.readable():
        raise ValueError("Input file must be readable.")
      else:
        self.fin = fin

我的问题是，如果文件被打开，关闭文件的正确方法是什么。我想象它可以在 __del__ 方法中关闭，但在阅读之后，似乎一致认为使用 __del__ 不是一个好主意。有更好的方法吗？

Answer 1

提供一个单独的类方法来处理使用上下文管理器打开和关闭文件。

from contextlib import contextmanager


class Parser:
    def __init__(self, fh: IO[str]) -> None:
        if not fh.readable():
            raise ValueError(...)
        self.fh = fh

    # Typing-hinting is out of scope for this question, so I'm
    # just using the Python-3.11-style Self hint for simplicity.
    @classmethod
    @contextmanager
    def from_filename(cls: Self, name: str) -> Self:
        with open(name) as fh:
            yield Parser(fh)

现在你可以使用任何一个

with open("some_file.txt") as fh:
    p = Parser(fh)
    ...

或

with Parser.from_filename("some_file.txt") as p:
    ...