Python：symbol 库高级用法举例和应用详解

import ast  # 导入抽象语法树库
import symbol  # 导入symbol库

code = """
def hello_world():
    print("Hello, World!")
"""

# 解析Python代码为抽象语法树(AST)
tree = ast.parse(code)

# 递归地打印语法树中的符号
def print_symbols(node, level=0):
    if isinstance(node, ast.AST):
        symbol_name = symbol.sym_name.get(node.__class__.__name__, node.__class__.__name__)
        print(" " * level + symbol_name)
        for child in ast.iter_child_nodes(node):
            print_symbols(child, level + 2)

print_symbols(tree)
# 此代码展示如何解析函数结构并通过symbol库打印出对应符号

import ast  # 导入抽象语法树库
import symbol  # 导入symbol库

code = """
x = 10

def example():
    y = 5
    return x
"""

# 解析Python代码为抽象语法树(AST)
tree = ast.parse(code)
  
# 遍历语法树查找全局变量
class GlobalVariableAnalyzer(ast.NodeVisitor):
    def visit_Assign(self, node):
        if isinstance(node.targets[0], ast.Name):
            if isinstance(getattr(node, 'ctx', None), ast.Store):
                symbol_name = symbol.sym_name.get(node.__class__.__name__, node.__class__.__name__)
                print(f"Found global variable: {node.targets[0].id} (symbol: {symbol_name})")
        self.generic_visit(node)

analyzer = GlobalVariableAnalyzer()
analyzer.visit(tree)
# 此代码展示如何通过symbol库和AST库实现简单的代码静态分析工具

import ast  # 导入抽象语法树库
import symbol  # 导入symbol库

code = """
def greet(name):
    print(f"Hello, {name}!")

greet("Alice")
"""

# 解析Python代码为抽象语法树(AST)
tree = ast.parse(code)

# 遍历语法树查找函数调用
class FunctionCallAnalyzer(ast.NodeVisitor):
    def visit_Call(self, node):
        function_name = node.func.id if isinstance(node.func, ast.Name) else 'unknown'
        symbol_name = symbol.sym_name.get(node.__class__.__name__, node.__class__.__name__)
        print(f"Function call: {function_name} (symbol: {symbol_name})")
        self.generic_visit(node)

analyzer = FunctionCallAnalyzer()
analyzer.visit(tree)
# 此代码展示如何通过symbol库解析并打印函数调用的符号结构