如何在Python嵌入场景截断调用栈？PypTeX调试优化问题

我开发了一款「编译器」PypTeX，可将包含Hello @{3+4}的输入文件a.tex转换为输出文件a.pyptex，其中@{3+4}会被计算为7。我通过eval(compile('3+4','a.tex',mode='eval'),myglobals)执行这类Python代码片段（myglobals为初始空字典），虽营造出嵌入式解释器的假象，但执行时的调用栈会回溯至PypTeX内部，而非终止于用户代码'3+4'。

请问是否存在类似eval但能截断调用栈顶部的实现方式？

调试场景动机 #

若Python代码片段在numpy内部抛出异常并启动pdb，用户输入up进入自身代码作用域后执行list，当前实现会正确显示a.tex文件；但用户再次输入up时，会进入PypTeX编译器的内部代码，带来困惑。这类似于g++报错时暴露自身内部调用栈的问题。

核心需求 #

当前eval继承调用者的调用栈，若能为其提供截断后的栈，或让pdb停止在指定栈帧（如截断回溯对象后传入pdb.post_mortem()），即可解决问题；此外，若能创建拥有独立调用栈、全局变量的干净嵌入式解释器（如from sys import Interpreter; foo = Interpreter(); foo.eval(...)），也可满足需求。

已否决方案 #

通过正则提取TeX中的Python代码存入临时文件a.py，调用新Python解释器执行的方案体验极差：a.py是自动生成的实现细节，用户难以关联其与a.tex的对应关系，大文档场景下可用性极低。可参考pythontex。

1. 截断异常回溯栈 #

捕获用户代码抛出的异常后，手动修改异常的__traceback__属性，移除PypTeX内部的栈帧：

def execute_user_code(code_str, filename):
    myglobals = {}
    try:
        compiled = compile(code_str, filename, mode='eval')
        return eval(compiled, myglobals)
    except Exception as e:
        # 遍历回溯栈，定位到用户代码对应的帧
        tb = e.__traceback__
        while tb and tb.tb_frame.f_code.co_filename != filename:
            tb = tb.tb_next
        # 将截断后的回溯栈赋值给异常
        e.__traceback__ = tb
        raise

这样异常抛出时，回溯只会从用户代码的栈帧开始，pdb调试时up不会进入编译器内部。

2. 自定义pdb栈过滤逻辑 #

通过自定义pdb的栈帧过滤，默认隐藏PypTeX内部的栈帧，让用户调试时只看到自己的代码：

import pdb

def filter_internal_frames(frame, event, arg):
    # 根据实际情况判断是否为PypTeX内部栈帧，比如通过文件路径前缀
    if frame.f_code.co_filename.startswith('/path/to/pyptex/'):
        return None
    return frame

# 在启动pdb时应用过滤
pdb_instance = pdb.Pdb()
pdb_instance.set_trace()

# 或者在异常处理中启动带过滤的pdb
def debug_user_exception(e):
    tb = e.__traceback__
    # 先定位到用户代码帧
    while tb and tb.tb_frame.f_code.co_filename != 'a.tex':
        tb = tb.tb_next
    pdb.post_mortem(tb)

3. 模拟独立嵌入式解释器 #

通过封装eval和隔离的全局作用域，结合回溯截断，模拟拥有独立全局变量的干净解释器：

class EmbeddedInterpreter:
    def __init__(self):
        self.globals = {}  # 独立的全局变量空间
    
    def eval(self, code_str, filename):
        try:
            compiled = compile(code_str, filename, mode='eval')
            return eval(compiled, self.globals)
        except Exception as e:
            tb = e.__traceback__
            while tb and tb.tb_frame.f_code.co_filename != filename:
                tb = tb.tb_next
            e.__traceback__ = tb
            raise

# 使用示例
interp = EmbeddedInterpreter()
result = interp.eval('3+4', 'a.tex')

关键注意点 #

• 栈帧判断逻辑需要适配PypTeX的实际文件路径或代码标识，确保准确跳过内部帧。
• 处理回溯栈时要注意边界条件，避免tb为None引发错误。
• 这种方案无法完全消除底层调用栈，但能让用户调试时看不到编译器内部细节，达到需求的体验。

内容的提问来源于stack exchange，提问作者Sébastien Loisel