SelfPrivacy
/
selfprivacy-rest-api
7
2
Fork 1
Code Issues 15 Pull Requests 4 Projects Releases 1 Wiki Activity

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

pull/26/head
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 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)
if token_dict is not None:
return Token(**token_dict)
return model_class(**token_dict)
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]:
token_dict = self._model_dict_from_hash(redis_key)
if token_dict is not None:
return RecoveryKey(**token_dict)
return None
return self._hash_as_model(redis_key, RecoveryKey)
def _store_model_as_hash(self, redis_key, model):
r = self.connection