我正在编写一个函数,该函数将文件指针作为参数并写入其中。我想向该参数添加类型提示:
def write_some_stuff(fp: _io.TextIOWrapper):
...
_io.TextIOWrapper
是type(open(...))
给我的。还有其他我应该使用的东西吗?
最佳答案
您应该使用python 3.5中引入的typing
模块:
typing.TextIO
最适合这种情况。
通用类型IO[AnyStr]
及其子类TextIO(IO[str])
和BinaryIO(IO[bytes])
表示I / O流的类型,例如open()
返回。
在您的示例中:
from typing import TextIO
def write_some_stuff(fp: TextIO):
...