注释函数参数类型

Question

有这样的方法：

def pdf_repr(self, users):
    ···

我知道我可以用

注释参数行

users: list

但是，如果列表由以下人员组成，比如：

class User:
    ...

objects，我能否以某种方式注释参数以不仅描述列表而且描述内部对象？

我需要这个，所以编辑器可以帮助我自动完成，而不必检查或记住用户对象的属性。

或者，如果它是另一种实现方式，我想知道。我正在使用 VS 代码。

Answer 1

您可以使用键入库中的 List 来显式键入自定义 class 对象的列表。

from typing import List

class User:
    pass

def pdf_repr(self, users: List[User]):
    pass

看看打字库doc。

Anotate a function argument type