upgrade.protocol might not be set

progrium · progrium · commit d65cd3a5dfb2 · 2011-09-25T01:37:29.000-05:00
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
+build
 *.egg-info
diff --git a/ws4py/server/wsgi/middleware.py b/ws4py/server/wsgi/middleware.py
@@ -64,7 +64,7 @@ def __init__(self, handle, fallback_app=None, protocols=None, extensions=None,
     def __call__(self, environ, start_response):        
         # Initial handshake validation
         try:
-            if 'websocket' not in environ.get('upgrade.protocol'):
+            if 'websocket' not in environ.get('upgrade.protocol', ''):
                 raise HandshakeError("Upgrade protocol is not websocket")
             
             if environ.get('REQUEST_METHOD') != 'GET':
