From 257096084f872f160e327bf1d2513b6ce3762578 Mon Sep 17 00:00:00 2001 From: Houkime <> Date: Wed, 14 Dec 2022 17:51:51 +0000 Subject: [PATCH] refactor(tokens-repo): split out date field detection --- .../repositories/tokens/redis_tokens_repository.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/selfprivacy_api/repositories/tokens/redis_tokens_repository.py b/selfprivacy_api/repositories/tokens/redis_tokens_repository.py index 8432709..3a23911 100644 --- a/selfprivacy_api/repositories/tokens/redis_tokens_repository.py +++ b/selfprivacy_api/repositories/tokens/redis_tokens_repository.py @@ -87,10 +87,15 @@ class RedisTokensRepository(AbstractTokensRepository): raise NotImplementedError @staticmethod - def _prepare_model_dict(d: dict): - for date in [ + def _is_date_key(key: str): + return key in [ "created_at", - ]: + ] + + @staticmethod + def _prepare_model_dict(d: dict): + date_keys = [key for key in d.keys() if RedisTokensRepository._is_date_key(key)] + for date in date_keys: if d[date] != "None": d[date] = datetime.fromisoformat(d[date]) for key in d.keys():