python - 在python中，如何强制函数参数属于特定类

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