piket/piket_client/set_active.py

"""
Provides a helper tool to (de-)activate multiple people at once.
"""

import math
import sys

# pylint: disable=E0611
import qdarkstyle
from PySide2.QtCore import QObject, QSize, Qt, Signal, Slot
from PySide2.QtGui import QIcon
from PySide2.QtWidgets import (
    QAction,
    QActionGroup,
    QApplication,
    QGridLayout,
    QInputDialog,
    QLineEdit,
    QMainWindow,
    QMessageBox,
    QPushButton,
    QSizePolicy,
    QToolBar,
    QWidget,
)

from piket_client.model import NetworkError, Person, ServerStatus

# pylint: enable=E0611


class ActivationButton(QPushButton):
    def __init__(self, person, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.setCheckable(True)
        self.person = person

        self.setText(self.person.name)
        self.setChecked(self.person.active)

        self.toggled.connect(self.process_toggled)

    def process_toggled(self, new_state: bool) -> None:
        self.person = self.person.set_active(new_state)


class ActivationButtons(QWidget):
    def __init__(self, *args, **kwargs) -> None:
        super().__init__(*args, **kwargs)

        self.layout = QGridLayout()
        self.setLayout(self.layout)
        self.init_ui()

    def init_ui(self) -> None:
        ps = Person.get_all()
        assert not isinstance(ps, NetworkError)
        num_columns = math.ceil(math.sqrt(len(ps)))

        for index, person in enumerate(ps):
            button = ActivationButton(person, self)
            self.layout.addWidget(button, index // num_columns, index % num_columns)


class ActiveStateMainWindow(QMainWindow):
    def __init__(self) -> None:
        super().__init__()

        self.toolbar = QToolBar()

        self.init_ui()

    def init_ui(self) -> None:
        """ Initialize the UI: construct main widget, toolbar. """

        self.setWindowState(Qt.WindowActive | Qt.WindowFullScreen)

        font_metrics = self.fontMetrics()
        icon_size = font_metrics.height() * 1.45

        # Toolbar
        self.toolbar.setToolButtonStyle(Qt.ToolButtonTextUnderIcon)
        self.toolbar.setIconSize(QSize(icon_size, icon_size))

        # Left
        self.toolbar.addAction("Afsluiten", self.quit)

        self.addToolBar(self.toolbar)

        # Build main widget
        ps = Person.get_all()
        self.main_widget = ActivationButtons()
        self.setCentralWidget(self.main_widget)

    def quit(self) -> None:
        """ Exit immediately. """
        QApplication.instance().quit()


def main() -> None:
    """ Main entry point of GUI client. """
    app = QApplication(sys.argv)

    # Set dark theme
    app.setStyleSheet(qdarkstyle.load_stylesheet_pyside2())

    # Enlarge font size
    font = app.font()
    size = font.pointSize()
    font.setPointSize(size * 0.8)
    app.setFont(font)

    # Test connectivity
    server_running = ServerStatus.is_server_running()

    if not isinstance(server_running, bool):
        QMessageBox.critical(
            None,
            "Help er is iets kapot",
            "Kan niet starten omdat de server niet reageert, stuur een foto van "
            "dit naar Maarten: " + repr(server_running.value),
        )
        return

    # Load main window
    main_window = ActiveStateMainWindow()
    main_window.show()

    app.exec_()


if __name__ == "__main__":
    main()