Python类型提示：如何限制函数仅接受A的子类实例而非A本身？

关于def fnc(x: A):的理解确认 #

你的理解完全正确：使用x: A作为类型提示时，函数会接受A类本身的实例，同时也会接受所有A的子类实例。因为类型提示中的A代表的是A类型及其所有子类型的实例。

仅接受子类实例的类型提示写法 #

根据子类数量是否确定，有几种不同的实现方式：

1. 子类数量确定时：使用Union #

如果已知所有需要支持的子类（比如B、C、D），可以直接用Union组合这些类型：

from typing import Union

def fnc(x: Union[B, C, D]):
    pass

这种方式简单直接，但如果后续新增子类，需要手动更新类型提示，维护成本较高。

2. 子类数量不确定/动态新增：使用Protocol（Python 3.8+） #

定义一个仅子类会实现的协议（Protocol），让函数接受符合该协议的实例。因为基类A不实现这个协议，类型检查器会自动排除A的实例：

from typing import Protocol

# 定义协议，包含子类特有的方法/属性
class SubclassOfAProtocol(Protocol):
    def subclass_specific_method(self) -> None:
        ...

# 让所有子类实现这个方法（比如B、C、D都要定义subclass_specific_method）
class B(A):
    def subclass_specific_method(self) -> None:
        pass

def fnc(x: SubclassOfAProtocol):
    pass

这种方式无需每次新增子类都修改函数的类型提示，只要子类实现协议要求的内容即可被接受。

3. 结合运行时检查与TypeGuard（Python 3.10+） #

如果需要在运行时强制限制，同时让类型检查器识别合法类型，可以使用TypeGuard：

from typing import TypeGuard

def is_valid_subclass(x: A) -> TypeGuard[B | C | D]:
    # 运行时判断x不是A的直接实例
    return type(x) is not A

def fnc(x: A):
    if not is_valid_subclass(x):
        raise TypeError("仅允许传入A的子类实例")
    # 在此之后，类型检查器会认为x是B/C/D的实例

注意：这种方式下，调用函数时传入A实例，静态检查工具可能不会提前报错，但运行时会抛出异常。

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