add campaign & round

src/warchron/view/view.py

@@ -1,18 +1,49 @@
 from pathlib import Path
+import calendar
 
 from PyQt6 import QtWidgets
+from PyQt6.QtCore import Qt
 from PyQt6.QtWidgets import QDialog, QFileDialog, QTreeWidgetItem
 from PyQt6.QtGui import QCloseEvent
 
 from warchron.view.ui.ui_main_window import Ui_MainWindow
 from warchron.view.ui.ui_player_dialog import Ui_playerDialog
 from warchron.view.ui.ui_war_dialog import Ui_warDialog
+from warchron.view.ui.ui_campaign_dialog import Ui_campaignDialog
+
+ROLE_TYPE = Qt.ItemDataRole.UserRole
+ROLE_ID = Qt.ItemDataRole.UserRole + 1
 
 class View(QtWidgets.QMainWindow, Ui_MainWindow):
     def __init__(self, parent=None):
         super(View, self).__init__(parent)
         self.setupUi(self)
         self.on_close_callback = None
+        self.on_selection_changed = None
+        self.on_add_campaign = None
+        self.on_add_round = None
+        self.warsTree.currentItemChanged.connect(self._emit_selection_changed)
+        self.addCampaignBtn.clicked.connect(self._on_add_campaign_clicked)
+        self.addRoundBtn.clicked.connect(self._on_add_round_clicked)
+
+    def _emit_selection_changed(self, current, previous):
+        if not self.on_tree_selection_changed:
+            return
+        if not current:
+            self.on_tree_selection_changed(None)
+            return
+        self.on_tree_selection_changed({
+            "type": current.data(0, ROLE_TYPE),
+            "id": current.data(0, ROLE_ID),
+        })
+        
+    def _on_add_campaign_clicked(self):
+        if self.on_add_campaign:
+            self.on_add_campaign()
+
+    def _on_add_round_clicked(self):
+        if self.on_add_round:
+            self.on_add_round()
 
     def closeEvent(self, event: QCloseEvent):
         if self.on_close_callback:
@@ -55,10 +86,36 @@ class View(QtWidgets.QMainWindow, Ui_MainWindow):
         tree.setHeaderLabels(["Wars"])
         for war in wars:
             war_item = QTreeWidgetItem([f"{war.name} ({war.year})"])
-            war_item.setData(0, 1, war.id)  # role=1 to store id
+            war_item.setData(0, ROLE_TYPE, "war")
+            war_item.setData(0, ROLE_ID, war.id)
             tree.addTopLevelItem(war_item)
+            for camp in war.get_all_campaigns():
+                camp_item = QTreeWidgetItem([f"{camp.name} ({calendar.month_name[camp.month]})"])
+                camp_item.setData(0, ROLE_TYPE, "campaign")
+                camp_item.setData(0, ROLE_ID, camp.id)
+                war_item.addChild(camp_item)
+                for index, rnd in enumerate(camp.get_all_rounds(), start=1):
+                    rnd_item = QTreeWidgetItem([f"Round {index}"])
+                    rnd_item.setData(0, ROLE_TYPE, "round")
+                    rnd_item.setData(0, ROLE_ID, rnd.id)
+                    camp_item.addChild(rnd_item)
         tree.expandAll()
 
+    def get_selected_tree_item(self):
+        item = self.warsTree.currentItem()
+        if not item:
+            return None
+        return {
+            "type": item.data(0, ROLE_TYPE),
+            "id": item.data(0, ROLE_ID)
+        }
+
+    def set_add_campaign_enabled(self, enabled: bool):
+        self.addCampaignBtn.setEnabled(enabled)
+
+    def set_add_round_enabled(self, enabled: bool):
+        self.addRoundBtn.setEnabled(enabled)
+
 class PlayerDialog(QDialog):
     def __init__(self, parent=None):
         super().__init__(parent)
@@ -76,3 +133,12 @@ class WarDialog(QDialog):
 
     def get_war_name(self) -> str:
         return self.ui.warName.text().strip()
+
+class CampaignDialog(QDialog):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.ui = Ui_campaignDialog()
+        self.ui.setupUi(self)
+
+    def get_campaign_name(self) -> str:
+        return self.ui.campaignName.text().strip()