raise NoResult in redis token store

2022-04-23 14:51:53 +02:00
parent 7636483346
commit 50ecb9d042
1 changed files with 3 additions and 1 deletions
--- a/avt_fresh/token.py
+++ b/avt_fresh/token.py
@@ -73,7 +73,9 @@ class TokenStoreOnRedis(Token):
        self.redis_client = redis.from_url(redis_url)

    def get(self):
-        return self.redis_client.get(TOKEN_KEY)
+        result = self.redis_client.get(TOKEN_KEY)
+        if result is None:
+            raise NoResult

    def set(self, token_dict: dict) -> None:
        self.redis_client.set(TOKEN_KEY, json.dumps(token_dict))