Use python-defined make_dsn() for ReplicationConnection class

1 files changed, 2 insertions, 13 deletions

diff --git a/lib/extras.py b/lib/extras.py
index 6e815d6..7845223 100644
--- a/lib/extras.py
+++ b/lib/extras.py
@@ -441,25 +441,14 @@ class MinTimeLoggingCursor(LoggingCursor):
         return LoggingCursor.callproc(self, procname, vars)
 
 
-class ReplicationConnectionBase(_replicationConnection):
-    """
-    Base class for Logical and Physical replication connection
-    classes.  Uses `ReplicationCursor` automatically.
-    """
-
-    def __init__(self, *args, **kwargs):
-        super(ReplicationConnectionBase, self).__init__(*args, **kwargs)
-        self.cursor_factory = ReplicationCursor
-
-
-class LogicalReplicationConnection(ReplicationConnectionBase):
+class LogicalReplicationConnection(_replicationConnection):
 
     def __init__(self, *args, **kwargs):
         kwargs['replication_type'] = REPLICATION_LOGICAL
         super(LogicalReplicationConnection, self).__init__(*args, **kwargs)
 
 
-class PhysicalReplicationConnection(ReplicationConnectionBase):
+class PhysicalReplicationConnection(_replicationConnection):
 
     def __init__(self, *args, **kwargs):
         kwargs['replication_type'] = REPLICATION_PHYSICAL