From 224d0570f55b8492be6df84cf1d5f12ac820c96b Mon Sep 17 00:00:00 2001
From: thec0sm0s <deepakrajko14@gmail.com>
Date: Thu, 14 May 2020 21:01:10 +0530
Subject: [PATCH] Add DISCORD_USER_ID in SESSION_KEYS. Try removing user from
 cache if exists

---
 flask_discord/_http.py  | 1 +
 flask_discord/client.py | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/flask_discord/_http.py b/flask_discord/_http.py
index cc63608..697f40c 100644
--- a/flask_discord/_http.py
+++ b/flask_discord/_http.py
@@ -34,6 +34,7 @@ class DiscordOAuth2HttpClient(abc.ABC):
     """
 
     SESSION_KEYS = [
+        "DISCORD_USER_ID",
         "DISCORD_OAUTH2_STATE",
         "DISCORD_OAUTH2_TOKEN",
     ]
diff --git a/flask_discord/client.py b/flask_discord/client.py
index 8ad31ff..0d029af 100644
--- a/flask_discord/client.py
+++ b/flask_discord/client.py
@@ -61,6 +61,10 @@ class DiscordOAuth2Session(_http.DiscordOAuth2HttpClient):
         to go through discord authorization token grant flow again.
 
         """
+
+        user_id = session.get("DISCORD_USER_ID")
+        self.users_cache.pop(user_id, None)
+
         for session_key in self.SESSION_KEYS:
             try:
                 session.pop(session_key)