def visit(self, node):
if self._return_level:
if self._node_level == self._return_level:
return None
elif self._node_level < self._return_level:
self._return_level = None
if isinstance(node, ast.Return):
self._return_level = self._node_level
self._node_level += 1
node = super().visit(node)
self._node_level -= 1
return node
评论列表
文章目录