From e77efde50344e3719c600edac622a799b5ce4bd3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gina=20H=C3=A4u=C3=9Fge?= Date: Fri, 29 Aug 2014 09:26:57 +0200 Subject: [PATCH] Make key removal from session more solid Do not die a horrible death if the key you are attempting to remove from the session upon logout doesn't exist. Attempt at fixing #556 --- src/octoprint/server/api/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/octoprint/server/api/__init__.py b/src/octoprint/server/api/__init__.py index de3cf188..48ed69fd 100644 --- a/src/octoprint/server/api/__init__.py +++ b/src/octoprint/server/api/__init__.py @@ -235,8 +235,9 @@ def login(): @restricted_access def logout(): # Remove session keys set by Flask-Principal - for key in ('identity.id', 'identity.auth_type'): - del session[key] + for key in ('identity.id', 'identity.name', 'identity.auth_type'): + if key in session: + del session[key] identity_changed.send(current_app._get_current_object(), identity=AnonymousIdentity()) logout_user()