StrEnums与mypy类型收窄问题：子类重写方法类型不兼容如何解决？

问题原因 #

你遇到的MyPy报错本质是违反了里氏替换原则（LSP）：父类Animal的speak方法声明接受任意StrEnum实例，但子类Frog的speak只接受FrogType——如果用Animal类型引用Frog实例，调用speak时传入其他StrEnum（比如另一种动物的枚举），运行时必然出错。静态类型检查会严格拦截这种不符合LSP的代码。

解决方案：使用泛型抽象基类 #

最规范的修复方式是给抽象基类引入泛型，让每个Animal子类绑定自己专属的枚举类型，既实现类型收窄，又符合类型系统规则：

from abc import ABC, abstractmethod
from enum import StrEnum
from typing import Generic, TypeVar

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

class Animal(ABC, Generic[T]):
    @staticmethod
    @abstractmethod
    def speak(animal_type: T):
        ...

class FrogType(StrEnum):
    frog = "frog"
    toad = "toad"

class Frog(Animal[FrogType]):
    @staticmethod
    def speak(animal_type: FrogType):
        print(f"{animal_type} goes ribbit")

验证 #

修改后运行代码和MyPy检查都能正常通过：

Frog.speak(FrogType.frog)  # 输出：frog goes ribbit

python3.11 -m mypy frog.py
# 无报错输出

这种方式明确了每个动物子类与对应枚举类型的绑定关系，既满足静态类型检查的要求，又保留了代码的灵活性。

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