Python如何在cd上下文管理器切换目录失败时跳过with语句块执行

问题根源说明 #

1. RuntimeError: generator didn't yield 报错原因：@contextmanager 装饰的生成器必须至少执行一次 yield，如果目录切换失败后直接跳过 yield 就会触发该错误。
2. 虚假报错原因：原有实现将 yield 放在了捕获目录切换异常的 try 代码块内，with 语句块执行时抛出的异常会传回生成器中 yield 的位置，被该 try 对应的 except 块捕获，导致误判为目录切换失败。

方案一：零改造兼容原有调用方式（推荐） #

如果你的业务场景允许目录切换失败时向上抛出异常，该方案可以完全兼容你期望的 with cd('here'): 用法，无需修改任何现有调用代码，同时完全避免虚假报错。

实现代码 #

import os
import traceback
import logging as log
from contextlib import contextmanager

@contextmanager
def cd(to_dir):
    from_dir = os.getcwd()
    # 仅捕获目录切换阶段的异常
    try:
        os.chdir(os.path.expanduser(to_dir))
    except Exception:
        log.error(traceback.format_exc())
        log.error(f"Failed to cd into {to_dir} from {from_dir}")
        # 抛出异常终止上下文进入，with 语句块不会执行
        raise
    try:
        # yield 独立于目录切换的异常捕获块，不会捕获with块内部异常
        yield
    finally:
        os.chdir(from_dir)

优势 #

• 零改造适配原有调用逻辑，项目改造成本为0
• 目录切换失败时 with 语句块绝对不会执行，符合核心需求
• 完全不会捕获 with 块内部的异常，没有虚假报错问题

方案二：静默失败不抛出上层异常 #

如果要求目录切换失败时仅打印日志、不向上抛出异常，你可以优化原有带状态返回的实现，解决虚假报错问题，仅需要在调用处增加少量判断逻辑。

实现代码 #

import os
import traceback
import logging as log
from contextlib import contextmanager

@contextmanager
def cd(to_dir):
    from_dir = os.getcwd()
    cd_success = False
    # 仅捕获目录切换阶段的异常
    try:
        os.chdir(os.path.expanduser(to_dir))
        cd_success = True
    except Exception:
        log.error(traceback.format_exc())
        log.error(f"Failed to cd into {to_dir} from {from_dir}")
    
    try:
        # yield 放在独立的块中，不会被目录切换的except捕获
        yield cd_success
    finally:
        # 只有切换成功才需要切回原目录
        if cd_success:
            os.chdir(from_dir)

使用方式 #

with cd('here') as success:
    if success:
        # 原有业务逻辑
        ...

优势 #

• 解决了原有实现的虚假报错问题
• 目录切换失败时不会向上抛出异常，上层业务无感知
• 仅需要在调用处增加一层判断，改造成本可控

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