fixes #6: save session variables correctly

example/app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template
+from flask import Flask, render_template, session
 from flask.ext.socketio import SocketIO, emit
 
 app = Flask(__name__)
@@ -14,17 +14,22 @@ def index():
 
 @socketio.on('my event', namespace='/test')
 def test_message(message):
-    emit('my response', {'data': message['data']})
+    session['receive_count'] = session.get('receive_count', 0) + 1
+    emit('my response',
+         {'data': message['data'], 'count': session['receive_count']})
 
 
 @socketio.on('my broadcast event', namespace='/test')
 def test_message(message):
-    emit('my response', {'data': message['data']}, broadcast=True)
+    session['receive_count'] = session.get('receive_count', 0) + 1
+    emit('my response',
+         {'data': message['data'], 'count': session['receive_count']},
+         broadcast=True)
 
 
 @socketio.on('connect', namespace='/test')
 def test_connect():
-    emit('my response', {'data': 'Connected'})
+    emit('my response', {'data': 'Connected', 'count': 0})
 
 
 @socketio.on('disconnect', namespace='/test')

example/templates/index.html

@@ -8,7 +8,7 @@
         $(document).ready(function(){
             var socket = io.connect('http://' + document.domain + ':' + location.port + '/test');
             socket.on('my response', function(msg) {
-                $('#log').append('<p>Received: ' + msg.data + '</p>');
+                $('#log').append('<p>Received #' + msg.count + ': ' + msg.data + '</p>');
             });
             $('form#emit').submit(function(event) {
                 socket.emit('my event', {data: $('#emit_data').val()});

flask_socketio.py

@@ -123,7 +123,7 @@ def dispatch_message(self, app, namespace, message, args=[]):
                     session[k] = v
                 self.messages[namespace.ns_name][message](*args)
                 for k, v in session.items():
-                    self.namespace[k] = v
+                    namespace.session[k] = v
 
     def on_message(self, message, handler, **options):
         namespace = options.pop('namespace', '/')

setup.py

@@ -9,7 +9,7 @@
 
 setup(
     name='Flask-SocketIO',
-    version='0.2.0',
+    version='0.2.1',
     url='http://github.com/miguelgrinberg/Flask-SocketIO/',
     license='MIT',
     author='Miguel Grinberg',