假设,我有这段代码:
class BaseModel(object):
pass
class AppleModel(BaseModel):
pass
class OrangeModel(BaseModel):
pass
def get_record(model, pk):
try:
result = model.objects.get(pk=pk)
# Some code working with BaseModel..
return result
except model.DoesNotExist:
return None
现在,我想像在此示例中那样使用它并带有适当的类型注释...
myApple = get_record(AppleModel, 12)
myApple. # type hints for AppleModel
myOrange = get_record(OrangeModel, 433)
myOrange. # type hints for OrangeModel
我应该如何注释
get_record
函数?我尝试了类似的东西def get_record(model: Type[BaseModel], id: int) -> Optional[BaseModel]:
# ...
但是随后,我丢失了有关子类的信息。我尝试使用Generic,但没有成功。
最佳答案
我假设您想做的是让get_record
接受BaseModel
的某些子类的类型,并返回相应类的实例。
在这种情况下,您可能需要使用TypeVars-特别是TypeVars with an upper bound restriction。
大致如下所示:
from typing import Type, Optional, TypeVar
# ...snip...
T = TypeVar('T', bound=BaseModel)
def get_record(model: Type[T], id: int) -> Optional[T]:
# ...
关于python - 通用代码的类型提示,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/48287588/