如何让自定义布尔函数实现类似isinstance的静态类型推断效果？

兄弟，你这个需求刚好对应Python类型系统里的**类型守卫（Type Guard）**特性！PyCharm、mypy这类静态类型检查工具都支持它，专门用来解决自定义判断函数的类型推断问题，完美替代isinstance的静态推断能力，甚至还能推断像List[Tuple]这样的复杂类型。

下面直接给你改造方案：

1. 使用TypeGuard注解你的判断函数 #

从Python 3.10开始，标准库的typing模块里新增了TypeGuard，它的作用就是告诉类型检查器：当这个函数返回True时，输入参数的类型就是你指定的那个类型。

改造你的dict_like函数，把返回类型从bool改成TypeGuard[List[Tuple[Any, ...]]]（这里的Tuple[Any, ...]表示任意元素的元组，你可以根据实际需求改成更精确的类型，比如Tuple[str, int]）：

from typing import TypeGuard, List, Tuple, Any

def dict_like(x: object) -> TypeGuard[List[Tuple[Any, ...]]]:
    if isinstance(x, list) and len(x) > 0 and isinstance(x[0], tuple):
        return True
    return False

def handle_dict(y: List[Tuple[Any, ...]]):
    # 这里可以放心使用y的List[Tuple]特性，IDE会自动提示
    pass

def handle_list(w: list):
    pass

def foobar(z: object):
    if dict_like(z):
        handle_dict(z)  # 现在PyCharm会自动推断z是List[Tuple[...]]，不会报类型错误了
    elif isinstance(z, list):
        handle_list(z)
    else:
        pass

2. 旧Python版本的兼容方案 #

如果你还在使用Python 3.9及以下版本，只需要先安装typing_extensions库（pip install typing_extensions），然后从typing_extensions导入TypeGuard即可，用法完全一样：

from typing_extensions import TypeGuard
from typing import List, Tuple, Any

# 后面的代码和上面完全相同

3. 更精确的类型推断 #

如果你的dict_like函数能确保元组里的元素是特定类型（比如每个元组都是(str, int)），直接把TypeGuard里的类型改成对应的精确类型就行：

def dict_like(x: object) -> TypeGuard[List[Tuple[str, int]]]:
    # 这里可以补充更严格的判断逻辑，确保每个元组都是(str, int)
    if isinstance(x, list) and len(x) > 0:
        return all(isinstance(item, tuple) and len(item)==2 and isinstance(item[0], str) and isinstance(item[1], int) for item in x)
    return False

这样类型检查器会在if dict_like(z):分支里直接把z推断成List[Tuple[str, int]]，比isinstance的基础推断还强大！

备注：内容来源于stack exchange，提问作者Daniel Lawson