17370845950

本文介绍一种结合 `pre_run_cell` 事件钩子与 ast 变换器的可靠方案，实现在单元执行前根据自定义逻辑（如异常、配置检查等）彻底阻止其运行，避免原生钩子无法中断执行的限制。

在 IPython 中，pre_run_cell 事件钩子虽然能在每个单元执行前被调用，但它无法中止后续执行——即使抛出异常，IPython 仍会继续解析并运行原始代码。这是由其设计定位决定的：该钩子用于“通知”而非“拦截”。若需真正实现条件化阻止执行（例如：禁止在生产环境运行调试语句、校验用户权限、检测敏感操作等），必须借助更底层的机制。

推荐方案是 AST 变换器（AST Transformer）配合 pre_run_cell 协同工作：

• pre_run_cell 负责判断条件（如是否满足拦截规则）；
• 若需拦截，则动态注册一个一次性 AST 变换器，将当前单元的抽象语法树（AST）主体清空（node.body.clear()）；
• 由于 AST 在执行前被修改为空，最终实际执行的是空模块，从而实现“逻辑上未运行”。

✅ 关键要点：

• AST 变换器必须仅对当前单元生效，避免污染后续输入，因此需在 visit() 中主动移除自身；
• pre_run_cell 中抛出异常本身不阻断执行，但可作为触发 AST 拦截的信号（如示例中的 except 分支）；
• 所有逻辑均发生在 IPython 输入处理管道的早期阶段，安全且无副作用。

以下是完整可运行示例（兼容 IPython 8+）：

from typing import Any
import ast
import random
from IPython import get_ipython

class BlockExecutionTransformer:
    """AST 变换器：清空当前单元全部语句，实现静默拦截"""
    def visit(self, node: ast.AST) -> Any:
        if not isinstance(node, ast.Module):
            return node
        # ✅ 确保仅作用于本次单元：立即移除自身
        ip = get_ipython()
        if self in ip.ast_transformers:
            ip.ast_transformers.remove(self)
        # ? 清空所有语句，保留模块结构
        node.body.clear()
        return node

def pre_run_cell(info):
    # ? 此处放置你的拦截条件逻辑
    # 示例：模拟随机失败（如检测到危险命令、环境变量不符、权限不足等）
    try:
        # 假设某业务规则：当 random.randint(0,1) == 0 时禁止执行
        denominator = random.randint(0, 1)
        quotient = 1 / denominator  # 可能触发 ZeroDivisionError
        print(f"[✓] 条件通过，允许执行 → {info.raw_cell}")
    except Exception as e:
        # ⚠️ 触发拦截：注册一次性 AST 变换器
        transformer = BlockExecutionTransformer()
        get_ipython().ast_transformers.append(transformer)
        print(f"[✗] 条件不满足（{type(e).__name__}），已屏蔽本次执行")

# ? 注册钩子
get_ipython().events.register("pre_run_cell", pre_run_cell)

? 使用说明：

1. 运行上述代码后，每次执行新单元前都会调用 pre_run_cell；
2. 若条件满足（如 random.randint(0,1) == 1），则正常输出并执行；
3. 若条件不满足（如抛出 ZeroDivisionError），则自动注入 BlockExecutionTransformer，使该单元实际执行为空操作（Out[] 不显示、无副作用、不写入历史）；
4. 变换器自动卸载，不影响下一个单元。

⚠️ 注意事项：

• 不要在 pre_run_cell 中直接 return 或 sys.exit() —— 它们无法中断执行流程；
• AST 变换器需继承自 ast.NodeTransformer 或实现 visit() 方法，且必须注册到 ip.ast_transformers 列表；
• 生产环境建议将条件判断封装为独立函数（如 should_block_cell(info)），提升可读性与可测性；
• 此方案对 %magic 和 !shell 命令同样有效（因其最终也被编译为 AST），但对纯预处理器指令（如 %%time）需额外适配。

总结：IPython 原生不支持 pre_run_cell 中断执行，但通过“钩子 + AST 动态改写”的组合策略，可精准、安全、可复用地实现运行前条件拦截，是构建交互式环境管控能力的核心技术路径。