无法导入os.DirEntry时，如何为对应函数添加类型提示？

这问题我之前也碰到过，没法直接导入os.DirEntry但又要加类型提示，其实有几个很实用的解决方案，结合你的FileSystemVisitor类来给你举例：

1. 字符串形式的类型标注 #

Python 3.6+支持直接用字符串来写类型提示，不用提前导入DirEntry，类型检查工具（比如mypy、PyCharm）会自动识别这个字符串对应的类型：

class FileSystemVisitor:
    def visit_dir(self, entry: 'os.DirEntry') -> None:
        # 处理目录的逻辑
        print(f"Visiting directory: {entry.name}")

    def visit_file(self, entry: 'os.DirEntry') -> None:
        # 处理文件的逻辑
        print(f"Visiting file: {entry.path}")

这种方法最直接，不需要额外的导入语句，兼容性也最好。

2. 结合TYPE_CHECKING的条件导入 #

如果你觉得字符串标注不够直观，可以用typing模块里的TYPE_CHECKING常量——它在运行时是False，所以导入语句不会被执行，不会影响程序运行；但类型检查工具会把它当作True，从而正确识别DirEntry类型：

from typing import TYPE_CHECKING

# 只有在类型检查时才导入DirEntry，运行时跳过
if TYPE_CHECKING:
    from os import DirEntry

class FileSystemVisitor:
    def visit_dir(self, entry: DirEntry) -> None:
        pass

    def visit_file(self, entry: DirEntry) -> None:
        pass

这个方案的优势是代码更清晰，类型提示和普通的类型标注写法一致，而且类型检查工具能完整识别DirEntry的所有属性（比如name、path、is_file()方法等），推荐优先用这个。

3. Python 3.10+：使用from __future__ import annotations #

如果你用的是Python 3.10及以上版本，可以通过导入annotations特性，让所有类型提示自动以字符串形式存储，不用手动加引号：

from __future__ import annotations

class FileSystemVisitor:
    def visit_dir(self, entry: os.DirEntry) -> None:
        pass

    def visit_file(self, entry: os.DirEntry) -> None:
        pass

注意这里不需要提前导入os，因为注解会被当作字符串处理，运行时不会解析，只有类型检查工具会去解析这个类型。

不推荐的方案：模拟Stub类 #

你提到的编写模拟类的方法虽然可行，但不建议用——因为类型检查工具无法识别模拟类和真实os.DirEntry的属性差异，可能导致你在代码中调用entry.is_dir()这类方法时，类型检查工具报错（因为模拟类没有这些方法）。如果实在要写，大概是这样：

# 仅用于类型提示的空类，运行时无实际作用
class DirEntry:
    pass

class FileSystemVisitor:
    def visit_dir(self, entry: DirEntry) -> None:
        pass

    def visit_file(self, entry: DirEntry) -> None:
        pass

但这个方案的局限性很大，除非你完全不需要类型检查工具验证DirEntry的属性，否则还是用前面几种方法更好。

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