如何在python*中分析python代码*以识别变量名、函数调用等

import ast

def analyze_code(code):
    tree = ast.parse(code)
    # 遍历抽象语法树，获取变量名、函数调用等信息
    # ...

# 调用示例
code = """
def add(a, b):
    return a + b

x = 1
y = 2
z = add(x, y)
"""
analyze_code(code)

import ast

class CodeAnalyzer(ast.NodeVisitor):
    def visit_Assign(self, node):
        # 处理赋值语句，获取变量名和值
        for target in node.targets:
            if isinstance(target, ast.Name):
                variable_name = target.id
                print("Variable:", variable_name)
        self.generic_visit(node)

    def visit_Call(self, node):
        # 处理函数调用，获取函数名和参数
        if isinstance(node.func, ast.Name):
            function_name = node.func.id
            print("Function call:", function_name)
        self.generic_visit(node)

def analyze_code(code):
    tree = ast.parse(code)
    analyzer = CodeAnalyzer()
    analyzer.visit(tree)

# 调用示例
code = """
def add(a, b):
    return a + b

x = 1
y = 2
z = add(x, y)
"""
analyze_code(code)