fix: Catch invalid token

ZiRO-Bot · Sep 16, 2024 · 583b3db · 583b3db
1 parent 1a359d6
commit 583b3db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nexus/core/middleware.py b/nexus/core/middleware.py
@@ -3,7 +3,7 @@
 from base64 import b64decode, b64encode
 from typing import Union
 
-from cryptography.fernet import Fernet
+from cryptography.fernet import Fernet, InvalidToken
 from itsdangerous import BadSignature
 from redis import asyncio as aioredis
 from starlette.datastructures import MutableHeaders
@@ -46,7 +46,7 @@ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
                 scope["__ssid"] = sessionId
                 scope["__sskey"] = sessionKey
                 initial_session_was_empty = False
-            except (BadSignature, AttributeError):
+            except (BadSignature, AttributeError, InvalidToken):
                 scope["session"] = {}
         else:
             scope["session"] = {}