SelfPrivacy/selfprivacy-rest-api
1
0
Fork 0
mirror of https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git synced 2024-11-18 08:29:14 +00:00
Code Issues Projects Releases Wiki Activity

feat(tokens-repo): redis store token

Browse source
This commit is contained in:
Houkime 2022-12-14 15:21:32 +00:00
parent 256c16fa9f
commit d8e3cd67e0
1 changed files with 10 additions and 2 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

@ -23,6 +23,7 @@ class RedisTokensRepository(AbstractTokensRepository):
def __init__(self): def __init__(self):
self.connection = RedisPool().get_connection() self.connection = RedisPool().get_connection()
@staticmethod
def token_key_for_device(device_name: str): def token_key_for_device(device_name: str):
return TOKENS_PREFIX + str(hash(device_name)) return TOKENS_PREFIX + str(hash(device_name))
@ -58,7 +59,8 @@ 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"""
raise NotImplementedError key = RedisTokensRepository.token_key_for_device(new_token.device_name)
self._store_token_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"""
@ -76,7 +78,7 @@ class RedisTokensRepository(AbstractTokensRepository):
"created_at", "created_at",
]: ]:
if token_dict[date] != "None": if token_dict[date] != "None":
token_dict[date] = datetime.datetime.fromisoformat(token_dict[date]) token_dict[date] = datetime.fromisoformat(token_dict[date])
for key in token_dict.keys(): for key in token_dict.keys():
if token_dict[key] == "None": if token_dict[key] == "None":
token_dict[key] = None token_dict[key] = None
@ -84,3 +86,9 @@ class RedisTokensRepository(AbstractTokensRepository):
return Token(**token_dict) return Token(**token_dict)
return None return None
def _store_token_as_hash(self, redis_key, model):
r = self.connection
for key, value in model.dict().items():
if isinstance(value, datetime):
value = value.isoformat()
r.hset(redis_key, key, str(value))