return tokens not dicts

This commit is contained in:
2022-04-23 15:07:14 +02:00
parent f5a78e8b05
commit 04ccc4e9ae

View File

@@ -54,7 +54,7 @@ class TokenStoreOnDisk(Token):
if not TOKEN_PATH.exists():
raise NoToken
with TOKEN_PATH.open(encoding="utf-8") as fin:
return json.load(fin)
return Token(**json.load(fin))
@classmethod
def set(cls, token_dict: dict) -> None:
@@ -76,7 +76,7 @@ class TokenStoreOnRedis(Token):
result = self.redis_client.get(TOKEN_KEY)
if result is None:
raise NoToken
return json.loads(result)
return Token(**json.loads(result))
def set(self, token_dict: dict) -> None:
self.redis_client.set(TOKEN_KEY, json.dumps(token_dict))