improve draw & battle display

src/warchron/view/battle_dialog.py

@@ -0,0 +1,74 @@
+from typing import cast, List
+
+from PyQt6.QtWidgets import QWidget, QDialog
+
+from warchron.controller.dtos import ParticipantOption, SectorDTO
+from warchron.view.helpers import select_if_exists
+from warchron.view.ui.ui_battle_dialog import Ui_battleDialog
+
+
+class BattleDialog(QDialog):
+    def __init__(
+        self,
+        parent: QWidget | None = None,
+        *,
+        sectors: List[SectorDTO],
+        default_sector_id: str | None = None,
+        players: List[ParticipantOption],
+        default_player_1_id: str | None = None,
+        default_player_2_id: str | None = None,
+        default_winner_id: str | None = None,
+        default_score: str | None = None,
+        default_victory_condition: str | None = None,
+        default_comment: str | None = None,
+    ) -> None:
+        super().__init__(parent)
+        self.ui: Ui_battleDialog = Ui_battleDialog()
+        self.ui.setupUi(self)  # type: ignore
+        for sect in sectors:
+            self.ui.sectorComboBox.addItem(sect.name, sect.id)
+        select_if_exists(self.ui.sectorComboBox, default_sector_id)
+        self.ui.sectorComboBox.setEnabled(False)
+
+        self.ui.player1ComboBox.addItem("(none)", None)
+        self.ui.player2ComboBox.addItem("(none)", None)
+        for play in players:
+            self.ui.player1ComboBox.addItem(play.name, play.id)
+            self.ui.player2ComboBox.addItem(play.name, play.id)
+        select_if_exists(self.ui.player1ComboBox, default_player_1_id)
+        select_if_exists(self.ui.player2ComboBox, default_player_2_id)
+        self.ui.winnerComboBox.addItem("(none)", None)
+        for play in players:
+            if play.id in (default_player_1_id, default_player_2_id):
+                self.ui.winnerComboBox.addItem(play.name, play.id)
+        select_if_exists(self.ui.winnerComboBox, default_winner_id)
+        self.ui.score.setText(default_score)
+        self.ui.victoryCondition.setText(default_victory_condition)
+        self.ui.battleComment.setPlainText(default_comment)
+
+    def get_sector_id(self) -> str:
+        return cast(str, self.ui.sectorComboBox.currentData())
+
+    def get_player_1_id(self) -> str | None:
+        text = cast(str, self.ui.player1ComboBox.currentData())
+        return text if text else None
+
+    def get_player_2_id(self) -> str | None:
+        text = cast(str, self.ui.player2ComboBox.currentData())
+        return text if text else None
+
+    def get_winner_id(self) -> str | None:
+        text = cast(str, self.ui.winnerComboBox.currentData())
+        return text if text else None
+
+    def get_score(self) -> str | None:
+        text = self.ui.score.text().strip()
+        return text if text else None
+
+    def get_victory_condition(self) -> str | None:
+        text = self.ui.victoryCondition.text().strip()
+        return text if text else None
+
+    def get_comment(self) -> str | None:
+        text = self.ui.battleComment.toPlainText().strip()
+        return text if text else None