如何提取Python with语句块代码并阻止其在本地进程执行？

标准Python上下文管理器的执行流程是解释器硬编码的，没有官方提供的跳过with块执行的接口，以下是两种可直接落地的实现方案：

方案1：利用异常跳转逻辑兼容现有with用法 #

你可以通过自定义私有异常，在__enter__阶段抛出异常跳过with块执行，再在__exit__阶段捕获异常完成远程执行逻辑，完全兼容你现有的调用方式，不需要修改用户侧代码：

# 内部私有异常，仅用于标记跳过本地执行，不对外暴露
class _SkipLocalExecution(Exception):
    pass

class PythonExecutor:
    def __init__(self, env: str = "local"):
        self.env = env

    def __enter__(self):
        # 远程模式直接抛出异常，跳过后续with块执行
        if self.env == "remote":
            raise _SkipLocalExecution()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 捕获我们自定义的跳过异常，执行远程逻辑
        if exc_type is _SkipLocalExecution:
            # 此处直接复用你已经实现的：提取with块代码、SSH远程执行逻辑即可
            print("远程执行完成")
            # 返回True表示异常已处理，不会向上抛出报错
            return True
        # 本地模式或真实运行异常，按默认逻辑处理
        return False

效果验证 #

# 本地模式：with块正常在本地执行
with PythonExecutor(env="local"):
    print("本地执行输出")
    x = 1
    assert x == 1

# 远程模式：with块不会在本地执行，仅触发__exit__中的远程逻辑
with PythonExecutor(env="remote"):
    print("这段内容不会在本地打印")
    x = 1
    assert x == 1

该方案的优势是对使用者完全透明，你已有的代码提取、SSH执行逻辑可以直接复用，额外开发量极低。

方案2：改用装饰器模式（更规范易维护） #

如果可以调整调用方式，更推荐使用装饰器封装执行逻辑，不需要依赖异常跳转的奇技淫巧，后续扩展参数传递、返回值接收等能力也更方便：

import inspect

class PythonExecutor:
    def __init__(self, env: str = "local"):
        self.env = env

    def run(self, func):
        if self.env == "local":
            # 本地模式直接执行函数
            return func()
        else:
            # 远程模式提取函数源码执行SSH逻辑
            func_source = inspect.getsource(func)
            # 此处复用你的远程执行逻辑即可
            print(f"远程执行代码:\n{func_source}")
            return None

调用方式 #

# 本地执行
local_executor = PythonExecutor(env="local")
@local_executor.run
def local_task():
    print("本地执行输出")
    x = 1
    assert x == 1

# 远程执行
remote_executor = PythonExecutor(env="remote")
@remote_executor.run
def remote_task():
    print("这段内容不会在本地打印")
    x = 1
    assert x == 1

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