python-3.x - Python 类型提示 : How to specify file pointer as an argument?

Question

我正在编写一个函数，它将文件指针作为参数并写入它。我想为该参数添加一个类型提示:

def write_some_stuff(fp: _io.TextIOWrapper):
    ...

_io.TextIOWrapper是什么 type(open(...))给我。

还有什么我应该使用的吗？

Answer 1

您应该使用 typing模块，在 Python 3.5 中引入:
typing.TextIO 最适合这种情况。

Generic type IO[AnyStr] and its subclasses TextIO(IO[str]) and BinaryIO(IO[bytes]) represent the types of I/O streams such as returned by open().

在您的示例中:

from typing import TextIO

def write_some_stuff(fp: TextIO):
    ...