From 8791462f87c4a134f5ccb7099a814892c110d43b Mon Sep 17 00:00:00 2001
From: Inex Code <inex.code@selfprivacy.org>
Date: Thu, 23 Nov 2023 16:15:53 +0300
Subject: [PATCH] fix: Migration from secrets to token

---
 selfprivacy_api/migrations/write_token_to_redis.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/selfprivacy_api/migrations/write_token_to_redis.py b/selfprivacy_api/migrations/write_token_to_redis.py
index e83eaf1..55609f6 100644
--- a/selfprivacy_api/migrations/write_token_to_redis.py
+++ b/selfprivacy_api/migrations/write_token_to_redis.py
@@ -1,4 +1,7 @@
+from datetime import datetime
+from typing import Optional
 from selfprivacy_api.migrations.migration import Migration
+from selfprivacy_api.models.tokens.token import Token
 
 from selfprivacy_api.repositories.tokens.redis_tokens_repository import (
     RedisTokensRepository,
@@ -23,10 +26,14 @@ class WriteTokenToRedis(Migration):
             return False
         return True
 
-    def get_token_from_json(self):
+    def get_token_from_json(self) -> Optional[Token]:
         try:
             with ReadUserData(UserDataFiles.SECRETS) as userdata:
-                return userdata["api"]["token"]
+                return Token(
+                    token=userdata["api"]["token"],
+                    device_name="Initial device",
+                    created_at=datetime.now(),
+                )
         except Exception as e:
             print(e)
             return None