Откат после неудачной транзакции

Question

У меня есть операция, когда родительская модель должна быть создана с некоторым количеством дочерних моделей. Если создание любого из этих экземпляров не выполняется, все это нужно отменить. Родительская модель и ее дети должны не существуют в базе данных при возникновении ошибки.

код у меня есть:

transaction = db.engine.connect().begin() 

try: 
    parent = ParentModel() 
    db.session.add(parent) 
    db.session.commit() 

    child = ChildModel(parent_id=parent.id) 
    db.session.add(child) 
    db.session.commit() 

    # An error occurs. We need to rollback the saved parent model. 
    raise HTTPException() # from werkzeug 
except: 
    transaction.rollback() 

transaction.commit() 

Мой тест:

def test(self): 
    # call the above operation 
    ParentModel.query.filter_by(id=1).first() # returns the parent model 

Answer 1

Колба-SQLAlchemy поворачивает на AUTOCOMMIT по умолчанию. Чтобы отменить транзакцию, вам нужно отключить ее. Вместо

db = SQLAlchemy(app) # or whatever variation you use 

использования

db = SQLAlchemy(app, session_options={'autocommit': False}) 

Это позволит вам добавить несколько объектов db.session, прежде чем либо фиксации или отката.

С помощью этого изменения вы можете удалить ваши ссылки на transaction.