我正在考虑一种情况,我有一个带有(几个)孩子的基类。我有一个函数,它接受基类对象的列表,并返回一个包含这些对象的新列表。
现在,如果我使用子类,显然返回的是这些子类对象的列表:考虑以下简单情况:
from typing import Sequence, List, TypeVar
class BaseClass:
def __init__(self, data=None, *args, **kwargs):
super().__init__()
self.CanCalculate = True
if data is None:
data = []
self.CanCalculate = False
self._mydata = list(data)
self.multiplier = 1
@property
def data(self):
return self._mydata
class ChildClass(BaseClass):
def sum_mul_data(self):
return self.multiplier * sum(self.data)
class SecondChildClass(BaseClass):
def sum_div_data(self):
return sum(self.data) / self.multiplier
def myFun(input: Sequence[BaseClass]) -> List[BaseClass]:
out = []
for n, i in enumerate(input):
if i.CanCalculate:
i.multiplier = 10**n
out.append(i)
return out
childs = [ChildClass([1,2,3,4]), ChildClass(), ChildClass([1,2,3,4])]
t = myFun(childs)
for idx in t:
print(idx.sum_mul_data())
childs = [SecondChildClass([1,2,3,4]), SecondChildClass(), SecondChildClass([1,2,3,4])]
t = myFun(childs)
for idx in t:
print(idx.sum_div_data())
合法代码:但是 pycharm(和标准类型提示)在静态代码分析期间显示错误: “未解析的属性引用”@
idx.sum_mul_data()
现在显然这是由于 pycharm 认为函数的返回类型是“BaseClass”——而不是子类型。那么我该如何声明:“返回与输入相同的类型”?
我尝试使用类型变量:
T = TypeVar("T", BaseClass)
,尽管这给出了一个实际错误,即单个约束不能在TypeVar
中使用。有趣的是,使用 T = TypeVar("T", BaseClass, ChildClass)
确实有效,并且 pycharm 正确推导了 sum_div_data 的类型(提示)。
您应该使用带有上限的类型变量:使用
T = TypeVar('T', bound=BaseClass)
而不是 T = TypeVar('T', BaseClass)
。
详情: