我想向 Python 函数添加类型,该函数接受类型作为参数(实际上,特定类的子类型),并返回该类型的实例。想想一个接受特定类型作为参数的工厂,例如:
T = TypeVar('T', bound=Animal)
def make_animal(animal_type: Type[T]) -> T: # <-- what should `Type[T]` be?
return animal_type()
(显然这是一个非常简单的例子,但它演示了这种情况)
这感觉应该是可能的,但我找不到如何正确输入提示。
答案 0 :(得分:0)
这样的事情怎么样?
from __future__ import annotations
from typing import Type
class Animal:
...
def make_animal(animal_type: Type[Animal]) -> Animal:
return animal_type()
答案 1 :(得分:0)
不确定您的问题是什么,您发布的代码是完全有效的 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 输出。