Повторите попытку в тупике для MySQL/SQLAlchemy

Я искал довольно давно и не могу найти решение моей проблемы. Мы используем SQLAlchemy в сочетании с MySQL для нашего проекта, и мы сталкиваемся с несколькими опасными ошибками:

1213, "Тупик обнаружен при попытке получить блокировку; попробуйте перезапустить транзакцию".

В этом случае мы попытаемся перезапустить транзакцию не более трех раз.

Я начал писать декоратор, который делает это, но я не знаю, как сохранить состояние сеанса перед сбоем и повторить одну и ту же транзакцию после него? (Поскольку SQLAlchemy требует откат при возникновении исключения)

Моя работа до сих пор,

def retry_on_deadlock_decorator(func):
    lock_messages_error = ['Deadlock found', 'Lock wait timeout exceeded']

    @wraps(func)
    def wrapper(*args, **kwargs):
        attempt_count = 0
        while attempt_count < settings.MAXIMUM_RETRY_ON_DEADLOCK:
            try:
                return func(*args, **kwargs)
            except OperationalError as e:
                if any(msg in e.message for msg in lock_messages_error) \
                        and attempt_count <= settings.MAXIMUM_RETRY_ON_DEADLOCK:
                    logger.error('Deadlock detected. Trying sql transaction once more. Attempts count: %s'
                                 % (attempt_count + 1))
                else:
                    raise
            attempt_count += 1
    return wrapper

Ответы

 @retry_on_deadlock_decorator
 def deadlock_safe_execute(db, stmt, *args, **kw):
     return db.execute(stmt, *args, **kw)

 db.execute("UPDATE users SET active=0")

 deadlock_safe_execute(db, "UPDATE users SET active=0")

try: 

     Perform table transaction 
     break 
except: 
     rollback 
     delay 
     try again to perform table transaction