openstack/deb-python-websocket-client
Code Issues Proposed changes

Merge pull request #27 from zakkie/master

Browse Source 
allow to set HTTP Header to WebSocketApp
This commit is contained in:
liris
2013-02-25 16:18:54 -08:00
parent ea03f7026b 1f87a4d33c
commit ae94a8d67c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
websocket.py
View File

@@ -683,7 +683,7 @@ class WebSocketApp(object):
Higher level of APIs are provided. Higher level of APIs are provided.
The interface is like JavaScript WebSocket object. The interface is like JavaScript WebSocket object.
""" """
def __init__(self, url, def __init__(self, url, header = [],
on_open = None, on_message = None, on_error = None, on_open = None, on_message = None, on_error = None,
on_close = None, keep_running = True, get_mask_key = None): on_close = None, keep_running = True, get_mask_key = None):
""" """
@@ -706,6 +706,7 @@ class WebSocketApp(object):
docstring for more information docstring for more information
""" """
self.url = url self.url = url
self.header = header
self.on_open = on_open self.on_open = on_open
self.on_message = on_message self.on_message = on_message
self.on_error = on_error self.on_error = on_error
@@ -739,7 +740,7 @@ class WebSocketApp(object):
raise WebSocketException("socket is already opened") raise WebSocketException("socket is already opened")
try: try:
self.sock = WebSocket(self.get_mask_key) self.sock = WebSocket(self.get_mask_key)
self.sock.connect(self.url) self.sock.connect(self.url, header = self.header)
self._run_with_no_err(self.on_open) self._run_with_no_err(self.on_open)
while self.keep_running: while self.keep_running:
data = self.sock.recv() data = self.sock.recv()