From d27c4ba2c9090b2decc8e09dbe5e685cc1dfd46a Mon Sep 17 00:00:00 2001 From: Robert Aistleitner Date: Thu, 28 Apr 2016 13:27:09 +0200 Subject: [PATCH] little bugfix to avoid exception on non-existing cookie --- flask_sockets.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flask_sockets.py b/flask_sockets.py index aa51fa9..691744b 100644 --- a/flask_sockets.py +++ b/flask_sockets.py @@ -39,7 +39,9 @@ class SocketMiddleware(object): try: handler, values = adapter.match() environment = environ['wsgi.websocket'] - cookie = parse_cookie(environ['HTTP_COOKIE']) + cookie = None + if 'HTTP_COOKIE' in environ: + cookie = parse_cookie(environ['HTTP_COOKIE']) with self.app.app_context(): with self.app.request_context(environ):