utils.py 文件源码

def get_variables(root):
    class Visitor(ast.NodeVisitor):
        def __init__(self):
            self.vars = set()

        def visit_Name(self, node):
            self.vars.add(node.id)

    var_visitor = Visitor()
    var_visitor.visit(root)
    return var_visitor.vars