From fe86382819bea75427dcae6d5ea8ec13570e0b81 Mon Sep 17 00:00:00 2001
From: Inex Code <inex.code@selfprivacy.org>
Date: Fri, 14 Jan 2022 09:24:33 +0200
Subject: [PATCH] Fix migration to run

---
 selfprivacy_api/migrations/__init__.py           | 3 ++-
 selfprivacy_api/migrations/create_tokens_json.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/selfprivacy_api/migrations/__init__.py b/selfprivacy_api/migrations/__init__.py
index 32c467f..8c40f7e 100644
--- a/selfprivacy_api/migrations/__init__.py
+++ b/selfprivacy_api/migrations/__init__.py
@@ -1,7 +1,8 @@
 from selfprivacy_api.utils import ReadUserData
 from selfprivacy_api.migrations.fix_nixos_config_branch import FixNixosConfigBranch
+from selfprivacy_api.migrations.create_tokens_json import CreateTokensJson
 
-migrations = [FixNixosConfigBranch()]
+migrations = [FixNixosConfigBranch(), CreateTokensJson()]
 
 
 def run_migrations():
diff --git a/selfprivacy_api/migrations/create_tokens_json.py b/selfprivacy_api/migrations/create_tokens_json.py
index 8f91dbf..7404301 100644
--- a/selfprivacy_api/migrations/create_tokens_json.py
+++ b/selfprivacy_api/migrations/create_tokens_json.py
@@ -35,7 +35,8 @@ class CreateTokensJson(Migration):
 
     def migrate(self):
         try:
-            with ReadUserData(USERDATA_FILE) as userdata:
+            print(f"Creating tokens.json file at {TOKENS_FILE}")
+            with ReadUserData() as userdata:
                 token = userdata["api"]["token"]
             # Touch tokens.json with 0600 permissions
             Path(TOKENS_FILE).touch(mode=0o600)