refactor(tokens-repo): split out date field detection

2022-12-14 17:51:51 +00:00 · 2022-12-14 17:51:51 +00:00 · 257096084f
parent bf6c230ae0
commit 257096084f
1 changed files with 8 additions and 3 deletions
--- 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():