piket/piket_client/gui.py

"""
Provides the graphical front-end for Piket.
"""
import os
import sys
from urllib.parse import urljoin

# pylint: disable=E0611
from PySide2.QtWidgets import (
    QApplication,
    QGridLayout,
    QInputDialog,
    QMainWindow,
    QPushButton,
    QSizePolicy,
    QToolBar,
    QWidget,
)
from PySide2.QtGui import QIcon
from PySide2.QtCore import QSize

# pylint: enable=E0611
import requests

from piket_client.sound import PLOP_WAVE


def plop() -> None:
    """ Asynchronously play the plop sound. """
    PLOP_WAVE.play()


SERVER_URL = "http://127.0.0.1:5000"


def get_people() -> [dict]:
    """ Request list of active people from the server. """
    request = requests.get(urljoin(SERVER_URL, "/people"))
    return request.json()["people"]


class NameButton(QPushButton):
    """ Wraps a QPushButton to provide a counter. """

    def __init__(self, person: dict, *args, **kwargs) -> None:
        self.person_id = person["person_id"]
        self.name = person["name"]
        self.count = person["consumptions"]["1"]

        super().__init__(self.current_label, *args, **kwargs)
        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)

        self.clicked.connect(self.plop)

    @property
    def current_label(self) -> str:
        """ Return the label to show on the button. """
        return f"{self.name} ({self.count})"

    def plop(self) -> None:
        """ Process a click on this button. """
        req = requests.post(
            urljoin(SERVER_URL, f"people/{self.person_id}/add_consumption")
        )
        if req.status_code == 200:
            json = req.json()
            person = json["person"]
            self.count = person["consumptions"]["1"]
            self.setText(self.current_label)
            plop()
        else:
            print("Oh shit kapot")


class NameButtons(QWidget):
    """ Main widget responsible for capturing presses and registering them.
    """

    def __init__(self) -> None:
        super().__init__()

        self.layout = None
        self.init_ui()

    def init_ui(self) -> None:
        """ Initialize UI: build GridLayout, retrieve People and build a button
        for each. """
        self.layout = QGridLayout()

        for index, person in enumerate(get_people()):
            button = NameButton(person)
            self.layout.addWidget(button, index // 2, index % 2)

        self.setLayout(self.layout)


class PiketMainWindow(QMainWindow):
    """ QMainWindow subclass responsible for showing the main application
    window. """

    def __init__(self) -> None:
        super().__init__()

        self.main_widget = None
        self.toolbar = None
        self.init_ui()

    def init_ui(self) -> None:
        """ Initialize the UI: construct main widget and toolbar. """
        self.main_widget = NameButtons()
        self.setCentralWidget(self.main_widget)

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

        # Initialize toolbar
        self.toolbar = QToolBar()
        self.toolbar.setIconSize(QSize(icon_size, icon_size))

        # Left
        self.toolbar.addAction(
            self.load_icon("add_person.svg"), "Nieuw persoon", self.add_person
        )
        self.toolbar.addAction(self.load_icon("undo.svg"), "Heydrich")

        self.toolbar.addWidget(self.create_spacer())

        # Right
        self.toolbar.addAction(self.load_icon("beer_bottle.svg"), "Bierrr")
        self.addToolBar(self.toolbar)

    def add_person(self) -> None:
        """ Ask for a new Person and register it, then rebuild the central
        widget. """
        name, ok = QInputDialog.getItem(
            self,
            "Persoon toevoegen",
            "Voer de naam van de nieuwe persoon in, of kies uit de lijst.",
            ["Cas", "Frenk"],
            0,
            True,
        )
        if ok and name:
            req = requests.post(
                urljoin(SERVER_URL, "people"), json={"person": {"name": name}}
            )

            self.main_widget = NameButtons()
            self.setCentralWidget(self.main_widget)

    @staticmethod
    def create_spacer() -> QWidget:
        """ Return an empty QWidget that automatically expands. """
        spacer = QWidget()
        spacer.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Preferred)
        return spacer

    icons_dir = os.path.join(os.path.dirname(__file__), "icons")

    @classmethod
    def load_icon(cls, filename: str) -> QIcon:
        """ Return a QtIcon loaded from the given `filename` in the icons
        directory. """

        return QIcon(os.path.join(cls.icons_dir, filename))


def main() -> None:
    """ Main entry point of GUI client. """
    app = QApplication(sys.argv)
    font = app.font()
    size = font.pointSize()
    font.setPointSize(size * 1.75)
    app.setFont(font)

    main_window = PiketMainWindow()
    main_window.show()

    app.exec_()


if __name__ == "__main__":
    main()