Python：鍵入一個通用 function 接收一個類型並返回該類型的實例

Question

我想將類型添加到 Python function 中，它將類型作為參數（實際上是特定類的子類型），並返回該類型的實例。 考慮一個將特定類型作為參數的工廠，例如：

T = TypeVar('T', bound=Animal)

def make_animal(animal_type: Type[T]) -> T:  # <-- what should `Type[T]` be?
    return animal_type()

（顯然這是一個非常簡單的例子，但它演示了案例）

這感覺應該是可能的，但我找不到如何正確輸入提示。

Answer 1

不確定您的問題是什么，您發布的代碼是完全有效的 Python 代碼。 有typing.Type完全符合您的要求：

from typing import Type, TypeVar

class Animal: ...
class Snake(Animal): ...

T = TypeVar('T', bound=Animal)

def make_animal(animal_type: Type[T]) -> T:
    return animal_type()

reveal_type(make_animal(Animal))  # Revealed type is 'main.Animal*'
reveal_type(make_animal(Snake))   # Revealed type is 'main.Snake*'

請參閱mypy-play上的 mypy output 。

Answer 2

這樣的事情怎么樣？

from __future__ import annotations
from typing import Type


class Animal:
    ...


def make_animal(animal_type: Type[Animal]) -> Animal:
    return animal_type()