factorise ranking icon mapper

src/warchron/controller/campaign_controller.py

@@ -1,15 +1,11 @@
 from typing import List, Dict, Tuple, TYPE_CHECKING
 
 from PyQt6.QtWidgets import QMessageBox, QDialog
-from PyQt6.QtGui import QIcon
 
 from warchron.constants import (
     RefreshScope,
     ContextType,
     ItemType,
-    Icons,
-    IconName,
-    RANK_TO_ICON,
 )
 
 if TYPE_CHECKING:
@@ -28,8 +24,8 @@ from warchron.model.campaign_participant import CampaignParticipant
 from warchron.model.sector import Sector
 from warchron.model.tie_manager import TieContext, TieResolver
 from warchron.model.score_service import ScoreService
-from warchron.model.result_checker import ResultChecker
 from warchron.controller.closure_workflow import CampaignClosureWorkflow
+from warchron.controller.ranking_icon import RankingIcon
 from warchron.view.campaign_dialog import CampaignDialog
 from warchron.view.campaign_participant_dialog import CampaignParticipantDialog
 from warchron.view.sector_dialog import SectorDialog
@@ -40,41 +36,6 @@ class CampaignController:
     def __init__(self, app: "AppController"):
         self.app = app
 
-    def _compute_campaign_ranking_icons(
-        self, war: War, campaign: Campaign
-    ) -> Dict[str, QIcon]:
-        scores = ScoreService.compute_scores(
-            war,
-            ContextType.CAMPAIGN,
-            campaign.id,
-        )
-        ranking = ResultChecker.get_effective_ranking(
-            war, ContextType.CAMPAIGN, campaign.id, scores
-        )
-        icon_map = {}
-        for rank, group, token_map in ranking:
-            base_icon = RANK_TO_ICON.get(rank, IconName.VPNTH)
-            vp = scores[group[0]].victory_points
-            original_group_size = sum(
-                1 for s in scores.values() if s.victory_points == vp
-            )
-            for pid in group:
-                spent = token_map.get(pid, 0)
-                if original_group_size == 1:
-                    icon_name = base_icon
-                elif len(group) == 1:
-                    if spent > 0:
-                        icon_name = getattr(IconName, f"{base_icon.name}BREAK")
-                    else:
-                        icon_name = base_icon
-                else:
-                    if spent > 0:
-                        icon_name = getattr(IconName, f"{base_icon.name}TIEDRAW")
-                    else:
-                        icon_name = getattr(IconName, f"{base_icon.name}DRAW")
-                icon_map[pid] = QIcon(Icons.get_pixmap(icon_name))
-        return icon_map
-
     def _fill_campaign_details(self, campaign_id: str) -> None:
         camp = self.app.model.get_campaign(campaign_id)
         self.app.view.show_campaign_details(name=camp.name, month=camp.month)
@@ -98,7 +59,9 @@ class CampaignController:
         rows: List[CampaignParticipantScoreDTO] = []
         icon_map = {}
         if camp.is_over:
-            icon_map = self._compute_campaign_ranking_icons(war, camp)
+            icon_map = RankingIcon.compute_icons(
+                war, ContextType.CAMPAIGN, campaign_id, scores
+            )
         for camp_part in camp.get_all_campaign_participants():
             war_part_id = camp_part.war_participant_id
             war_part = war.get_war_participant(war_part_id)