Raise a runtime error when gevent-websocket's custom server is not used

miguelgrinberg · Oct 7, 2015 · e38cad9 · e38cad9
1 parent 895e1bd
commit e38cad9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/engineio/async_gevent.py b/engineio/async_gevent.py
@@ -30,8 +30,11 @@ def __init__(self, app):
         self.app = app
 
     def __call__(self, environ, start_response):
-
-        self._sock = environ["wsgi.websocket"]
+        if 'wsgi.websocket' not in environ:
+            raise RuntimeError('You need to use the gevent-websocket server. '
+                               'See the Deployment section of the '
+                               'documentation for more information.')
+        self._sock = environ['wsgi.websocket']
         self.environ = environ
         self.version = self._sock.version
         self.path = self._sock.path