python - 正确输入 sys.stdout 和文件

Question

我正在尝试创建一个方法来处理表示文件名的字符串、Path 对象或已打开的输出流(sys.stdout、打开('...'，'w'))。我正在尝试正确设置 mypy 的类型以检查它们。

到目前为止我得到了:

import io
from pathlib import Path
from typing import Union, TextIO, Text


def generate(output: Union[Text, Path, TextIO]) -> None:
    if isinstance(output, io.IOBase):
        output.write("data")
    else:
        if isinstance(output, Text):
            output = Path(output)
        with output.open("w") as output_file:
            output_file.write("data")

但是 mypy 一直提示

Item "TextIO" of "Union[Path, TextIO]" has no attribute "open"

据我所知，TextIO 是用于文本文件的正确类型，但无法针对此类型进行 isinstance 检查。代码结构确保在错误点我们不能有 TestIO 对象，因为它已在上一个分支中处理过。

我应该如何标记这里的所有类型？

Answer 1

这段代码可以重写如下，以隔离 Path 分支中的 open，因为 Mypy 理解 isinstance 检查，并在一个中调用 write地点:

import io
from pathlib import Path
from typing import Union, TextIO, Text
from contextlib import ExitStack


def generate(output: Union[Text, Path, TextIO]) -> None:
    with ExitStack() as stack:
        if isinstance(output, Text):
            output = Path(output)
        if isinstance(output, Path):
            output = stack.enter_context(output.open("w"))
        output.write("data")