def mtgox_websocket_connect(s, proxytype=None,proxy=None):
host='websocket.mtgox.com'
port=80
if proxytype=='http':
s.connect(proxy)
Proxies.do_http_connect(s,host,port)
elif proxytype=='socks4a':
s.connect(proxy)
Proxies.do_socks4a(s,host,port)
else:
s.connect((host,port))
querylines = [
'GET /mtgox HTTP/1.1',
'Upgrade: WebSocket',
'Connection: Upgrade',
'Host: websocket.mtgox.com',
'Origin: null',
]
s.send('\r\n'.join(querylines)+'\r\n\r\n')
lines = []
while True:
line = recv_line(s)
if line=='': break
lines.append(line)
if lines[0]<>'HTTP/1.1 101 Web Socket Protocol Handshake':
raise Exception('server did not say "101" (said %s)'%repr(lines[0]))
return s
