From b21c8f7a4e4c64795e8a10c30f68531ad1d9580f Mon Sep 17 00:00:00 2001 From: Oleksandr Shulgin Date: Tue, 8 Mar 2016 18:34:22 +0100 Subject: Move replication-related imports to extras.py --- lib/extensions.py | 2 -- lib/extras.py | 8 ++++---- 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): -- cgit v1.2.1