如何为函数参数添加类型提示，仅接受指定类层级的对象？

首先明确两种需求对应的解决方案：

需求1：接受Computer及其所有子类的实例 #

你原本写的def buy(item: Computer):是完全正确的写法。Laptop和AutomatedTellerMachine作为Computer的子类，它们的实例都符合这个类型提示，而不属于该类层级的对象（比如自定义的非继承类实例）会被mypy拦截。

如果mypy没有正确执行限制，检查以下几点：

• 确认mypy已正确扫描你的代码文件，没有被配置规则排除
• 检查mypy配置（如mypy.ini），建议开启strict = true启用严格类型检查，避免遗漏问题
• 测试示例：传入非该类层级的对象，比如：

class Phone:
    pass

buy(Phone())  # mypy应报错：Argument 1 to "buy" has incompatible type "Phone"; expected "Computer"

需求2：仅接受Computer基类的实例，拒绝子类 #

Python静态类型系统没有原生支持直接限制仅接受基类实例，但可以通过两种方式实现：

方式1：运行时检查 #

在函数内部添加类型判断，直接拦截子类实例：

def buy(item: Computer):
    if type(item) is not Computer:
        raise TypeError("仅允许传入Computer基类实例，不接受子类")
    # 函数实现逻辑

方式2：结合mypy的类型守卫实现静态检查 #

通过自定义类型守卫，让mypy在静态检查阶段识别子类实例：

from typing import TypeGuard

def is_exact_computer(obj: Computer) -> TypeGuard[Computer]:
    return type(obj) is Computer

def buy(item: Computer):
    if not is_exact_computer(item):
        raise TypeError("仅允许传入Computer基类实例")
    # 后续代码中，mypy会将item视为严格的Computer类型，而非子类

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