A replacement for playhouse.shortcuts.RetryOperationalError?

Commit 54a672547f549eb381cd9578a2b19310bcf5ef1d removed the class playhouse.shortcuts.RetryOperationalError. Is there a new recommended way to automatically reconnect and retry on failed queries?

[gzxultra]

This should work if you have legacy code.

from peewee import MySQLDatabase
from peewee import OperationalError


class RetryOperationalError(object):

    def execute_sql(self, sql, params=None, commit=True):
        try:
            cursor = super(RetryOperationalError, self).execute_sql(
                sql, params, commit)
        except OperationalError:
            if not self.is_closed():
                self.close()
            with __exception_wrapper__:
                cursor = self.cursor()
                cursor.execute(sql, params or ())
                if commit and not self.in_transaction():
                    self.commit()
        return cursor


class MyRetryDB(RetryOperationalError, MySQLDatabase):
    pass

[coleifer]

Small note:

__exception_wrapper__ should not have parentheses afterwards – you don’t need to call it. I went ahead and edited your code snippet to make the change.