如何为Jupyter Notebook/Lab单元格配置类似Decorators的全局代码包裹？

核心方案：利用IPython钩子或自定义扩展实现全局拦截 #

Jupyter底层依赖IPython内核，通过IPython提供的事件钩子或自定义扩展，就能实现类似装饰器的效果，自动为所有单元格添加异常捕获和日志记录逻辑，无需手动编写try-except块。

方法1：通过IPython事件钩子实现（推荐） #

IPython提供pre_run_cell和post_run_cell两个事件钩子，分别在单元格执行前后触发。可以利用post_run_cell钩子捕获执行异常并记录日志：

1. 创建一个启动脚本（比如在~/.ipython/profile_default/startup/目录下新建cell_error_logger.py），内容如下：

import logging
from IPython.core.events import post_run_cell

# 配置日志输出，可自定义路径和格式
logging.basicConfig(
    filename='jupyter_cell_errors.log',
    level=logging.ERROR,
    format='%(asctime)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

def log_cell_errors(result):
    # 检查单元格执行是否抛出异常
    if result.error_in_exec is not None:
        # 记录完整异常回溯
        logger.exception("单元格执行失败", exc_info=result.error_in_exec)
        # 重新抛出异常，保留原有报错提示
        raise result.error_in_exec

# 注册事件钩子
ip = get_ipython()
ip.events.register('post_run_cell', log_cell_errors)

2. 重启Jupyter后，所有单元格执行时的异常都会自动被捕获并记录到指定日志文件，同时正常抛出异常不影响原有流程。

方法2：自定义IPython扩展（更灵活） #

如果需要更复杂的前后执行逻辑，可以编写正式的IPython扩展：

1. 创建扩展文件cell_decorator_ext.py：

import logging
from IPython.core.events import pre_run_cell, post_run_cell

logging.basicConfig(
    filename='jupyter_cell_errors.log',
    level=logging.ERROR,
    format='%(asctime)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

def pre_execution_hook(info):
    # 单元格执行前的逻辑（比如记录开始时间）
    pass

def post_execution_hook(result):
    if result.error_in_exec:
        logger.exception("单元格执行异常")
        raise result.error_in_exec

def load_ipython_extension(ipython):
    ipython.events.register('pre_run_cell', pre_execution_hook)
    ipython.events.register('post_run_cell', post_execution_hook)

def unload_ipython_extension(ipython):
    ipython.events.unregister('pre_run_cell', pre_execution_hook)
    ipython.events.unregister('post_run_cell', post_execution_hook)

2. 加载扩展：
   • 临时加载：在Notebook单元格中运行%load_ext cell_decorator_ext
   • 永久加载：修改~/.ipython/profile_default/ipython_config.py，添加c.InteractiveShellApp.extensions = ['cell_decorator_ext']，确保扩展文件在IPython搜索路径中

关键注意事项 #

• 日志文件默认生成在当前工作目录，可通过修改filename参数指定绝对路径
• 上述方案不会改变单元格原有执行逻辑，仅在异常发生时追加日志记录
• Jupyter Lab完全兼容这些配置，因为底层使用的是同一个IPython内核

内容的提问来源于stack exchange，提问作者Yura Lytkin