Move creation of websocket connection to its own method, so subclasses can override

This commit is contained in:
Aanand Prasad 2013-09-05 22:46:24 -04:00
parent c2d867b117
commit 6e659ba12a
1 changed files with 4 additions and 2 deletions

View File

@ -155,8 +155,10 @@ class Client(requests.Session):
url = self._url("/containers/{0}/attach/ws".format(container)) url = self._url("/containers/{0}/attach/ws".format(container))
req = requests.Request("POST", url, params=self._attach_params(params)) req = requests.Request("POST", url, params=self._attach_params(params))
full_url = req.prepare().url.replace("http://", "ws://", 1) full_url = req.prepare().url.replace("http://", "ws://", 1)
print full_url return self._create_websocket_connection(full_url)
return websocket.create_connection(full_url)
def _create_websocket_connection(self, url):
return websocket.create_connection(url)
def attach(self, container): def attach(self, container):
socket = self.attach_socket(container) socket = self.attach_socket(container)