如何在Python（打字通用物）中专业班级

Question

如何为专业化（键入 Sense）类提供不同的定义？

例如，为什么这可能有用：

TElement = TypeVar('TElement')

class BaseCollection(Generic[TElement]):
    name: str
    data: List[TElement]

    def __add__(self, other: 'BaseCollection[TElement]'):
        return CombinedCollection[TElement].from_collections(self, other)
    ...

class Collection(BaseCollection):
    pass

# how do I do this specialization
class Collection[int](BaseCollection[int]):
    def sum(self):
        return sum(self.data)


# so that CombinedCollection[int] has also the sum method
class CombinedCollection(Collection[TElement]):
    @classmethod
    def from_collections(cls, *lists: Collection[TElement]):

        return CombinedCollection[TElement]('(' + '+'.join(l.name for l in lists) + ')', 
            [x for l in lists for x in l])

# i.e. I can do
c = Collection[int]('my_collection c', [1,2])
d = Collection[int]('my_collection d', [-1, -2, -3])

cd = c + d
# and now I can do this:
cd.sum()
# -3
cd.name
# (my_collection c+my_collection d)

Answer 1

实际上有一种方法可以做到这一点，但是您不必创建类的“专业”，而需要定义类定义中类的所有方法（如果没有类型的注释，这就是Python的样子），并添加一个type constraint on the self parameter of the method:

class Collection(Generic[T]):
    def sum(self: "Collection[int]") -> int:
        ...

a = C[int]([1, 2, 3])
reveal_type(a.sum())  # int
b = C[str](["a", "b", "c"])
reveal_type(b.sum())  # error: Invalid self argument "C[str]" to attribute function "sum"

See a concrete example on

请注意，这仅适用于具体类型，而不适用于类型。如果您将int替换为typevar用bunds（typevar（“ t”，t'，bound = nucts.real））或用约束（typevar（“ t”，int，float）），mypy似乎忽略了它们，并且会接受任何类型。上面的示例也证明了这一点。我相信这是一个错误或监督。