Move replication-related imports to extras.py

2 files changed, 4 insertions, 6 deletions

diff --git a/lib/extensions.py b/lib/extensions.py
index 65792fc..2130098 100644
--- a/lib/extensions.py
+++ b/lib/extensions.py
@@ -65,8 +65,6 @@ from psycopg2._psycopg import string_types, binary_types, new_type, new_array_ty
 from psycopg2._psycopg import ISQLQuote, Notify, Diagnostics, Column
 
 from psycopg2._psycopg import QueryCanceledError, TransactionRollbackError
-from psycopg2._psycopg import REPLICATION_PHYSICAL, REPLICATION_LOGICAL
-from psycopg2._psycopg import ReplicationConnection, ReplicationCursor, ReplicationMessage
 
 try:
     from psycopg2._psycopg import set_wait_callback, get_wait_callback
diff --git a/lib/extras.py b/lib/extras.py
index 7845223..6ae9851 100644
--- a/lib/extras.py
+++ b/lib/extras.py
@@ -39,12 +39,12 @@ import psycopg2
 from psycopg2 import extensions as _ext
 from psycopg2.extensions import cursor as _cursor
 from psycopg2.extensions import connection as _connection
-from psycopg2.extensions import REPLICATION_PHYSICAL, REPLICATION_LOGICAL
-from psycopg2.extensions import ReplicationConnection as _replicationConnection
-from psycopg2.extensions import ReplicationCursor as _replicationCursor
-from psycopg2.extensions import ReplicationMessage
 from psycopg2.extensions import adapt as _A, quote_ident
 from psycopg2.extensions import b
+from psycopg2._psycopg import REPLICATION_PHYSICAL, REPLICATION_LOGICAL
+from psycopg2._psycopg import ReplicationConnection as _replicationConnection
+from psycopg2._psycopg import ReplicationCursor as _replicationCursor
+from psycopg2._psycopg import ReplicationMessage
 
 
 class DictCursorBase(_cursor):