SelfPrivacy/selfprivacy-rest-api
1
0
Fork 0
mirror of https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git synced 2024-11-15 23:23:17 +00:00
Code Issues Projects Releases Wiki Activity

refactor(tokens-repo): break out generic hash_as_model casting

Browse source
This commit is contained in:
Houkime 2022-12-14 18:45:12 +00:00
parent 6f6a9f5ef0
commit 0ae7c43ebf
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
selfprivacy_api/repositories/tokens/redis_tokens_repository.py
View file

@ -124,17 +124,17 @@ class RedisTokensRepository(AbstractTokensRepository):
return token_dict return token_dict
return None return None
def _token_from_hash(self, redis_key: str) -> Optional[Token]: def _hash_as_model(self, redis_key: str, model_class):
token_dict = self._model_dict_from_hash(redis_key) token_dict = self._model_dict_from_hash(redis_key)
if token_dict is not None: if token_dict is not None:
return Token(**token_dict) return model_class(**token_dict)
return None return None
def _token_from_hash(self, redis_key: str) -> Optional[Token]:
return self._hash_as_model(redis_key, Token)
def _recovery_key_from_hash(self, redis_key: str) -> Optional[RecoveryKey]: def _recovery_key_from_hash(self, redis_key: str) -> Optional[RecoveryKey]:
token_dict = self._model_dict_from_hash(redis_key) return self._hash_as_model(redis_key, RecoveryKey)
if token_dict is not None:
return RecoveryKey(**token_dict)
return None
def _store_model_as_hash(self, redis_key, model): def _store_model_as_hash(self, redis_key, model):
r = self.connection r = self.connection