crafty-4/app/classes/shared/import3.py

import json
import os
import logging

from app.classes.controllers.users_controller import users_helper
from app.classes.shared.main_controller import Controller
from app.classes.shared.console import console

logger = logging.getLogger(__name__)


class import3:
    def __init__(self):
        self.controller = Controller()

    def start_import(self):
        folder = os.path.normpath(
            input(
                "Please input the path to the migrations folder "
                "in your installation of Crafty 3: "
            )
        )
        if not os.path.exists(folder):
            console.info(
                "Crafty cannot find the path you entered. "
                "Does Crafty's user have permission to access it?"
            )
            console.info("Please run the import3 command again and enter a valid path.")
        else:
            with open(os.path.join(folder, "users.json"), encoding="utf-8") as f:
                user_json = json.loads(f.read())
            with open(os.path.join(folder, "mc_settings.json"), encoding="utf-8") as f:
                servers_json = json.loads(f.read())
            self.import_users(user_json)
            self.import_servers(servers_json, self.controller)

    @staticmethod
    def import_users(json_data):
        # If there is only one user to import json needs to call the data differently
        if isinstance(json_data, list):
            for user in json_data:
                users_helper.add_rawpass_user(user["username"], user["password"])
                console.info(f"Imported user {user['username']} from Crafty 3")
                logger.info(f"Imported user {user['username']} from Crafty 3")
        else:
            console.info(
                "There is only one user detected. "
                "Cannot create duplicate Admin account."
            )
            logger.info(
                "There is only one user detected. "
                "Cannot create duplicate Admin account."
            )

    @staticmethod
    def import_servers(json_data, controller):
        # If there is only one server to import json needs to call the data differently
        if isinstance(json_data, list):
            for server in json_data:
                new_server_id = controller.import_jar_server(
                    server_name=server["server_name"],
                    server_path=server["server_path"],
                    server_jar=server["server_jar"],
                    min_mem=(int(server["memory_min"]) / 1000),
                    max_mem=(int(server["memory_max"]) / 1000),
                    port=server["server_port"],
                )
                console.info(
                    f"Imported server {server['server_name']}[{server['id']}] "
                    f"from Crafty 3 to new server id {new_server_id}"
                )
                logger.info(
                    f"Imported server {server['server_name']}[{server['id']}] "
                    f"from Crafty 3 to new server id {new_server_id}"
                )
        else:
            new_server_id = controller.import_jar_server(
                server_name=json_data["server_name"],
                server_path=json_data["server_path"],
                server_jar=json_data["server_jar"],
                min_mem=(int(json_data["memory_min"]) / 1000),
                max_mem=(int(json_data["memory_max"]) / 1000),
                port=json_data["server_port"],
            )
            console.info(
                f"Imported server {json_data['server_name']}[{json_data['id']}] "
                f"from Crafty 3 to new server id {new_server_id}"
            )
            logger.info(
                f"Imported server {json_data['server_name']}[{json_data['id']}] "
                f"from Crafty 3 to new server id {new_server_id}"
            )


import3 = import3()