python 3：用mypy類型推斷？

Question

在python 3中給出以下代碼段

def foo() -> List[X]: pass

class X:
    def bar(self) -> MYTYPE: pass

對於以下表達式：

[x.bar() for x in foo()]

我可以利用mypy包正確解析上面表達式的AST並猜測結果的類型是List[MYTYPE]嗎？

如果不能，我的最佳選擇是什么？ 有什么想法/套餐可以推薦嗎？

Answer 1

是的，它會推斷出來。 這是一個演示：

from typing import List

class MyType:
    pass

class X:
    def bar(self) -> MyType: pass

def foo() -> List[X]: pass  # note the square brackets

bars = [x.bar() for x in foo()]
reveal_type(bars)

然后，當您在終端中運行mypy script.py ，您會看到一條消息：

Revealed type is 'builtins.list[script.MyType*]'

有關reveal_type文檔在此處 。

順便說一下，您的代碼有一個小問題，應該是List[X]而不是List(X) 。

PyCharm也會識別類型。 bars. 將為列表方法和bars[0].提供自動完成選項bars[0]. 將為MyType提供選項。