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

feat(tokens-repo): get new device key

Browse source
This commit is contained in:
Houkime 2022-12-14 17:20:09 +00:00
parent b98ccb88d1
commit 9ffd67fa19
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -13,6 +13,7 @@ from selfprivacy_api.models.tokens.recovery_key import RecoveryKey
from selfprivacy_api.models.tokens.new_device_key import NewDeviceKey from selfprivacy_api.models.tokens.new_device_key import NewDeviceKey
TOKENS_PREFIX = "token_repo:tokens:" TOKENS_PREFIX = "token_repo:tokens:"
NEW_DEVICE_KEY_REDIS_KEY = "token_repo:new_device_key"
class RedisTokensRepository(AbstractTokensRepository): class RedisTokensRepository(AbstractTokensRepository):
@ -53,7 +54,9 @@ class RedisTokensRepository(AbstractTokensRepository):
def get_new_device_key(self) -> NewDeviceKey: def get_new_device_key(self) -> NewDeviceKey:
"""Creates and returns the new device key""" """Creates and returns the new device key"""
raise NotImplementedError new_device_key = NewDeviceKey.generate()
self._store_model_as_hash(NEW_DEVICE_KEY_REDIS_KEY, new_device_key)
return NewDeviceKey
def delete_new_device_key(self) -> None: def delete_new_device_key(self) -> None:
"""Delete the new device key""" """Delete the new device key"""
@ -66,7 +69,7 @@ class RedisTokensRepository(AbstractTokensRepository):
def _store_token(self, new_token: Token): def _store_token(self, new_token: Token):
"""Store a token directly""" """Store a token directly"""
key = RedisTokensRepository._token_redis_key(new_token) key = RedisTokensRepository._token_redis_key(new_token)
self._store_token_as_hash(key, new_token) self._store_model_as_hash(key, new_token)
def _decrement_recovery_token(self): def _decrement_recovery_token(self):
"""Decrement recovery key use count by one""" """Decrement recovery key use count by one"""
@ -101,7 +104,7 @@ class RedisTokensRepository(AbstractTokensRepository):
return Token(**token_dict) return Token(**token_dict)
return None return None
def _store_token_as_hash(self, redis_key, model): def _store_model_as_hash(self, redis_key, model):
r = self.connection r = self.connection
for key, value in model.dict().items(): for key, value in model.dict().items():
if isinstance(value, datetime): if isinstance(value, datetime):