如何为NamedTemporaryFile添加类型注解？使用typing.IO遇mypy报错

这个问题的核心在于你混淆了NamedTemporaryFile()返回的对象类型与标准IO类型，我来一步步帮你理清并解决：

为什么会报错？ #

当你调用NamedTemporaryFile()时，它返回的并不是标准的IO文件对象，而是tempfile._TemporaryFileWrapper类的实例——这是一个包装器，用来管理临时文件的生命周期（比如自动删除）。这个包装器确实有file（指向实际文件对象）和name（临时文件路径）属性，但标准的typing.IO类型只定义了常规文件操作的接口（比如read、write、fileno），并不包含这两个属性，所以mypy会提示属性不存在，但Python运行时能正常访问（因为实际对象确实有这些属性）。

三种解决方案 #

1. 使用具体的包装器类型（简单但依赖内部实现） #

直接用tempfile._TemporaryFileWrapper作为类型注解，mypy就能识别file和name属性了：

from __future__ import annotations
from tempfile import NamedTemporaryFile, _TemporaryFileWrapper

def example(tmp: _TemporaryFileWrapper) -> str:
    print(tmp.file)
    return tmp.name

print(example(NamedTemporaryFile()))

⚠️ 注意：下划线开头的类属于模块内部实现，可能在Python版本迭代中发生变化，这是一个快速但不够健壮的方案。

2. 自定义Protocol（优雅且安全） #

用typing.Protocol定义一个符合你需求的接口协议，既不依赖内部类，又能让类型检查准确生效：

from __future__ import annotations
from tempfile import NamedTemporaryFile
from typing import IO, Protocol

class TemporaryFileWrapper(Protocol):
    file: IO[bytes]  # 若你用文本模式打开，可改为IO[str]
    name: str

def example(tmp: TemporaryFileWrapper) -> str:
    print(tmp.file)
    return tmp.name

print(example(NamedTemporaryFile()))

这个方案最推荐，因为它明确了你需要的属性，即使_TemporaryFileWrapper的内部逻辑变化，只要它还提供file和name，类型检查就不会出错。

3. 用Any跳过检查（不推荐） #

如果只是临时绕过报错，可以用Any，但这会丢失类型检查的所有好处，不建议长期使用：

from __future__ import annotations
from tempfile import NamedTemporaryFile
from typing import Any

def example(tmp: Any) -> str:
    print(tmp.file)
    return tmp.name

print(example(NamedTemporaryFile()))

额外提示 #

如果你是在上下文管理器中使用NamedTemporaryFile（这是更常见的用法），此时with块内的对象是实际的IO文件对象，这时候用typing.IO注解就是完全正确的：

from tempfile import NamedTemporaryFile
from typing import IO

def process_file(fp: IO[bytes]) -> None:
    fp.write(b"test content")

with NamedTemporaryFile() as tmp:
    process_file(tmp)  # 这里mypy不会报错

内容的提问来源于stack exchange，提问作者Martin Thoma