如何缩短包含长with上下文管理器的try/except代码块？

你目前想到的将with块内逻辑抽离为独立函数的写法已经是Python中最常用、可读性最高的实现方案，完全符合最佳实践。如果你追求更简洁、代码量更少的写法，可以根据你的异常捕获需求选择以下方案：

方案1：拆分异常捕获范围（仅需处理连接建立失败时使用） #

ConnectionError通常仅会在数据库连接建立阶段（也就是with语句执行上下文管理器的__enter__方法时）抛出，连接建立成功后的普通数据库操作异常一般为OperationalError、ProgrammingError等其他类型。如果你不需要捕获with块内部业务逻辑的异常，可以把try/except仅套住连接建立步骤，手动管理上下文的退出：

try:
    connection = db.connect().__enter__()
except ConnectionError:
    # 处理连接失败的逻辑
    handle_connection_err()
else:
    # 连接成功后执行的长逻辑，完全不被try/except包裹
    try:
        # many lines interacting with the database
        # and more lines
        # and yet more lines
    finally:
        # 手动关闭连接，避免资源泄漏
        connection.__exit__(None, None, None)

方案2：封装通用安全连接管理器（需要捕获全生命周期连接异常时使用） #

如果你需要覆盖连接建立后、使用过程中抛出的ConnectionError，又不想每次写数据库操作都重复写try/except，可以用contextlib封装一次通用的安全连接上下文管理器，后续所有场景都可以直接复用：

from contextlib import contextmanager

@contextmanager
def safe_db_connection():
    try:
        with db.connect() as conn:
            yield conn
    except ConnectionError:
        # 统一处理连接错误逻辑
        handle_connection_err()

# 业务代码无需重复写try/except
with safe_db_connection() as connection:
    # many lines interacting with the database
    # and more lines
    # and yet more lines

这种写法一次封装多次复用，长期来看整体代码量远低于每次重复写异常捕获逻辑。

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