Merge pull request #4 from noirbizarre/werkzeug-debug

Werkzeug debugger and reloader support

docs/index.rst

@@ -22,15 +22,15 @@ The following code example shows how to add Flask-SocketIO to a Flask applicatio
 
     from flask import Flask, render_template
     from flask.ext.socketio import SocketIO
-    
+
     app = Flask(__name__)
     app.config['SECRET_KEY'] = 'secret!'
     socketio = SocketIO(app)
-    
+
     if __name__ == '__main__':
         socketio.run(app)
 
-The ``init_app()`` style of initialization is also supported. Note the way the web server is started. The ``socketio.run()`` function encapsulates the start up of the gevent web server. When using this extension the Werkzeug server cannot be used.
+The ``init_app()`` style of initialization is also supported. Note the way the web server is started. The ``socketio.run()`` function encapsulates the start up of the gevent web server.
 
 Receiving Messages
 ------------------

example/app.py

@@ -2,6 +2,7 @@
 from flask.ext.socketio import SocketIO, emit
 
 app = Flask(__name__)
+app.debug=True
 app.config['SECRET_KEY'] = 'secret!'
 socketio = SocketIO(app)
 

flask_socketio.py

@@ -5,13 +5,17 @@
 from socketio.namespace import BaseNamespace
 from flask import request, session
 from flask.ctx import RequestContext
+from werkzeug.debug import DebuggedApplication
+from werkzeug.serving import run_with_reloader
 
 monkey.patch_all()
 
 
 class SocketIOMiddleware(object):
     def __init__(self, app, socket):
         self.app = app
+        if app.debug:
+            app.wsgi_app = DebuggedApplication(app.wsgi_app, evalex=True)
         self.wsgi_app = app.wsgi_app
         self.socket = socket
 
@@ -142,7 +146,14 @@ def run(self, app, host=None, port=None):
                 port = int(server_name.rsplit(':', 1)[1])
             else:
                 port = 5000
-        SocketIOServer((host, port), app.wsgi_app, resource='socket.io').serve_forever()
+        if app.debug:
+            @run_with_reloader
+            def run_server():
+                server = SocketIOServer((host, port), app.wsgi_app, resource='socket.io')
+                server.serve_forever()
+            run_server()
+        else:
+            SocketIOServer((host, port), app.wsgi_app, resource='socket.io').serve_forever()
 
 
 def emit(event, *args, **kwargs):