From 557d19ecd48c38cfccb02111137c8d54e14db041 Mon Sep 17 00:00:00 2001
From: Altair Bow <altair.bow@foxmail.com>
Date: Sat, 14 Oct 2023 18:49:27 +0800
Subject: [PATCH] postgres: replace `connection.info` with
 `connection.connection.info`

---
 dj_db_conn_pool/backends/postgresql/base.py   |  8 +-----
 dj_db_conn_pool/backends/postgresql/mixins.py | 28 ++++++++++++-------
 2 files changed, 19 insertions(+), 17 deletions(-)

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