Commit ad32a2ef authored by Беклемишев Петр Алексеевич's avatar Беклемишев Петр Алексеевич
Browse files

Drop mysql restarts

parent e59e1f50
from contextvars import ContextVar from contextvars import ContextVar
from peewee import PostgresqlDatabase, InterfaceError as PeeWeeInterfaceError, MySQLDatabase, _ConnectionState from peewee import PostgresqlDatabase, InterfaceError as PeeWeeInterfaceError, _ConnectionState
from playhouse.db_url import register_database from playhouse.db_url import register_database
from playhouse.pool import PooledPostgresqlDatabase, PooledMySQLDatabase from playhouse.pool import PooledPostgresqlDatabase
from playhouse.shortcuts import ReconnectMixin from playhouse.shortcuts import ReconnectMixin
from psycopg2 import OperationalError from psycopg2 import OperationalError
from psycopg2.errors import InterfaceError from psycopg2.errors import InterfaceError
...@@ -26,7 +26,7 @@ class PeeweeConnectionState(_ConnectionState): ...@@ -26,7 +26,7 @@ class PeeweeConnectionState(_ConnectionState):
class CustomReconnectMixin(ReconnectMixin): class CustomReconnectMixin(ReconnectMixin):
reconnect_errors = ( reconnect_errors = (
# default ReconnectMixin exceptions (MySQL specific) # default ReconnectMixin exceptions
*ReconnectMixin.reconnect_errors, *ReconnectMixin.reconnect_errors,
# psycopg2 # psycopg2
(OperationalError, 'termin'), (OperationalError, 'termin'),
...@@ -44,16 +44,6 @@ class ReconnectingPooledPostgresqlDatabase(CustomReconnectMixin, PooledPostgresq ...@@ -44,16 +44,6 @@ class ReconnectingPooledPostgresqlDatabase(CustomReconnectMixin, PooledPostgresq
pass pass
class ReconnectingMySQLDatabase(CustomReconnectMixin, MySQLDatabase):
pass
class ReconnectingPooledMySQLDatabase(CustomReconnectMixin, PooledMySQLDatabase):
pass
def register_peewee_databases(): def register_peewee_databases():
register_database(MySQLDatabase, 'mysql')
register_database(PooledMySQLDatabase, 'mysql+pool')
register_database(ReconnectingPostgresqlDatabase, 'postgres', 'postgresql') register_database(ReconnectingPostgresqlDatabase, 'postgres', 'postgresql')
register_database(ReconnectingPooledPostgresqlDatabase, 'postgres+pool', 'postgresql+pool') register_database(ReconnectingPooledPostgresqlDatabase, 'postgres+pool', 'postgresql+pool')
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment