更改继承方法的类型提示

Question

问题

假设您有一个类A 和一个子类B。如果A 的方法foo 返回与B 不同的变量类型，您将如何在不重新定义/覆盖方法本身的情况下覆盖该方法的类型提示？

现实例子

class RGB:
    def __init__(self, color: Union[tuple[int, int, int], tuple[int, int, int, int]]) -> None:
        self.__color: Union[tuple[int, int, int], tuple[int, int, int, int]] = color
        return

    def __eq__(self, other: Union[RGB, RGBA]) -> bool:
        return isinstance(other, RGB) and self.__color == other.color

    @property
    def color(self) -> tuple[int, int, int]:
        return self.__color


class RGBA(RGB):
    def __init__(self, color: tuple[int, int, int, int]) -> None:
        super().__init__(color)
        return

    @property
    def color(self) -> tuple[int, int, int, int]:
        return self.__color

如您所见，color 属性会根据给定颜色是否具有 alpha 值而有所不同。但我不想/不需要重新定义/覆盖 color 属性方法，功能不会改变。如何在不覆盖方法的情况下为 color 属性定义返回类型提示？

理想情况下，RGBA 类将只包含 __init__ 函数。

Answer 1

您可以有一个通用的Color 类，它作为RGB 和RGBA 的共同父类，由用于表示它们的元组类型参数化。例如，

from typing import Generic, TypeVar


C = TypeVar('C')


class Color(Generic[C]):
    def __init__(self, c: C):
        self.__color = c

    @property
    def color(self) -> C:
        return self.__color


class RGB(Color[tuple[int,int,int]]):
    pass


class RGBA(Color[tuple[int,int,int,int]]):
    pass

这只是一个开始；当您向类添加更多详细信息时，您可能会遇到其他问题。