Rare/rare/lgndr/cli.py

import logging

import legendary.cli
from PyQt5.QtWidgets import QLabel, QMessageBox
from legendary.cli import LegendaryCLI as LegendaryCLIReal, logger

from .core import LegendaryCore
from .exception import LgndrException, LgndrLogHandler


def get_boolean_choice(message):
    choice = QMessageBox.question(None, "Import DLCs?", message)
    return True if choice == QMessageBox.StandardButton.Yes else False


class UILogHandler(logging.Handler):
    def __init__(self, dest: QLabel):
        super(UILogHandler, self).__init__()
        self.widget = dest

    def emit(self, record: logging.LogRecord) -> None:
        self.widget.setText(record.getMessage())


class LegendaryCLI(LegendaryCLIReal):

    def __init__(self):
        self.core = None
        self.logger = logging.getLogger('cli')
        self.logging_queue = None

    def import_game(self, args):
        handler = LgndrLogHandler()
        logger.addHandler(handler)
        old_choice = legendary.cli.get_boolean_choice
        legendary.cli.get_boolean_choice = get_boolean_choice
        try:
            super(LegendaryCLI, self).import_game(args)
        except LgndrException as e:
            raise e
        finally:
            legendary.cli.get_boolean_choice = old_choice
            logger.removeHandler(handler)