59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
from datetime import datetime
|
|
import os
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from selfprivacy_api.migrations.migration import Migration
|
|
from selfprivacy_api.utils import USERDATA_FILE, 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") as tokens:
|
|
json.dump(structure, tokens, indent=4)
|
|
print("Done")
|
|
except Exception as e:
|
|
print(e)
|
|
print("Error creating tokens.json")
|