crafty-4/main.py

import os
import sys
import json
import time
import argparse
import logging.config

""" Our custom classes / pip packages """
from app.classes.shared.console import console
from app.classes.shared.helpers import helper
from app.classes.shared.models import installer
from app.classes.shared.tasks import tasks_manager

def do_intro():
    logger.info("***** Commander Agent Launched *****")

    version_data = helper.get_version()

    # set some defaults if we don't get version_data from our helper
    version = "{}.{}.{}".format(version_data.get('major', '?'),
                                version_data.get('minor', '?'),
                                version_data.get('sub', '?'))

    intro = """
    {lines}
    #\t\tWelcome to Crafty Controller - v.{version}\t\t      #
    #\t\t\t   Codename: Commander \t\t\t\t      #
    {lines}
    #   \tServer Manager / Web Portal for your Minecraft server \t      #
    #   \t\tHomepage: www.craftycontrol.com\t\t\t      #
    {lines}
    """.format(lines="/" * 75, version=version)

    console.magenta(intro)


def setup_logging(debug=False):
    logging_config_file = os.path.join(os.path.curdir,
                                       'app',
                                       'config',
                                       'logging.json'
                                       )

    if os.path.exists(logging_config_file):
        # open our logging config file
        with open(logging_config_file, 'rt') as f:
            logging_config = json.load(f)
            if debug:
                logging_config['loggers']['']['level'] = 'DEBUG'
            logging.config.dictConfig(logging_config)
    else:
        logging.basicConfig(level=logging.DEBUG)
        logging.warning("Unable to read logging config from {}".format(logging_config_file))
        console.critical("Unable to read logging config from {}".format(logging_config_file))


""" Our Main Starter """
if __name__ == '__main__':

    parser = argparse.ArgumentParser("Crafty Commander - A Server Management System")

    parser.add_argument('-i', '--ignore',
                        action='store_true',
                        help="Ignore session.json files"
                        )

    parser.add_argument('-v', '--verbose',
                        action='store_true',
                        help="Sets logging level to debug."
                        )

    args = parser.parse_args()

    helper.ensure_logging_setup()

    setup_logging(debug=args.verbose)

    # setting up the logger object
    logger = logging.getLogger(__name__)

    # print our pretty start message
    do_intro()

    # our session file, helps prevent multiple commander agents on the same machine.
    helper.create_session_file(ignore=args.ignore)

    tasks_manager.start_webserver()

    # this should always be last
    tasks_manager.start_main_kill_switch_watcher()

    # do our installer stuff
    if installer.is_fresh_install():
        installer.create_tables()
        installer.default_settings()

    # our main loop
    while True:
        if tasks_manager.get_main_thread_run_status():
            sys.exit(0)
        time.sleep(1)