Python中带默认值的泛型参数定义问题

问题描述 #

现有如下泛型类代码：

from typing import Literal, TypeVar, Generic

T = TypeVar("T", bound=str)

class Foo(Generic[T]):
    def foo(self, arg: T = "foo") -> T:
        return arg

需求是让子类对方法参数做更严格的类型限制，例如子类仅接受"bar"作为参数：

class Bar(Foo[Literal["bar"]]):
    pass

但上述代码无法通过mypy或pyright的类型检查，mypy报错信息如下：

test.py:6: error: Incompatible default for argument "arg" (default has type "str", argument has type "T")

问题根源在于：当子类指定T为Literal["bar"]时，父类方法的默认值"foo"不属于T的合法取值范围，类型检查器会判定类型不兼容。仅用bound=str只能保证T是字符串子类，但无法确保"foo"属于T的取值集合。

解决方案 #

根据需求的不同，有两种可行的解决方式：

方式一：约束T必须包含"foo"字面量 #

修改TypeVar的定义，确保T的取值集合一定包含"foo"，这样父类的默认值始终合法，子类可以扩展其他字符串字面量：

from typing import Literal, TypeVar, Generic

# 约束T必须是包含"foo"的字符串类型，可扩展其他字面量
T = TypeVar("T", bound=Literal["foo"] | str)

class Foo(Generic[T]):
    def foo(self, arg: T = "foo") -> T:
        return arg

# 子类可接受"foo"或"bar"
class Bar(Foo[Literal["foo", "bar"]]):
    pass

这种方式下，子类无法仅限制为"bar"（因为T必须包含"foo"），但能保证父类默认值的合法性，符合类型安全要求。

方式二：子类重写方法并提供符合自身类型的默认值 #

如果确实需要子类仅接受"bar"，则子类必须重写foo方法，替换为符合T类型的默认值：

from typing import Literal, TypeVar, Generic

T = TypeVar("T", bound=str)

class Foo(Generic[T]):
    def foo(self, arg: T = "foo") -> T:
        return arg

class Bar(Foo[Literal["bar"]]):
    # 重写方法，使用符合Literal["bar"]的默认值
    def foo(self, arg: Literal["bar"] = "bar") -> Literal["bar"]:
        return arg

这样既满足子类的严格类型限制，又保证了默认值与参数类型的兼容性，能顺利通过类型检查。

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