Launcher/main.py

import os
import sys
import argparse
import logging

from app import Launcher
from tkinter import messagebox

from core_lib.log.default import DEFAULT_FORMAT, DEFAULT_DATE_FORMAT, get_colorful_handler


def main():
    root_logger = logging.getLogger()
    logger = logging.getLogger("Launcher.Main")

    # parse arguments
    parser = argparse.ArgumentParser()
    parser.add_argument("-m", "--mode", choices=['lightweight', 'full', 'choice'], default='choice',
                        help="Select mode (lightweight for tui, full is gui,"
                             "choice for a mode select window) (TUI is not implemented)", required=False)
    parser.add_argument("-d", "--debug", action="store_true",
                        help="Enable debug mode", default=False)
    parser.add_argument("-wd", "--work-dir", default=os.getcwd(), help="Work directory")

    args, unknown = parser.parse_known_args()
    for arg in unknown:
        logger.warning("Unknown argument '%s'" % arg)

    debug = args.debug

    # check workdir
    if not os.path.exists(args.work_dir) or not os.path.isdir(args.work_dir):
        messagebox.showerror("Error", f"Specified work directory \"{args.work_dir}\" does not exist or"
                                      " is not a directory.")
        logger.error("Work directory '%s' does not exist. Exiting..." % args.work_dir)
        sys.exit(-1)

    if os.getcwd() != args.work_dir:
        os.chdir(args.work_dir)

    logger.info("Current working directory is {}".format(args.work_dir))

    # Init lib
    level = logging.DEBUG if debug else logging.INFO

    # Set root logger level and configure handler for colorful log (not sure why I added this)
    root_logger.setLevel(level)
    root_logger.addHandler(get_colorful_handler())

    # Logger configure
    logging.basicConfig(format=DEFAULT_FORMAT, datefmt=DEFAULT_DATE_FORMAT, level=level)

    # Start (the real) launcher
    app = Launcher()
    sys.exit(app.exec())


if __name__ == '__main__':
    main()