From 1f87a4d33cab63f13a0b845d6ff8de8b1fb458b0 Mon Sep 17 00:00:00 2001
From: zakkie <zakkie@zakkie.net>
Date: Mon, 25 Feb 2013 00:25:50 +0900
Subject: [PATCH] allow to set HTTP Header to WebSocketApp

---
 websocket.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/websocket.py b/websocket.py
index 0ba0d00..013e38c 100644
--- a/websocket.py
+++ b/websocket.py
@@ -683,7 +683,7 @@ class WebSocketApp(object):
     Higher level of APIs are provided.
     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_close = None, keep_running = True, get_mask_key = None):
         """
@@ -706,6 +706,7 @@ class WebSocketApp(object):
          docstring for more information
         """
         self.url = url
+        self.header = header
         self.on_open = on_open
         self.on_message = on_message
         self.on_error = on_error
@@ -739,7 +740,7 @@ class WebSocketApp(object):
             raise WebSocketException("socket is already opened")
         try:
             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)
             while self.keep_running:
                 data = self.sock.recv()