fspachinko 0.0.2__py3-none-any.whl

fspachinko/gui/actions.py

@@ -0,0 +1,68 @@
+"""Actions module for QActions."""
+
+from dataclasses import dataclass, field
+
+from PySide6.QtGui import QAction, QIcon
+
+from ..utils import IconFilename, Paths
+from .components import set_qt_tips
+
+
+@dataclass(slots=True)
+class FileActions:
+    """Main file menu actions."""
+
+    save: QAction = field(init=False)
+    save_as: QAction = field(init=False)
+    load: QAction = field(init=False)
+    autosave: QAction = field(init=False)
+    exit: QAction = field(init=False)
+
+    def __post_init__(self) -> None:
+        """Initialize the main actions."""
+        self.save = QAction(icon=QIcon(Paths.icon(IconFilename.SAVE)), text="&Save Profile")
+        self.save.setShortcut("Ctrl+S")
+        set_qt_tips(self.save, "Save current GUI profile (Ctrl+S)")
+
+        self.save_as = QAction(icon=QIcon(Paths.icon(IconFilename.SAVE_AS)), text="Save Profile &As")
+        self.save_as.setShortcut("Ctrl+Shift+S")
+        set_qt_tips(self.save_as, "Save current GUI profile as ... (Ctrl+Shift+S)")
+
+        self.load = QAction(icon=QIcon(Paths.icon(IconFilename.OPEN)), text="&Load Profile")
+        self.load.setShortcut("Ctrl+O")
+        set_qt_tips(self.load, "Load GUI profile (Ctrl+O)")
+
+        self.autosave = QAction(
+            icon=QIcon(Paths.icon(IconFilename.AUTOSAVE)), text="A&utosave Profile", checkable=True, checked=True
+        )
+        set_qt_tips(self.autosave, "Automatically save profile on exit")
+
+        self.exit = QAction(icon=QIcon(Paths.icon(IconFilename.CLOSE)), text="&Exit")
+        self.exit.setShortcut("Ctrl+W")
+        set_qt_tips(self.exit, "Exit application (Ctrl+W)")
+
+
+@dataclass(slots=True)
+class RunActions:
+    """Main run actions."""
+
+    start: QAction = field(init=False)
+    stop: QAction = field(init=False)
+
+    def __post_init__(self) -> None:
+        """Initialize the main actions."""
+        self.start = QAction(icon=QIcon(Paths.icon(IconFilename.START)), text="&Start")
+        self.start.setShortcut("Ctrl+R")
+        set_qt_tips(self.start, "Start (Ctrl+R)")
+
+        self.stop = QAction(icon=QIcon(Paths.icon(IconFilename.STOP)), text="S&top")
+        self.stop.setShortcut("ESC")
+        set_qt_tips(self.stop, "Stop (ESC)")
+
+
+@dataclass(slots=True)
+class Actions:
+    """Main actions."""
+
+    file: FileActions = field(default_factory=FileActions)
+    run: RunActions = field(default_factory=RunActions)

fspachinko/gui/centralwidget.py

@@ -0,0 +1,70 @@
+"""Main module."""
+
+import logging
+
+from PySide6.QtCore import Slot
+from PySide6.QtWidgets import QGroupBox, QWidget
+
+from ..utils import PERCENTAGE_100, GUIName
+from .components import ProgressBinder
+from .qthelpers import set_qt_name
+from .uibuilder import UIBuilder
+from .workers import MainThread, MainWorker
+
+logger = logging.getLogger(__name__)
+
+
+class CentralWidget(QWidget):
+    """Main application window."""
+
+    def __init__(self) -> None:
+        """Initialize the main window."""
+        super().__init__()
+        set_qt_name(self, GUIName.CENTRAL_WIDGET)
+        self.main_thread: MainThread | None = None
+        self.ui = UIBuilder()
+        layout = self.ui.build_layout()
+        self.setLayout(layout)
+        self.progress_binder = ProgressBinder(self.ui.progress, self.ui.logging)
+        self.window_title_original = ""
+
+    @Slot()
+    def on_start(self) -> None:
+        """Start the process and disable UI elements."""
+        try:
+            config = self.ui.get_config()
+        except Exception:
+            logger.exception("")
+            return
+
+        self.main_thread = MainThread(MainWorker.from_config(config))
+        self.ui.progress.reset()
+        self.window_title_original = self.window().windowTitle()
+        self.progress_binder.bind(self.main_thread.worker.signals)
+        self.progress_binder.count.connect(self.update_title_progress)
+        self.progress_binder.finished.connect(self.on_finished)
+        self.toggle_ui(is_enabled=False)
+        self.main_thread.start()
+
+    @Slot()
+    def on_stop(self) -> None:
+        """Stop the process."""
+        if self.main_thread and self.main_thread.isRunning():
+            self.main_thread.stop()
+
+    @Slot(int)
+    def update_title_progress(self, val: int) -> None:
+        """Update window title with progress percentage."""
+        pct = int((val / self.ui.progress.progbar_dir.maximum()) * PERCENTAGE_100)
+        self.window().setWindowTitle(f"[{pct}%] {self.window_title_original}")
+
+    @Slot()
+    def on_finished(self) -> None:
+        """Handle worker finished signal."""
+        self.toggle_ui(is_enabled=True)
+        self.window().setWindowTitle(self.window_title_original)
+
+    def toggle_ui(self, *, is_enabled: bool) -> None:
+        """Lock or unlock UI elements."""
+        for child in self.findChildren(QGroupBox):
+            child.setEnabled(is_enabled)