python类型提示,返回与输入相同的类型

时间:2017-09-14 14:50:02

标签: python pycharm type-hinting

好吧,我正在考虑一个我有一个基类与(几个)孩子的情况。我有一个函数,它接受一个基类对象列表,并返回一个包含这些对象的新列表。

现在,如果我明确地使用子类,则返回是这些子类对象的列表:请考虑以下简单的情况:

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” - 而不是孩子。那么我怎么说:“返回与输入相同的类型”?

我尝试使用typevar:T = TypeVar("T", BaseClass),虽然这给出了实际错误,但TypeVar中无法使用单个约束。有趣的是,使用T = TypeVar("T", BaseClass, ChildClass)确实有效,并且pycharm正确地推导出了sum_div_data的类型(提示)。

1 个答案:

答案 0 :(得分:5)

您应该使用带有上限的typevars:执行T = TypeVar('T', bound=BaseClass)而不是T = TypeVar('T', BaseClass)

详细说明:

  • 当您执行T = TypeVar('T', ClassA, ClassB, ClassC...)之类的操作时,您正在创建type variable with a value restriction。也就是说,您坚持认为T必须恰好是您列出的类别之一。

    这就是禁止执行T = TypeVar('T', ClassA)的原因:typevar只能等于一个类,所以你也可以直接使用ClassA类型。

  • 当您执行T = TypeVar('T', bound=ClassA)之类的操作时,您正在创建type variable with an upper bound。您坚持T必须是ClassA或其任何子类。