Source code for retype.controllers.menu_controller

from qt import QAction, QObject

from retype.constants import (
    RETYPE_ISSUE_TRACKER_URL, RETYPE_DOCUMENTATION_URL, iswindows)
from retype.resource_handler import getIcon


class MenuController(QObject):
    def __init__(self, main_controller, menu):
        super().__init__()
        self.controller = main_controller
        self._menu = menu
        self._initMenuBar()

    def _initMenuBar(self):
        self._fileMenu()
        self._viewMenu()
        self._optionsMenu()
        self._helpMenu()

    def _makeAction(self, name, connect_to, shortcuts=None, icon_name=None):
        action = QAction(name, self)
        action.triggered.connect(connect_to)
        if shortcuts:
            action.setShortcuts(shortcuts)
        if icon_name:
            action.setIcon(getIcon(icon_name))
        return action

    def _addAction(self, menu, action_name, connect_to, shortcuts=None,
                   icon_name=None):
        action = self._makeAction(action_name, connect_to, shortcuts,
                                  icon_name)
        menu.addAction(action)

    def _fileMenu(self):
        fileMenu = self._menu.addMenu('&File')
        self._addAction(fileMenu, '&Quit', self.controller.quit,
                        ['Alt+F4'], 'door')

    def _viewMenu(self):
        viewMenu = self._menu.addMenu('&View')
        self._addAction(viewMenu, '&Shelf View',
                        lambda: self.controller.setViewByEnum(1), ['Ctrl+1'],
                        'shelf_view')
        self._addAction(viewMenu, '&Book View',
                        lambda: self.controller.setViewByEnum(2), ['Ctrl+2'],
                        'open_book')

        if iswindows:
            viewMenu.addSeparator()
            self._addAction(viewMenu, 'Toggle System &Console',
                            lambda: self.controller.toggleConsoleWindow(),
                            icon_name='console')

    def _optionsMenu(self):
        optionsMenu = self._menu.addMenu('&Options')
        self._addAction(optionsMenu, '&Customise retype',
                        self.controller.showCustomisationDialog, ['Ctrl+O'],
                        'customise')

    def _helpMenu(self):
        helpMenu = self._menu.addMenu('&Help')
        self._addAction(helpMenu, "&About", self.controller.showAboutDialog,
                        icon_name='about')
        self._addAction(helpMenu, "&Documentation (opens in browser)", lambda:
                        self.controller.openUrl(RETYPE_DOCUMENTATION_URL),
                        icon_name='documentation')
        self._addAction(helpMenu,
                        '&Report issue (opens in browser)', lambda:
                        self.controller.openUrl(RETYPE_ISSUE_TRACKER_URL),
                        icon_name='issue')