Skip to content

Commit

Permalink
fixes #14: access to server object when using gunicorn
Browse files Browse the repository at this point in the history
  • Loading branch information
miguelgrinberg committed Mar 31, 2014
1 parent d4e0a36 commit 1fb766c
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions flask_socketio/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ def __init__(self, app=None):
self.init_app(app)
self.messages = {}
self.rooms = {}
self.server = None

def init_app(self, app):
app.wsgi_app = SocketIOMiddleware(app, self)
Expand Down Expand Up @@ -59,6 +60,8 @@ def leave_room(self, room):
self.rooms.remove(room)

def recv_connect(self):
if self.socketio.server is None:
self.socketio.server = self.environ['socketio'].server
ret = super(GenericNamespace, self).recv_connect()
app = self.request
self.socketio._dispatch_message(app, self, 'connect')
Expand Down Expand Up @@ -164,9 +167,10 @@ def emit(self, event, *args, **kwargs):
for client in self.rooms.get(ns_name, {}).get(room, set()):
client.base_emit(event, *args, **kwargs)
else:
for sessid, socket in self.server.sockets.items():
if socket.active_ns.get(ns_name):
socket[ns_name].base_emit(event, *args, **kwargs)
if self.server:
for sessid, socket in self.server.sockets.items():
if socket.active_ns.get(ns_name):
socket[ns_name].base_emit(event, *args, **kwargs)

def send(self, message, json=False, namespace=None, room=None):
ns_name = namespace
Expand All @@ -176,9 +180,10 @@ def send(self, message, json=False, namespace=None, room=None):
for client in self.rooms.get(ns_name, {}).get(room, set()):
client.base_send(message, json)
else:
for sessid, socket in self.server.sockets.items():
if socket.active_ns.get(ns_name):
socket[ns_name].base_send(message, json)
if self.server:
for sessid, socket in self.server.sockets.items():
if socket.active_ns.get(ns_name):
socket[ns_name].base_send(message, json)

def run(self, app, host=None, port=None, **kwargs):
if host is None:
Expand Down

0 comments on commit 1fb766c

Please sign in to comment.