84 lines
3.1 KiB
Python
84 lines
3.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from warchron.constants import ContextType
|
||
|
|
from warchron.model.tie_manager import ResolutionContext
|
||
|
|
from warchron.model.war import War
|
||
|
|
from warchron.model.campaign import Campaign
|
||
|
|
from warchron.model.round import Round
|
||
|
|
|
||
|
|
|
||
|
|
class ClosureService:
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def close_round(round: Round) -> List[ResolutionContext]:
|
||
|
|
if not round.all_battles_finished():
|
||
|
|
raise RuntimeError("All battles must be finished to close their round")
|
||
|
|
ties = []
|
||
|
|
for battle in round.battles.values():
|
||
|
|
if battle.is_draw():
|
||
|
|
participants: list[str] = []
|
||
|
|
if battle.player_1_id is not None:
|
||
|
|
participants.append(battle.player_1_id)
|
||
|
|
if battle.player_2_id is not None:
|
||
|
|
participants.append(battle.player_2_id)
|
||
|
|
ties.append(
|
||
|
|
ResolutionContext(
|
||
|
|
context_type=ContextType.BATTLE,
|
||
|
|
context_id=battle.sector_id,
|
||
|
|
# TODO ref to War.participants at some point
|
||
|
|
participant_ids=participants,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
if ties:
|
||
|
|
return ties
|
||
|
|
round.is_over = True
|
||
|
|
return []
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def close_campaign(campaign: Campaign) -> List[ResolutionContext]:
|
||
|
|
if not campaign.all_rounds_finished():
|
||
|
|
raise RuntimeError("All rounds must be finished to close their campaign")
|
||
|
|
ties: List[ResolutionContext] = []
|
||
|
|
# for round in campaign.rounds:
|
||
|
|
# # compute score
|
||
|
|
# # if participants have same score
|
||
|
|
# ties.append(
|
||
|
|
# ResolutionContext(
|
||
|
|
# context_type=ContextType.CAMPAIGN,
|
||
|
|
# context_id=campaign.id,
|
||
|
|
# participant_ids=[
|
||
|
|
# # TODO ref to War.participants at some point
|
||
|
|
# campaign.participants[campaign_participant_id],
|
||
|
|
# campaign.participants[campaign_participant_id],
|
||
|
|
# ],
|
||
|
|
# )
|
||
|
|
# )
|
||
|
|
if ties:
|
||
|
|
return ties
|
||
|
|
campaign.is_over = True
|
||
|
|
return []
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def close_war(war: War) -> List[ResolutionContext]:
|
||
|
|
if not war.all_campaigns_finished():
|
||
|
|
raise RuntimeError("All campaigns must be finished to close their war")
|
||
|
|
ties: List[ResolutionContext] = []
|
||
|
|
# for campaign in war.campaigns:
|
||
|
|
# # compute score
|
||
|
|
# # if participants have same score
|
||
|
|
# ties.append(
|
||
|
|
# ResolutionContext(
|
||
|
|
# context_type=ContextType.WAR,
|
||
|
|
# context_id=war.id,
|
||
|
|
# participant_ids=[
|
||
|
|
# war.participants[war_participant_id],
|
||
|
|
# war.participants[war_participant_id],
|
||
|
|
# ],
|
||
|
|
# )
|
||
|
|
# )
|
||
|
|
if ties:
|
||
|
|
return ties
|
||
|
|
war.is_over = True
|
||
|
|
return []
|