add/edit/delete sectors and participants in campaign

src/warchron/model/model.py

@@ -5,7 +5,7 @@ from datetime import datetime
 
 from warchron.model.player import Player
 from warchron.model.war import War, Objective, WarParticipant
-from warchron.model.campaign import Campaign
+from warchron.model.campaign import Campaign, Sector, CampaignParticipant
 from warchron.model.round import Round
 
 class Model:
@@ -156,7 +156,7 @@ class Model:
             if not war.has_participant(player.id)
         ]
 
-    def add_war_participant(self, war_id: str, player_id: str, faction: str) -> Objective:
+    def add_war_participant(self, war_id: str, player_id: str, faction: str) -> WarParticipant:
         war = self.get_war(war_id)
         return war.add_war_participant(player_id, faction)
 
@@ -198,7 +198,21 @@ class Model:
             if camp is not None:
                 return camp
         raise KeyError(f"Round {round_id} not found")
-    
+
+    def get_campaign_by_campaign_participant(self, participant_id: str) -> Campaign:
+        for war in self.wars.values():
+            camp = war.get_campaign_by_campaign_participant(participant_id)
+            if camp is not None:
+                return camp
+        raise KeyError(f"Participant {participant_id} not found")
+
+    def get_campaign_by_sector(self, sector_id: str) -> Campaign:
+        for war in self.wars.values():
+            camp = war.get_campaign_by_sector(sector_id)
+            if camp is not None:
+                return camp
+        raise KeyError(f"Sector {sector_id} not found")
+
     def update_campaign(self, campaign_id: str, *, name: str, month: int):
         war = self.get_war_by_campaign(campaign_id)
         war.update_campaign(campaign_id, name=name, month=month)
@@ -207,16 +221,64 @@ class Model:
         war = self.get_war_by_campaign(campaign_id)
         war.remove_campaign(campaign_id)
 
+# Sector methods
+
+    def add_sector(self, campaign_id: str, name: str, round_id: str, major_id: str, minor_id: str, influence_id: str) -> Sector:
+        camp = self.get_campaign(campaign_id)
+        return camp.add_sector(name, round_id, major_id, minor_id, influence_id)
+
+    def get_sector(self, sector_id) -> Sector:
+        for war in self.wars.values():
+            for camp in war.campaigns:
+                for sect in camp.sectors.values():
+                    if sect.id == sector_id:
+                        return sect
+        raise KeyError("Sector not found")
+
+    def update_sector(self, sector_id: str, *, name: str, round_id: str, major_id: str, minor_id: str, influence_id: str):
+        camp = self.get_campaign_by_sector(sector_id)
+        camp.update_sector(sector_id, name=name, round_id=round_id, major_id=major_id, minor_id=minor_id, influence_id=influence_id)
+
+    def remove_sector(self, sector_id: str):
+        camp = self.get_campaign_by_sector(sector_id)
+        camp.remove_sector(sector_id)
+
+# Campaign participant methods
+
+    def get_available_war_participants(self, campaign_id: str) -> list[WarParticipant]:
+        war = self.get_war_by_campaign(campaign_id)
+        return war.get_available_war_participants(campaign_id)
+
+    def add_campaign_participant(self, camp_id: str, player_id: str, leader: str, theme: str) -> CampaignParticipant:
+        camp = self.get_campaign(camp_id)
+        return camp.add_campaign_participant(player_id, leader, theme)
+
+    def get_campaign_participant(self, participant_id) -> CampaignParticipant:
+        for war in self.wars.values():
+            for camp in war.campaigns:
+                for part in camp.participants.values():
+                    if part.id == participant_id:
+                        return part
+        raise KeyError("Participant not found")
+
+    def update_campaign_participant(self, participant_id: str, *, leader: str, theme: str):
+        camp = self.get_campaign_by_campaign_participant(participant_id)
+        camp.update_campaign_participant(participant_id, leader=leader, theme=theme)
+
+    def remove_campaign_participant(self, participant_id: str):
+        camp = self.get_campaign_by_campaign_participant(participant_id)
+        camp.remove_campaign_participant(participant_id)
+
 # Round methods
 
     def add_round(self, campaign_id: str) -> Round:
-        campaign = self.get_campaign(campaign_id)
-        return campaign.add_round()
+        camp = self.get_campaign(campaign_id)
+        return camp.add_round()
     
     def get_round(self, round_id: str) -> Round:
         for war in self.wars.values():
-            for campaign in war.campaigns:
-                for rnd in campaign.rounds:
+            for camp in war.campaigns:
+                for rnd in camp.rounds:
                     if rnd.id == round_id:
                         return rnd
         raise KeyError("Round not found")