如何让Python函数根据输入自动推断派生类的返回类型？

你遇到的核心问题是：当前的类型注解没有把Animal子类和它对应的Food子类类型关联起来，导致get_food无法自动推断出派生类实例对应的具体Food类型。

最优解决方案：将Animal定义为泛型类 #

通过给Animal添加泛型参数绑定对应的Food类型，让类型系统能够关联Animal实例和其food的具体类型，从而让get_food自动返回正确的类型，完全不需要依赖子类信息或者显式类型转换。

修改后的Module 1代码：

from typing import TypeVar, Generic

# 定义绑定Food的类型变量
T = TypeVar("T", bound="Food")

class Food:
    pass

# 让Animal继承Generic[T]，将food的类型和泛型参数绑定
class Animal(Generic[T]):
    def __init__(self, food: T) -> None:
        self.food: T = food

# 利用Animal的泛型参数，直接返回对应的T类型
def get_food(animal: Animal[T]) -> T:
    return animal.food

# 基类测试：类型推断正确
food = get_food(Animal(Food()))
reveal_type(food)  # 输出类型为Food

Module 2代码（无需修改get_food）：

from module1 import Food, Animal, get_food

class Carrot(Food):
    pass

# 继承Animal时指定泛型参数为Carrot，明确Rabbit对应的Food类型
class Rabbit(Animal[Carrot]):
    def __init__(self, food: Carrot) -> None:
        super().__init__(food)

# 派生类测试：类型自动推断为Carrot
food = get_food(Rabbit(Carrot()))
reveal_type(food)  # 输出类型为Carrot，符合期望

方案优势 #

• 符合开闭原则：模块1无需知晓任何子类实现，新增Animal或Food子类时，只需要在子类中指定泛型参数即可
• 无冗余代码：不需要额外定义重载、重新实现get_food或使用cast进行类型转换
• 类型安全：类型推断完全由类型系统自动完成，避免手动转换带来的潜在类型错误

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