bugfix in websocket

ytjohn · Jun 16, 2012 · a16c1b7 · a16c1b7
1 parent 02a4e64
commit a16c1b7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cyclone/websocket.py b/cyclone/websocket.py
@@ -70,6 +70,9 @@ def _execute(self, transforms, *args, **kwargs):
         except:
             return self.forbidConnection("Expected WebSocket Headers")
 
+        self._connectionMade = functools.partial(self.connectionMade,
+                                                 *args, **kwargs)
+
         if "Sec-Websocket-Version" in self.request.headers and \
             self.request.headers['Sec-Websocket-Version'] in ('7', '8', '13'):
             self.ws_protocol = WebSocketProtocol17(self)
@@ -85,8 +88,6 @@ def _execute(self, transforms, *args, **kwargs):
         self.request.connection.rawDataReceived = \
             self.ws_protocol.rawDataReceived
         self.ws_protocol.acceptConnection()
-        self._connectionMade = functools.partial(self.connectionMade,
-                                                 *args, **kwargs)
 
     def forbidConnection(self, message):
         self.transport.write(