allow connect event handler to send data to client

engineio/server.py

@@ -207,19 +207,22 @@ def handle_request(self, environ, start_response):
     def _handle_connect(self, environ):
         """Handle a client connection request."""
         sid = self.generate_id()
-        if self._trigger_event('connect', sid, environ) is False:
-            self.logger.warning('Application rejected connection')
-            return self._unauthorized()
-        self.clients[sid] = socket.Socket(self, sid)
+        s = socket.Socket(self, sid)
+        self.clients[sid] = s
         pkt = packet.Packet(
             packet.OPEN, {'sid': sid,
                           'upgrades': self._upgrades(sid),
                           'pingTimeout': int(self.ping_timeout * 1000),
                           'pingInterval': int(self.ping_interval * 1000)})
+        s.send(pkt)
+        if self._trigger_event('connect', sid, environ) is False:
+            self.logger.warning('Application rejected connection')
+            del self.clients[sid]
+            return self._unauthorized()
         headers = None
         if self.cookie:
             headers = [('Set-Cookie', self.cookie + '=' + sid)]
-        return self._ok([pkt], headers=headers)
+        return self._ok(s.poll(), headers=headers)
 
     def _upgrades(self, sid):
         """Return the list of possible upgrades for a client connection."""