def widget_test(model_test, monkeypatch, qtbot):
# Fake registry.
def list_plugins(plugin_type):
assert plugin_type == 'dictionary'
for name, readonly in (
('bad', False),
('json', False),
('ro', True),
):
obj = SimpleNamespace(readonly=readonly)
yield SimpleNamespace(name=name, obj=obj)
registry = mock.MagicMock(spec=['list_plugins'])
registry.list_plugins.side_effect = list_plugins
monkeypatch.setattr('plover.gui_qt.dictionaries_widget.registry', registry)
# Fake file dialog.
file_dialog = mock.MagicMock(spec='''
getOpenFileNames
getSaveFileName
'''.split())
monkeypatch.setattr('plover.gui_qt.dictionaries_widget.QFileDialog',
file_dialog)
# Fake `create_dictionary`.
def create_dictionary(filename, threaded_save=True):
pass
steno_dict = mock.create_autospec(StenoDictionary)
create_dictionary = mock.create_autospec(create_dictionary,
return_value=steno_dict)
monkeypatch.setattr('plover.gui_qt.dictionaries_widget.create_dictionary',
create_dictionary)
# Patch `DictionariesModel` constructor to use our own instance.
monkeypatch.setattr('plover.gui_qt.dictionaries_widget.DictionariesModel',
lambda engine, icons: model_test.model)
widget = DictionariesWidget()
widget.setup(model_test.engine)
qtbot.addWidget(widget)
test = WidgetTest(registry, qtbot, widget, file_dialog, create_dictionary,
model_test)
return test
def __init__(self, engine, use_qt_notifications):
super(MainWindow, self).__init__()
self.setupUi(self)
if hasattr(self, 'setUnifiedTitleAndToolBarOnMac'):
self.setUnifiedTitleAndToolBarOnMac(True)
self._engine = engine
self._active_dialogs = {}
self._dialog_class = {
'about' : AboutDialog,
'configuration' : ConfigWindow,
}
all_actions = find_menu_actions(self.menubar)
# Dictionaries.
self.dictionaries = DictionariesWidget(engine)
self.dictionaries.add_translation.connect(self._add_translation)
self.scroll_area.setWidget(self.dictionaries)
self.dictionaries.setFocus()
edit_menu = all_actions['menu_Edit'].menu()
edit_menu.addAction(self.dictionaries.action_Undo)
edit_menu.addSeparator()
edit_menu.addAction(self.dictionaries.action_AddDictionaries)
edit_menu.addAction(self.dictionaries.action_EditDictionaries)
edit_menu.addAction(self.dictionaries.action_RemoveDictionaries)
edit_menu.addSeparator()
edit_menu.addAction(self.dictionaries.action_MoveDictionariesUp)
edit_menu.addAction(self.dictionaries.action_MoveDictionariesDown)
# Tray icon.
self._trayicon = TrayIcon()
self._trayicon.enable()
self._trayicon.clicked.connect(self._engine.toggle_output)
if use_qt_notifications:
handler = NotificationHandler()
handler.emitSignal.connect(self._trayicon.log)
log.add_handler(handler)
popup_menu = QMenu()
for action_name in (
'action_ToggleOutput',
'action_Reconnect',
'',
'menu_Tools',
'',
'action_Configure',
'',
'menu_Help',
'',
'action_Show',
'action_Quit',
):
if action_name:
popup_menu.addAction(all_actions[action_name])
else:
popup_menu.addSeparator()
self._trayicon.set_menu(popup_menu)
engine.signal_connect('machine_state_changed', self._trayicon.update_machine_state)
engine.signal_connect('quit', self.on_quit)
self.action_Quit.triggered.connect(engine.quit)
# Populate tools bar/menu.
tools_menu = all_actions['menu_Tools'].menu()
# Toolbar popup menu for selecting which tools are shown.
self.toolbar_menu = QMenu()
self.toolbar.setContextMenuPolicy(Qt.CustomContextMenu)
self.toolbar.customContextMenuRequested.connect(
lambda: self.toolbar_menu.popup(QCursor.pos())
)
for tool_plugin in registry.list_plugins('gui.qt.tool'):
tool = tool_plugin.obj
action_parameters = []
if tool.ICON is not None:
icon = tool.ICON
# Internal QT resources start with a `:`.
if not icon.startswith(':'):
icon = resource_filename(icon)
action_parameters.append(QIcon(icon))
action_parameters.append(tool.TITLE)
toolbar_action = None
for parent in (tools_menu, self.toolbar, self.toolbar_menu):
action = parent.addAction(*action_parameters)
action.setObjectName(tool_plugin.name)
if tool.__doc__ is not None:
action.setToolTip(tool.__doc__)
if tool.SHORTCUT is not None:
action.setShortcut(QKeySequence.fromString(tool.SHORTCUT))
if parent == self.toolbar_menu:
action.setCheckable(True)
action.setChecked(True)
assert toolbar_action is not None
action.toggled.connect(toolbar_action.setVisible)
else:
if parent == self.toolbar:
toolbar_action = action
action.triggered.connect(partial(self._activate_dialog,
tool_plugin.name,
args=()))
self._dialog_class[tool_plugin.name] = tool
engine.signal_connect('output_changed', self.on_output_changed)
# Machine.
self.machine_type.addItems(
_(plugin.name)
for plugin in registry.list_plugins('machine')
)
engine.signal_connect('config_changed', self.on_config_changed)
engine.signal_connect('machine_state_changed',
lambda machine, state:
self.machine_state.setText(_(state.capitalize()))
)
self.restore_state()
# Commands.
engine.signal_connect('add_translation', partial(self._add_translation, manage_windows=True))
engine.signal_connect('focus', self._focus)
engine.signal_connect('configure', partial(self._configure, manage_windows=True))
engine.signal_connect('lookup', partial(self._activate_dialog, 'lookup',
manage_windows=True))
# Load the configuration (but do not start the engine yet).
if not engine.load_config():
self.on_configure()
# Apply configuration settings.
config = self._engine.config
self.machine_type.setCurrentText(config['machine_type'])
self._configured = False
self.dictionaries.on_config_changed(config)
self.set_visible(not config['start_minimized'])
# Start the engine.
engine.start()
def __init__(self, engine, use_qt_notifications):
super(MainWindow, self).__init__()
self.setupUi(self)
if hasattr(self, 'setUnifiedTitleAndToolBarOnMac'):
self.setUnifiedTitleAndToolBarOnMac(True)
self._engine = engine
self._active_dialogs = {}
self._dialog_class = {
'about': AboutDialog,
'add_translation': AddTranslation,
'configuration': ConfigWindow,
'lookup': LookupDialog,
'paper_tape': PaperTape,
'suggestions': SuggestionsDialog,
}
self.action_Quit.triggered.connect(QCoreApplication.quit)
all_actions = find_menu_actions(self.menubar)
# Dictionaries.
self.dictionaries = DictionariesWidget(engine)
self.dictionaries.add_translation.connect(self._add_translation)
self.scroll_area.setWidget(self.dictionaries)
self.dictionaries.setFocus()
edit_menu = all_actions['menu_Edit'].menu()
edit_menu.addAction(self.dictionaries.action_Undo)
edit_menu.addSeparator()
edit_menu.addAction(self.dictionaries.action_AddDictionaries)
edit_menu.addAction(self.dictionaries.action_EditDictionaries)
edit_menu.addAction(self.dictionaries.action_RemoveDictionaries)
edit_menu.addSeparator()
edit_menu.addAction(self.dictionaries.action_MoveDictionariesUp)
edit_menu.addAction(self.dictionaries.action_MoveDictionariesDown)
# Tray icon.
self._trayicon = TrayIcon()
self._trayicon.enable()
self._trayicon.clicked.connect(self._engine.toggle_output)
if use_qt_notifications:
handler = NotificationHandler()
handler.emitSignal.connect(self._trayicon.log)
log.add_handler(handler)
popup_menu = QMenu()
for action_name in (
'action_ToggleOutput',
'action_Reconnect',
'',
'menu_Tools',
'',
'action_Configure',
'',
'menu_Help',
'',
'action_ShowHide',
'action_Quit',
):
if action_name:
popup_menu.addAction(all_actions[action_name])
else:
popup_menu.addSeparator()
self._trayicon.set_menu(popup_menu)
engine.signal_connect('machine_state_changed',
self._trayicon.update_machine_state)
engine.signal_connect('output_changed', self.on_output_changed)
# Machine.
self.machine_type.addItems(
_(machine) for machine in engine.list_plugins('machine'))
engine.signal_connect('config_changed', self.on_config_changed)
engine.signal_connect(
'machine_state_changed',
lambda machine, state: self.machine_state.setText(
_(state.capitalize())))
self.restore_state()
# Commands.
engine.signal_connect(
'add_translation',
partial(self._add_translation, manage_windows=True))
engine.signal_connect('focus', self._focus)
engine.signal_connect('configure',
partial(self._configure, manage_windows=True))
engine.signal_connect('lookup',
partial(self._lookup, manage_windows=True))
# Load the configuration (but do not start the engine yet).
if not engine.load_config():
self.on_configure()
# Apply configuration settings.
config = self._engine.config
self.on_config_changed(config)
self.dictionaries.on_config_changed(config)
self.set_visible(not config['start_minimized'])
if config['show_suggestions_display']:
self.on_suggestions()
if config['show_stroke_display']:
self.on_paper_tape()
# Start the engine.
engine.start()
