SQLAlchemy1.4搭配Session上下文管理器插入重复主键异常未捕获问题

• SQLAlchemy的session.add()方法仅会将实体对象加入当前会话的待操作队列，不会立刻向数据库发送SQL请求，也不会执行主键唯一性校验。真正的INSERT语句、约束校验都是在会话触发flush（刷写）或者commit（提交）操作时才会执行，因此主键重复导致的IntegrityError不会在add调用时抛出。
• 你自定义的MockSession是直接在add方法里主动抛出异常，所以可以被包裹add的try-except捕获，和真实SQLAlchemy Session的运行逻辑完全不同，这就是两者表现不一致的根源。
• 再看session_scope上下文管理器的执行流程：当with块里的代码（也就是你写的session.add(label2)那部分）执行完成没有抛出异常时，才会执行yield之后的session.commit()逻辑，此时已经脱离了你包裹add的try-except作用域，异常会直接从上下文管理器内部抛出，自然进不到你写的except块里。

方案1：add后手动触发flush，在with块内部捕获异常 #

修改代码，add之后立刻调用session.flush()主动触发数据库交互，此时异常会在try块内抛出：

with session_scope(session_cls=Session) as session:
    try:
        session.add(label2)
        session.flush() # 手动触发刷写，立刻执行SQL校验
    except Exception as e:
        print("Cannot insert another row with existing id")
        print(e)

方案2：将整个with块包裹在try-except中 #

因为commit阶段的异常会从整个with语句向外抛出，直接在外层捕获即可：

try:
    with session_scope(session_cls=Session) as session:
        session.add(label2)
except Exception as e:
    print("Cannot insert another row with existing id")
    print(e)

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