warchron_app/src/warchron/model/campaign.py

from __future__ import annotations
from uuid import uuid4 
from datetime import datetime

from warchron.model.round import Round

class Campaign:
    def __init__(self, name: str, month: int):
        self.id: str = str(uuid4())
        self.name: str = name
        self.month: int = month
        self.participants = {}
        self.sectors = {}
        self.rounds = []
        self.is_over = False

    def set_id(self, new_id: str):
        self.id = new_id

    def set_name(self, new_name: str):
        self.name = new_name

    def set_month(self, new_month: int):
        self.month = new_month
        
    def set_state(self, new_state: bool):
        self.is_over = new_state

    def toDict(self):
        return {
            "id" : self.id,
            "name" : self.name,
            "month" : self.month,
            # "participants" : self.participants,
            "rounds": [rnd.toDict() for rnd in self.rounds],
            "is_over": self.is_over
        }
    
    @staticmethod
    def fromDict(data: dict):
        camp = Campaign(name=data["name"], month=data["month"])
        camp.set_id(data["id"])
        # camp.participants = data.get("participants", {})
        for rnd_data in data.get("rounds", []):
            camp.rounds.append(Round.fromDict(rnd_data))
        camp.set_state(data.get("is_over", False))
        return camp

    def has_round(self, round_id: str) -> bool:
        return any(r.id == round_id for r in self.rounds)
    
    def get_round(self, round_id: str) -> Round:
        return self.rounds[round_id]

    def get_all_rounds(self) -> list[Round]:
        return list(self.rounds)
    
    def add_round(self) -> Round:
        round = Round()
        self.rounds.append(round)
        return round
    
    def remove_round(self, round_id: str):
        rnd = next((r for r in self.rounds if r.id == round_id), None)
        if rnd:
            self.rounds.remove(rnd)

    def get_round_index(self, round_id: str) -> int:
        for index, rnd in enumerate(self.rounds, start=1):
            if rnd.id == round_id:
                return index
        raise KeyError("Round not found in campaign")

class CampaignParticipant:
    def __init__(self,war_participant_id: str, leader: str):
        self.id: str = war_participant_id # ref to War.participants
        self.leader: str = leader
        self.victory_points = 0
        self.objective_points = {}

class Sector:
    def __init__(self, name: str, round_id: str, major_id: str, minor_id: str, influence_id: str):
        self.id: str = str(uuid4())
        self.name: str = name
        self.major_objective_id: str = major_id
        self.minor_objective_id: str = minor_id
        self.influence_objective_id: str = influence_id
        self.round_id: str = round_id