Python:解决类的定义内部无法使用类名为函数注解及代码补全带来的不便
假定我们要定义一个结点Node类,并实现删除右结点时递归删除后续结点的功能,简单写一下代码如下:
class Node:
def __init__(self, val, right):
self.val = val
self.right = right
def deleteRight(self):
self.right.deleteRight()
为了美观以及便于代码维护的需要,我们一般会使用函数注解标明每个参数以及返回值的类型。另外,如果写了函数注解,我们在编写代码时编译器也会根据参数的类型提供代码补全提示。预期情况下,我们的函数注解应该像是下面这种形式:
class Node:
def __init__(self, val: int, right: Node):
self.val = val
self.right = right
def deleteRight(self) -> None:
self.right.deleteRight()
但是这样写编译器是会有错误提示的,执行代码会报错“NameError: name ‘Node’ is not defined.”,并且使用self.right时也没有Node相关属性及方法的补全提示(这里以Pycharm为例)。简单来说,这是因为在内部Node类还没有完成定义,现在类名是无法使用的状态,更加具体的解释感兴趣的同学可以自行搜索,这里不再赘述。
解决方法也很简单,就是给类名加上引号,代码如下:
class Node:
def __init__(self, val: int, right: 'Node'):
self.val = val
self.right = right
def deleteRight(self) -> None:
self.right.deleteRight()
可以看到现在编写代码时编译器也能正常给出Node类的补全提示了: