selfprivacy-rest-api/selfprivacy_api/migrations/create_tokens_json.py

from datetime import datetime
import os
import json
from pathlib import Path

from selfprivacy_api.migrations.migration import Migration
from selfprivacy_api.utils import TOKENS_FILE, ReadUserData


class CreateTokensJson(Migration):
    def get_migration_name(self):
        return "create_tokens_json"

    def get_migration_description(self):
        return """Selfprivacy API used a single token in userdata.json for authentication.
        This migration creates a new tokens.json file with the old token in it.
        This migration runs if the tokens.json file does not exist.
        Old token is located at ["api"]["token"] in userdata.json.
        tokens.json path is declared in TOKENS_FILE imported from utils.py
        tokens.json must have the following format:
        {
            "tokens": [
                {
                    "token": "token_string",
                    "name": "Master Token",
                    "date": "current date from str(datetime.now())",
                }
            ]
        }
        tokens.json must have 0600 permissions.
        """

    def is_migration_needed(self):
        return not os.path.exists(TOKENS_FILE)

    def migrate(self):
        try:
            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)
            # Write token to tokens.json
            structure = {
                "tokens": [
                    {
                        "token": token,
                        "name": "primary_token",
                        "date": str(datetime.now()),
                    }
                ]
            }
            with open(TOKENS_FILE, "w", encoding="utf-8") as tokens:
                json.dump(structure, tokens, indent=4)
            print("Done")
        except Exception as e:
            print(e)
            print("Error creating tokens.json")