Python中如何将字符串代码片段插入到原代码并编译运行？

import ast
import sys
import types

def execute_code_snippet(original_code: str, snippet: str, insertion_point: str) -> dict:
    """
    将代码片段插入到原始代码中并执行
    
    Args:
        original_code: 原始代码字符串
        snippet: 要插入的代码片段
        insertion_point: 插入位置标识（如函数名、类名后）
    
    Returns:
        执行结果的命名空间字典
    """
    # 1. 解析原始代码为AST
    original_ast = ast.parse(original_code)
    
    # 2. 解析代码片段为AST
    snippet_ast = ast.parse(snippet)
    
    # 3. 查找插入位置
    class InsertionVisitor(ast.NodeVisitor):
        def __init__(self, insertion_point):
            self.insertion_point = insertion_point
            self.target_node = None
            
        def visit_FunctionDef(self, node):
            if node.name == self.insertion_point:
                self.target_node = node
                
        def visit_ClassDef(self, node):
            if node.name == self.insertion_point:
                self.target_node = node
    
    visitor = InsertionVisitor(insertion_point)
    visitor.visit(original_ast)
    
    if not visitor.target_node:
        raise ValueError(f"未找到插入点: {insertion_point}")
    
    # 4. 插入代码片段
    if isinstance(visitor.target_node, ast.FunctionDef):
        # 插入到函数体开头
        visitor.target_node.body = snippet_ast.body + visitor.target_node.body
    elif isinstance(visitor.target_node, ast.ClassDef):
        # 插入到类定义中
        visitor.target_node.body = snippet_ast.body + visitor.target_node.body
    
    # 5. 编译和执行
    code_object = compile(original_ast, filename='<string>', mode='exec')
    namespace = {}
    exec(code_object, namespace)
    
    return namespace

# 示例使用
original_code = """
def process_data(x):
    result = x * 2
    return result

class Calculator:
    def add(self, a, b):
        return a + b
"""

snippet = """
    print(f"正在处理数据: {x}")
    # 新增的预处理逻辑
    if x < 0:
        x = abs(x)
"""

# 在process_data函数开头插入代码片段
result_namespace = execute_code_snippet(original_code, snippet, "process_data")

# 测试执行
calc = result_namespace['Calculator']()
print(calc.add(2, 3))  # 输出: 5

result = result_namespace['process_data'](-5)
# 输出:
# 正在处理数据: -5
# 返回: 10

核心是使用AST解析和修改，比eval()或exec()直接拼接字符串更安全可控。

动态代码加载如果还觉得不够的话, 可能只能和楼上说的一样, 用元编程定义一个 DSL 了. 如果不想搞那么复杂, 搞一个中间文件格式出来, 所有客户的输入都统一转成这个格式, 然后再统一处理也行

干脆自己写一个编译器 2333333333333333333333333333333333333333。