rename lan

src/ezgg_lan_manager/pages/SeatingPlanPage.py

@@ -0,0 +1,151 @@
+import logging
+from asyncio import sleep
+from decimal import Decimal
+from typing import Optional
+
+from rio import Text, Column, TextStyle, Component, event, PressEvent, ProgressCircle
+
+from src.ezgg_lan_manager import ConfigurationService, SeatingService, TicketingService, UserService
+from src.ezgg_lan_manager.components.MainViewContentBox import MainViewContentBox
+from src.ezgg_lan_manager.components.SeatingPlan import SeatingPlan, SeatingPlanLegend
+from src.ezgg_lan_manager.components.SeatingPlanInfoBox import SeatingPlanInfoBox
+from src.ezgg_lan_manager.components.SeatingPurchaseBox import SeatingPurchaseBox
+from src.ezgg_lan_manager.services.SeatingService import NoTicketError, SeatNotFoundError, WrongCategoryError, SeatAlreadyTakenError
+from src.ezgg_lan_manager.types.Seat import Seat
+from src.ezgg_lan_manager.types.SessionStorage import SessionStorage
+from src.ezgg_lan_manager.types.User import User
+
+logger = logging.getLogger(__name__.split(".")[-1])
+
+
+class SeatingPlanPage(Component):
+    seating_info: Optional[list[Seat]] = None
+    current_seat_id: Optional[str] = None
+    current_seat_occupant: Optional[str] = None
+    current_seat_price: Decimal = Decimal("0")
+    current_seat_is_blocked: bool = False
+    user: Optional[User] = None
+    show_info_box: bool = True
+    show_purchase_box: bool = False
+    purchase_box_loading: bool = False
+    purchase_box_success_msg: Optional[str] = None
+    purchase_box_error_msg: Optional[str] = None
+    seating_info_text = ""
+    is_booking_blocked: bool = False
+
+    @event.on_populate
+    async def on_populate(self) -> None:
+        await self.session.set_title(f"{self.session[ConfigurationService].get_lan_info().name} - Sitzplan")
+        self.seating_info = await self.session[SeatingService].get_seating()
+        self.user = await self.session[UserService].get_user(self.session[SessionStorage].user_id)
+        if not self.user:
+            self.is_booking_blocked = True
+        else:
+            for seat in self.seating_info:
+                if not seat.user or not self.user:
+                    continue
+                if seat.user.user_id == self.user.user_id:
+                    self.is_booking_blocked = True
+
+    async def on_seat_clicked(self, seat_id: str, _: PressEvent) -> None:
+        self.seating_info_text = ""
+        self.show_info_box = True
+        self.show_purchase_box = False
+        seat = next(filter(lambda s: s.seat_id == seat_id, self.seating_info), None)
+        if not seat:
+            return
+        self.current_seat_is_blocked = seat.is_blocked
+        self.current_seat_id = seat.seat_id
+        ticket_info = self.session[TicketingService].get_ticket_info_by_category(seat.category)
+        price = Decimal("0") if not ticket_info else ticket_info.price
+        self.current_seat_price = price
+        if seat.user:
+            self.current_seat_occupant = seat.user.user_name
+        else:
+            self.current_seat_occupant = None
+
+    async def on_info_clicked(self, text: str) -> None:
+        self.show_info_box = True
+        self.show_purchase_box = False
+        self.current_seat_id = None
+        self.seating_info_text = text
+
+    def set_error(self, msg: str) -> None:
+        self.purchase_box_error_msg = msg
+        self.purchase_box_success_msg = None
+
+    def set_success(self, msg: str) -> None:
+        self.purchase_box_error_msg = None
+        self.purchase_box_success_msg = msg
+
+    async def on_purchase_clicked(self) -> None:
+        self.show_info_box = False
+        self.show_purchase_box = True
+
+    async def on_purchase_confirmed(self) -> None:
+        self.purchase_box_loading = True
+        self.force_refresh()
+        await sleep(0.5)
+        try:
+            await self.session[SeatingService].seat_user(self.user.user_id, self.current_seat_id)
+        except (NoTicketError, WrongCategoryError):
+            self.set_error("Du besitzt kein gültiges Ticket für diesen Platz")
+        except SeatNotFoundError:
+            self.set_error("Der angegebene Sitzplatz existiert nicht")
+        except SeatAlreadyTakenError:
+            self.set_error("Dieser Platz ist bereits vergeben")
+        except Exception as e:
+            self.set_error("Ein unbekannter Fehler ist aufgetreten")
+            logger.error(e)
+        else:
+            self.set_success("Platz erfolgreich gebucht!")
+        self.purchase_box_loading = False
+        await self.on_populate()
+
+    async def on_purchase_cancelled(self) -> None:
+        self.purchase_box_loading = False
+        self.show_info_box = True
+        self.show_purchase_box = False
+        self.purchase_box_error_msg = None
+        self.purchase_box_success_msg = None
+
+    def build(self) -> Component:
+        if not self.seating_info:
+            return Column(
+                MainViewContentBox(
+                    ProgressCircle(
+                        color="secondary",
+                        align_x=0.5,
+                        margin_top=2,
+                        margin_bottom=2
+                    )
+                ),
+                align_y=0
+            )
+        return Column(
+            MainViewContentBox(
+                Column(
+                    SeatingPlanInfoBox(seat_id=self.current_seat_id, seat_occupant=self.current_seat_occupant, seat_price=self.current_seat_price,
+                                       is_blocked=self.current_seat_is_blocked, is_booking_blocked=self.is_booking_blocked, show=self.show_info_box,
+                                       purchase_cb=self.on_purchase_clicked, override_text=self.seating_info_text),
+                    SeatingPurchaseBox(
+                        show=self.show_purchase_box,
+                        seat_id=self.current_seat_id,
+                        is_loading=self.purchase_box_loading,
+                        confirm_cb=self.on_purchase_confirmed,
+                        cancel_cb=self.on_purchase_cancelled,
+                        error_msg=self.purchase_box_error_msg,
+                        success_msg=self.purchase_box_success_msg
+                    )
+                )
+            ),
+            MainViewContentBox(
+                SeatingPlan(seat_clicked_cb=self.on_seat_clicked, seating_info=self.seating_info, info_clicked_cb=self.on_info_clicked) if self.seating_info else
+                Column(ProgressCircle(color=self.session.theme.secondary_color, margin=3),
+                       Text("Sitzplan wird geladen", style=TextStyle(fill=self.session.theme.neutral_color), align_x=0.5, margin=1))
+            ),
+            MainViewContentBox(
+                SeatingPlanLegend(),
+            ),
+            align_y=0
+        )