Add Connection.create_transport method

This makes it possible to extend Connection class and use a customized
transport.

1 files changed, 5 insertions, 2 deletions

diff --git a/amqp/connection.py b/amqp/connection.py
index c93d91f..f5fc127 100644
--- a/amqp/connection.py
+++ b/amqp/connection.py
@@ -37,7 +37,7 @@ from .exceptions import (
 from .five import items, range, values, monotonic
 from .method_framing import MethodReader, MethodWriter
 from .serialization import AMQPWriter
-from .transport import create_transport
+from . import transport
 
 HAS_MSG_PEEK = hasattr(socket, 'MSG_PEEK')
 
@@ -162,7 +162,7 @@ class Connection(AbstractChannel):
         # Let the transport.py module setup the actual
         # socket connection to the broker.
         #
-        self.transport = create_transport(host, connect_timeout, ssl)
+        self.transport = self.create_transport(host, connect_timeout, ssl)
 
         self.method_reader = MethodReader(self.transport)
         self.method_writer = MethodWriter(self.transport, self.frame_max)
@@ -182,6 +182,9 @@ class Connection(AbstractChannel):
 
         return self._x_open(virtual_host)
 
+    def create_transport(self, host, connect_timeout, ssl=False):
+        return transport.create_transport(host, connect_timeout, ssl)
+
     @property
     def connected(self):
         return self.transport and self.transport.connected