From ba6a5261fa39e428c76b73394d8a17f263ff8ffb Mon Sep 17 00:00:00 2001 From: Houkime <> Date: Wed, 14 Dec 2022 15:34:25 +0000 Subject: [PATCH] refactor(tokens-repo): redis token key func --- .../repositories/tokens/redis_tokens_repository.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/selfprivacy_api/repositories/tokens/redis_tokens_repository.py b/selfprivacy_api/repositories/tokens/redis_tokens_repository.py index 72a3cec..44c32f3 100644 --- a/selfprivacy_api/repositories/tokens/redis_tokens_repository.py +++ b/selfprivacy_api/repositories/tokens/redis_tokens_repository.py @@ -57,9 +57,13 @@ class RedisTokensRepository(AbstractTokensRepository): """Delete the new device key""" raise NotImplementedError + @staticmethod + def _token_redis_key(token: Token) -> str: + return RedisTokensRepository.token_key_for_device(token.device_name) + def _store_token(self, new_token: Token): """Store a token directly""" - key = RedisTokensRepository.token_key_for_device(new_token.device_name) + key = RedisTokensRepository._token_redis_key(new_token) self._store_token_as_hash(key, new_token) def _decrement_recovery_token(self):