在python中，如何强制将函数参数设为特定的Class

Question

在以下示例中

class A():
    def __init__(self):
        self.foo = 'foo'

class B():
    def __init__(self, a):
        self.a = a

a = A()
B(a)    # right
B('a')  # wrong

我想做类似B.__init__(self, a:A)的事情，以便类B中的参数是A的对象。

Answer 1

您可以随时提出一个ValueError：

class B():
    def __init__(self, a: A):
        if not isinstance(a, A):
            raise ValueError("a must an object of class A")
        self.a = a