bugfix. wrong tokens is stuck in cache for good. (#1484)

* fix [BUG] #1407 * Update typings.py remove duplicate import * bugfix. wrong tokens is stuck in cache for good. --------- Co-authored-by: vgdh <info@we4.ru>
acheong08 · Aug 2, 2023 · 54cea93 · 54cea93
1 parent 36d1c8f
commit 54cea93
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/revChatGPT/V1.py b/src/revChatGPT/V1.py
@@ -427,13 +427,27 @@ def __get_cached_access_token(self, email: str | None) -> str | None:
                 d_access_token = base64.b64decode(s_access_token[1])
                 d_access_token = json.loads(d_access_token)
             except binascii.Error:
+                del cache["access_tokens"][email]
+                self.__write_cache(cache)
                 error = t.Error(
                     source="__get_cached_access_token",
                     message="Invalid access token",
                     code=t.ErrorType.INVALID_ACCESS_TOKEN_ERROR,
                 )
+                del cache["access_tokens"][email]
                 raise error from None
             except json.JSONDecodeError:
+                del cache["access_tokens"][email]
+                self.__write_cache(cache)
+                error = t.Error(
+                    source="__get_cached_access_token",
+                    message="Invalid access token",
+                    code=t.ErrorType.INVALID_ACCESS_TOKEN_ERROR,
+                )
+                raise error from None
+            except IndexError:
+                del cache["access_tokens"][email]
+                self.__write_cache(cache)
                 error = t.Error(
                     source="__get_cached_access_token",
                     message="Invalid access token",