From 7663d9f108edf2580f81342eaaf9fc9a66312ea9 Mon Sep 17 00:00:00 2001 From: Mark McGuire Date: Mon, 17 Feb 2014 21:16:25 +0000 Subject: [PATCH] Add returns for methods --- flask_socketio/__init__.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/flask_socketio/__init__.py b/flask_socketio/__init__.py index a674e11a..56fc92bb 100644 --- a/flask_socketio/__init__.py +++ b/flask_socketio/__init__.py @@ -44,7 +44,7 @@ def process_event(self, packet): message = packet['name'] args = packet['args'] app = self.request - self.socketio.dispatch_message(app, self, message, args) + return self.socketio.dispatch_message(app, self, message, args) def recv_connect(self): ret = super(GenericNamespace, self).recv_connect() @@ -59,11 +59,11 @@ def recv_disconnect(self): def recv_message(self, data): app = self.request - self.socketio.dispatch_message(app, self, 'message', [data]) + return self.socketio.dispatch_message(app, self, 'message', [data]) def recv_json(self, data): app = self.request - self.socketio.dispatch_message(app, self, 'json', [data]) + return self.socketio.dispatch_message(app, self, 'json', [data]) def emit(self, event, *args, **kwargs): ns_name = kwargs.pop('namespace', None) @@ -114,9 +114,10 @@ def dispatch_message(self, app, namespace, message, args=[]): request.namespace = namespace for k, v in namespace.session.items(): session[k] = v - self.messages[namespace.ns_name][message](*args) + ret = self.messages[namespace.ns_name][message](*args) for k, v in session.items(): namespace.session[k] = v + return ret def on_message(self, message, handler, **options): ns_name = options.pop('namespace', '')