From 5a25e2a2706ef2492e225ec1d0777c9944dd0152 Mon Sep 17 00:00:00 2001 From: Houkime <> Date: Wed, 14 Dec 2022 18:55:26 +0000 Subject: [PATCH] feat(tokens-repo): getting stored device key --- .../repositories/tokens/redis_tokens_repository.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/selfprivacy_api/repositories/tokens/redis_tokens_repository.py b/selfprivacy_api/repositories/tokens/redis_tokens_repository.py index 833679f..b1fb4b0 100644 --- a/selfprivacy_api/repositories/tokens/redis_tokens_repository.py +++ b/selfprivacy_api/repositories/tokens/redis_tokens_repository.py @@ -97,7 +97,7 @@ class RedisTokensRepository(AbstractTokensRepository): def _get_stored_new_device_key(self) -> Optional[NewDeviceKey]: """Retrieves new device key that is already stored.""" - raise NotImplementedError + return self._new_device_key_from_hash(NEW_DEVICE_KEY_REDIS_KEY) @staticmethod def _is_date_key(key: str): @@ -136,6 +136,9 @@ class RedisTokensRepository(AbstractTokensRepository): def _recovery_key_from_hash(self, redis_key: str) -> Optional[RecoveryKey]: return self._hash_as_model(redis_key, RecoveryKey) + def _new_device_key_from_hash(self, redis_key: str) -> Optional[NewDeviceKey]: + return self._hash_as_model(redis_key, NewDeviceKey) + def _store_model_as_hash(self, redis_key, model): r = self.connection for key, value in model.dict().items():