diff --git a/backend/apps/webui/internal/wrappers.py b/backend/apps/webui/internal/wrappers.py index 406599b5a..53869b916 100644 --- a/backend/apps/webui/internal/wrappers.py +++ b/backend/apps/webui/internal/wrappers.py @@ -1,8 +1,8 @@ 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.pool import PooledPostgresqlDatabase, PooledMySQLDatabase +from playhouse.pool import PooledPostgresqlDatabase from playhouse.shortcuts import ReconnectMixin from psycopg2 import OperationalError from psycopg2.errors import InterfaceError @@ -26,7 +26,7 @@ class PeeweeConnectionState(_ConnectionState): class CustomReconnectMixin(ReconnectMixin): reconnect_errors = ( - # default ReconnectMixin exceptions (MySQL specific) + # default ReconnectMixin exceptions *ReconnectMixin.reconnect_errors, # psycopg2 (OperationalError, 'termin'), @@ -44,16 +44,6 @@ class ReconnectingPooledPostgresqlDatabase(CustomReconnectMixin, PooledPostgresq pass -class ReconnectingMySQLDatabase(CustomReconnectMixin, MySQLDatabase): - pass - - -class ReconnectingPooledMySQLDatabase(CustomReconnectMixin, PooledMySQLDatabase): - pass - - def register_peewee_databases(): - register_database(MySQLDatabase, 'mysql') - register_database(PooledMySQLDatabase, 'mysql+pool') register_database(ReconnectingPostgresqlDatabase, 'postgres', 'postgresql') register_database(ReconnectingPooledPostgresqlDatabase, 'postgres+pool', 'postgresql+pool')