mirror of
https://github.com/jeertmans/manim-slides.git
synced 2025-05-20 03:57:38 +08:00
ae8d5b6aab
This PR refactors a lot of the code in order to have a better organized codebase. If will that I needed to create a second level of submodules, to better distinguish the different parts of this project.
122 lines
3.6 KiB
Python
122 lines
3.6 KiB
Python
from functools import partial
|
|
from typing import Any
|
|
|
|
from qtpy.QtCore import Qt
|
|
from qtpy.QtGui import QIcon, QKeyEvent
|
|
from qtpy.QtWidgets import (
|
|
QDialog,
|
|
QDialogButtonBox,
|
|
QGridLayout,
|
|
QLabel,
|
|
QMessageBox,
|
|
QPushButton,
|
|
QVBoxLayout,
|
|
QWidget,
|
|
)
|
|
|
|
from ..config import Config, Key
|
|
from ..logger import logger
|
|
from ..resources import * # noqa: F403
|
|
|
|
WINDOW_NAME: str = "Configuration Wizard"
|
|
|
|
keymap = {}
|
|
for key in Qt.Key:
|
|
keymap[key.value] = key.name.partition("_")[2]
|
|
|
|
|
|
class KeyInput(QDialog): # type: ignore
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.key = None
|
|
|
|
self.layout = QVBoxLayout()
|
|
|
|
self.setWindowTitle("Keyboard Input")
|
|
self.label = QLabel("Press any key to register it")
|
|
self.layout.addWidget(self.label)
|
|
self.setLayout(self.layout)
|
|
|
|
def keyPressEvent(self, event: QKeyEvent) -> None: # noqa: N802
|
|
self.key = event.key()
|
|
self.deleteLater()
|
|
event.accept()
|
|
|
|
|
|
class Wizard(QWidget): # type: ignore
|
|
def __init__(self, config: Config):
|
|
super().__init__()
|
|
|
|
self.setWindowTitle(WINDOW_NAME)
|
|
self.config = config
|
|
self.icon = QIcon(":/icon.png")
|
|
self.setWindowIcon(self.icon)
|
|
self.closed_without_saving = False
|
|
|
|
button = QDialogButtonBox.Save | QDialogButtonBox.Cancel
|
|
|
|
self.button_box = QDialogButtonBox(button)
|
|
self.button_box.accepted.connect(self.save_config)
|
|
self.button_box.rejected.connect(self.close_without_saving)
|
|
|
|
self.buttons = []
|
|
|
|
self.layout = QGridLayout()
|
|
|
|
for i, (key, value) in enumerate(self.config.keys.dict().items()):
|
|
# Create label for key name information
|
|
label = QLabel()
|
|
key_info = value["name"] or key
|
|
label.setText(key_info.title())
|
|
self.layout.addWidget(label, i, 0)
|
|
|
|
# Create button that will pop-up a dialog and ask to input a new key
|
|
value = value["ids"].pop()
|
|
button = QPushButton(keymap[value])
|
|
button.setToolTip(
|
|
f"Click to modify the key associated to action {key_info}"
|
|
)
|
|
self.buttons.append(button)
|
|
button.clicked.connect(
|
|
partial(self.open_dialog, i, getattr(self.config.keys, key))
|
|
)
|
|
self.layout.addWidget(button, i, 1)
|
|
|
|
self.layout.addWidget(self.button_box, len(self.buttons), 1)
|
|
|
|
self.setLayout(self.layout)
|
|
|
|
def close_without_saving(self) -> None:
|
|
logger.debug("Closing configuration wizard without saving")
|
|
self.closed_without_saving = True
|
|
self.deleteLater()
|
|
|
|
def closeEvent(self, event: Any) -> None: # noqa: N802
|
|
self.close_without_saving()
|
|
event.accept()
|
|
|
|
def save_config(self) -> None:
|
|
try:
|
|
Config.model_validate(self.config.dict())
|
|
except ValueError:
|
|
msg = QMessageBox()
|
|
msg.setIcon(QMessageBox.Critical)
|
|
msg.setText("Error")
|
|
msg.setInformativeText(
|
|
"Two or more actions share a common key: make sure actions have distinct key codes."
|
|
)
|
|
msg.setWindowTitle("Error: duplicated keys")
|
|
msg.exec()
|
|
return
|
|
|
|
self.deleteLater()
|
|
|
|
def open_dialog(self, button_number: int, key: Key) -> None:
|
|
button = self.buttons[button_number]
|
|
dialog = KeyInput()
|
|
dialog.exec()
|
|
if dialog.key is not None:
|
|
key_name = keymap[dialog.key]
|
|
key.set_ids(dialog.key)
|
|
button.setText(key_name)
|