Make gevent websocket optional in example app

miguelgrinberg · Aug 20, 2015 · c6b1ae9 · c6b1ae9
1 parent 48eb073
commit c6b1ae9
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/example/app.py b/example/app.py
@@ -42,6 +42,15 @@ def disconnect(sid):
     elif async_mode == 'gevent':
         # deploy with gevent
         from gevent import pywsgi
-        from geventwebsocket.handler import WebSocketHandler
-        pywsgi.WSGIServer(('', 5000), app,
-                          handler_class=WebSocketHandler).serve_forever()
+        try:
+            from geventwebsocket.handler import WebSocketHandler
+            websocket = True
+        except ImportError:
+            websocket = False
+        if websocket:
+            pywsgi.WSGIServer(('', 5000), app,
+                              handler_class=WebSocketHandler).serve_forever()
+        else:
+            pywsgi.WSGIServer(('', 5000), app).serve_forever()
+    else:
+        print('Unknown async_mode: ' + async_mode)