如何使用SQLAlchemy contextmanager仍然获取行ID？

Question

我正在使用SQLAlchemy的provided contextmanager为我处理会话。我不明白的是如何获取自动生成的ID，因为（1）在调用commit()之后才创建ID（2）新创建的实例仅在上下文管理器的范围内可用：

def save_soft_file(name, is_geo=False):
    with session_scope() as session:
        soft_file = models.SoftFile(name=name, is_geo=is_geo)
        session.add(soft_file)
        # id is not available here, because the session has not been committed
    # soft_file is not available here, because the session is out of context
    return soft_file.id

我错过了什么？

Answer 1

使用session.flush()在当前事务中执行挂起的命令。

def save_soft_file(name, is_geo=False):
    with session_scope() as session:
        soft_file = models.SoftFile(name=name, is_geo=is_geo)
        session.add(soft_file)
        session.flush()
        return soft_file.id

如果在flush之后但在会话超出范围之前发生异常，则更改将回滚到事务的开头。在这种情况下，你的soft_file实际上不会被写入数据库，即使它已经被赋予了ID。