diff --git a/dj_db_conn_pool/backends/postgresql/base.py b/dj_db_conn_pool/backends/postgresql/base.py index 911d095..bd4b64e 100644 --- a/dj_db_conn_pool/backends/postgresql/base.py +++ b/dj_db_conn_pool/backends/postgresql/base.py @@ -6,10 +6,4 @@ class DatabaseWrapper(PGDatabaseWrapperMixin, base.DatabaseWrapper): - def get_new_connection(self, conn_params): - connection = super().get_new_connection(conn_params) - - if not connection.info: - connection.info = connection.connection.info - - return connection + pass diff --git a/dj_db_conn_pool/backends/postgresql/mixins.py b/dj_db_conn_pool/backends/postgresql/mixins.py index 0199542..d6b557f 100644 --- a/dj_db_conn_pool/backends/postgresql/mixins.py +++ b/dj_db_conn_pool/backends/postgresql/mixins.py @@ -1,10 +1,18 @@ -# coding: utf-8 - -from sqlalchemy.dialects.postgresql.psycopg2 import PGDialect_psycopg2 - -from dj_db_conn_pool.core.mixins import PersistentDatabaseWrapperMixin - - -class PGDatabaseWrapperMixin(PersistentDatabaseWrapperMixin): - class SQLAlchemyDialect(PGDialect_psycopg2): - pass +# coding: utf-8 + +from sqlalchemy.dialects.postgresql.psycopg2 import PGDialect_psycopg2 + +from dj_db_conn_pool.core.mixins import PersistentDatabaseWrapperMixin + + +class PGDatabaseWrapperMixin(PersistentDatabaseWrapperMixin): + class SQLAlchemyDialect(PGDialect_psycopg2): + pass + + def get_new_connection(self, conn_params): + connection = super().get_new_connection(conn_params) + + if not connection.info: + connection.info = connection.connection.info + + return connection