[IMPR] Guess the last needed token key if the token is not found

Bug: T334288
Change-Id: I0514f460384a736d583849078f813c14c72bc129
---
M pywikibot/site/_tokenwallet.py
1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/pywikibot/site/_tokenwallet.py b/pywikibot/site/_tokenwallet.py
index 5ffd72f..0d7ec5a 100644
--- a/pywikibot/site/_tokenwallet.py
+++ b/pywikibot/site/_tokenwallet.py
@@ -29,6 +29,8 @@
         self.site: APISite = site
         self._tokens: Dict[str, str] = {}
         self._currentuser: Optional[str] = site.user()
+        # guess the needed token in update_tokens
+        self._last_token_key: Optional[str] = None
 
     def __getitem__(self, key: str) -> str:
         """Get token value for the given key."""
@@ -58,6 +60,7 @@
                 f'Invalid token {key!r} for user {self._currentuser!r} on '
                 f'{self.site} wiki.') from None
 
+        self._last_token_key = key
         return token
 
     def __contains__(self, key) -> bool:
@@ -119,7 +122,7 @@
         # find the token types
         types = [key
                  for key, value in self._tokens.items() for token in tokens
-                 if value == token]
+                 if value == token] or [self._last_token_key]
         self.clear()  # clear the cache
         return [self[token_type] for token_type in types]
 

