Add set_active script

piket_client/gui.py

@@ -34,7 +34,13 @@ except ImportError:
     dbus = None
 
 from piket_client.sound import PLOP_WAVE, UNDO_WAVE
-from piket_client.model import Person, ConsumptionType, Consumption, ServerStatus, Settlement
+from piket_client.model import (
+    Person,
+    ConsumptionType,
+    Consumption,
+    ServerStatus,
+    Settlement,
+)
 import piket_client.logger
 
 LOG = logging.getLogger(__name__)
@@ -87,18 +93,17 @@ class NameButton(QPushButton):
 
     def process_click(self) -> None:
         """ Process a click on this button. """
-        LOG.debug('Button clicked.')
+        LOG.debug("Button clicked.")
         result = self.person.add_consumption(self.active_id)
         if result:
             plop()
             self.setText(self.current_label)
             self.consumption_created.emit(result)
         else:
-            LOG.error("Failed to add consumption", extra={'person':
-                self.person})
+            LOG.error("Failed to add consumption", extra={"person": self.person})
 
     def confirm_hide(self) -> None:
-        LOG.debug('Button right-clicked.')
+        LOG.debug("Button right-clicked.")
         ok = QMessageBox.warning(
             self.window(),
             "Persoon verbergen?",
@@ -132,8 +137,7 @@ class NameButtons(QWidget):
     def consumption_type_changed(self, new_id: str):
         """ Process a change of the consumption type and propagate to the
         contained buttons. """
-        LOG.debug('Consumption type updated in NameButtons.',
-                extra={'new_id': new_id})
+        LOG.debug("Consumption type updated in NameButtons.", extra={"new_id": new_id})
         self.active_consumption_type_id = new_id
         self.new_id_set.emit(new_id)
 
@@ -141,7 +145,7 @@ class NameButtons(QWidget):
         """ Initialize UI: build GridLayout, retrieve People and build a button
         for each. """
 
-        LOG.debug('Initializing NameButtons.')
+        LOG.debug("Initializing NameButtons.")
 
         ps = Person.get_all(True)
         num_columns = round(len(ps) / 10) + 1
@@ -170,7 +174,7 @@ class PiketMainWindow(QMainWindow):
     consumption_type_changed = Signal(str)
 
     def __init__(self) -> None:
-        LOG.debug('Initializing PiketMainWindow.')
+        LOG.debug("Initializing PiketMainWindow.")
         super().__init__()
 
         self.main_widget = None
@@ -442,40 +446,39 @@ def main() -> None:
     # Test unsettled consumptions
     status = ServerStatus.unsettled_consumptions()
 
-    unsettled = status['unsettled']['amount']
+    unsettled = status["unsettled"]["amount"]
 
     if unsettled > 0:
-        first = status['unsettled']['first']
-        first_date = first.strftime('%c')
+        first = status["unsettled"]["first"]
+        first_date = first.strftime("%c")
         ok = QMessageBox.information(
             None,
             "Onafgesloten lijst",
             f"Wil je verdergaan met een lijst met {unsettled} onafgesloten "
             f"consumpties sinds {first_date}?",
             QMessageBox.Yes,
-            QMessageBox.No
+            QMessageBox.No,
         )
         if ok == QMessageBox.No:
             main_window.show_keyboard()
             name, ok = QInputDialog.getText(
                 None,
                 "Lijst afsluiten",
-                "Voer een naam in voor de lijst of druk op OK. Laat de datum "
-                "staan.",
+                "Voer een naam in voor de lijst of druk op OK. Laat de datum " "staan.",
                 QLineEdit.Normal,
-                f"{first.strftime('%Y-%m-%d')}"
+                f"{first.strftime('%Y-%m-%d')}",
             )
             main_window.hide_keyboard()
 
             if name and ok:
                 settlement = Settlement.create(name)
-                info = [f'{item["count"]} {item["name"]}' for item in
-                settlement.consumption_summary.values()]
-                info = ', '.join(info)
+                info = [
+                    f'{item["count"]} {item["name"]}'
+                    for item in settlement.consumption_summary.values()
+                ]
+                info = ", ".join(info)
                 QMessageBox.information(
-                    None,
-                    "Lijst afgesloten",
-                    f"VO! Op deze lijst stonden: {info}"
+                    None, "Lijst afgesloten", f"VO! Op deze lijst stonden: {info}"
                 )
 
                 main_window = PiketMainWindow()

piket_client/model.py

@@ -52,6 +52,7 @@ class Person(NamedTuple):
     """ Represents a Person, as retrieved from the database. """
 
     name: str
+    active: bool
     person_id: int = None
     consumptions: dict = {}
 
@@ -156,8 +157,11 @@ class Person(NamedTuple):
     @classmethod
     def get_all(cls, active=None) -> ["Person"]:
         """ Get all active People. """
-        active = int(active)
-        req = requests.get(urljoin(SERVER_URL, "/people"), params={"active": active})
+        params = {}
+        if active is not None:
+            params["active"] = int(active)
+
+        req = requests.get(urljoin(SERVER_URL, "/people"), params=params)
 
         try:
             data = req.json()
@@ -180,6 +184,7 @@ class Person(NamedTuple):
         """ Reconstruct a Person object from a dict. """
         return Person(
             name=data["name"],
+            active=data["active"],
             person_id=data["person_id"],
             consumptions=data["consumptions"],
         )

piket_client/set_active.py

@@ -0,0 +1,135 @@
+"""
+Provides a helper tool to (de-)activate multiple people at once.
+"""
+
+import sys
+
+# pylint: disable=E0611
+from PySide2.QtWidgets import (
+    QAction,
+    QActionGroup,
+    QApplication,
+    QGridLayout,
+    QInputDialog,
+    QLineEdit,
+    QMainWindow,
+    QMessageBox,
+    QPushButton,
+    QSizePolicy,
+    QToolBar,
+    QWidget,
+)
+from PySide2.QtGui import QIcon
+from PySide2.QtCore import QObject, QSize, Qt, Signal, Slot
+
+# pylint: enable=E0611
+
+import qdarkstyle
+
+from piket_client.model import Person, ServerStatus
+
+
+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()
+        num_columns = round(len(ps) / 10) + 1
+
+        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 = None
+
+        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 = QToolBar()
+        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 * 1.5)
+    app.setFont(font)
+
+    # Test connectivity
+    server_running, info = ServerStatus.is_server_running()
+
+    if not server_running:
+        LOG.critical("Could not connect to server", extra={"info": info})
+        QMessageBox.critical(
+            None,
+            "Help er is iets kapot",
+            "Kan niet starten omdat de server niet reageert, stuur een foto van "
+            "dit naar Maarten: " + repr(info),
+        )
+        return 1
+
+    # Load main window
+    main_window = ActiveStateMainWindow()
+    main_window.show()
+
+    app.exec_()
+
+
+if __name__ == "__main__":
+    main()

piket_server/__init__.py

@@ -41,6 +41,7 @@ class Person(db.Model):
     def as_dict(self) -> dict:
         return {
             "person_id": self.person_id,
+            "active": self.active,
             "name": self.name,
             "consumptions": {
                 ct.consumption_type_id: Consumption.query.filter_by(person=self)