如何为自定义类的参数指定类型提示？ [复制]

Question

假设我创建了一个定义如下的类，并在其上调用了方法：

class Student:

    def __init__(self, name):
        self.name = name
        self.friends = []

    def add_friend(self, new_friend: Student):
        self.friends.append(new_friend)

student1 = Student("Brian")
student2 = Student("Kate")
student1.add_friend(student2)

add_friend 方法有一个名为new_friend 的参数，它是一个Student 对象。如何使用类型提示来指定它？我假设您只需输入类的名称，例如new_friend: Student，但它不起作用。当我运行它时，我得到一个NameError: name 'Student' is not defined。我也试过new_friend: __main__.Student，但它给了我同样的错误。我做错了什么？

Answer 1

根据PEP-484，使用类的字符串名称进行前向引用：

class Student:

    def __init__(self, name):
        self.name = name
        self.friends = []

    def add_friend(self, new_friend: 'Student'):
        self.friends.append(new_friend)