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

refactor(tokens-repo): detach preparing a dict before a model cast

Browse source
This commit is contained in:
Houkime 2022-12-14 17:03:22 +00:00
parent e504585437
commit 3cb7f29593
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -76,18 +76,22 @@ class RedisTokensRepository(AbstractTokensRepository):
"""Retrieves new device key that is already stored.""" """Retrieves new device key that is already stored."""
raise NotImplementedError raise NotImplementedError
@staticmethod
def _prepare_model_dict(d: dict):
for date in [
"created_at",
]:
if d[date] != "None":
d[date] = datetime.fromisoformat(d[date])
for key in d.keys():
if d[key] == "None":
d[key] = None
def _token_from_hash(self, redis_key: str) -> Token: def _token_from_hash(self, redis_key: str) -> Token:
r = self.connection r = self.connection
if r.exists(redis_key): if r.exists(redis_key):
token_dict = r.hgetall(redis_key) token_dict = r.hgetall(redis_key)
for date in [ RedisTokensRepository._prepare_model_dict(token_dict)
"created_at",
]:
if token_dict[date] != "None":
token_dict[date] = datetime.fromisoformat(token_dict[date])
for key in token_dict.keys():
if token_dict[key] == "None":
token_dict[key] = None
return Token(**token_dict) return Token(**token_dict)
return None return None