如何为 Python 变量显式指定 class 类型？

Question

我对 Python 相当陌生，来自 Java 背景。 我有这样的事情：

class A:
    def __init__(self):
        self.var1 = 1
        self.var2 = 2

class B:
    def __init__(self):
        self.my_list = []

    def add(self, a_object):
        self.my_list.append(a_object)

    def show(self):
        for a_object in self.my_list:
            print(a_object.var1, a_object.var2)

现在，我知道这段代码会运行，但我的问题是是否有任何方法可以在show方法中指定a_object变量实际上是 A 类型的 object （就像类型转换 - 我会在 java 中写的东西，比如(A)a_object )。 我首先希望这是为了更好地阅读代码以及自动完成。 我猜另一种解决方案是键入列表，如果可能的话，我也是古玩。

谢谢你。

Answer 1

您可以使用类型提示。 但是请注意，这不是强制执行的，而是引导您 - 以及 IDE - 了解您是否传递了正确的 arguments。

如果您对 static 打字感兴趣，您也可以查看mypy 。

from typing import List


class A:
    def __init__(self):
        self.var1 = 1
        self.var2 = 2

class B:
    def __init__(self):
        self.my_list: List[A] = []

    def add(self, a_object: A):
        self.my_list.append(a_object)

    def show(self):
        for a_object in self.my_list:
            print(a_object.var1, a_object.var2)