Factor out Coalition from Game.

qt_ui/windows/AirWingDialog.py

@@ -3,12 +3,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import Optional, Iterator
 
-from PySide2.QtCore import (
-    QItemSelectionModel,
-    QModelIndex,
-    Qt,
-    QSize,
-)
+from PySide2.QtCore import QItemSelectionModel, QModelIndex, QSize
 from PySide2.QtWidgets import (
     QAbstractItemView,
     QCheckBox,
@@ -183,7 +178,7 @@ class AirInventoryView(QWidget):
         self.table.setSortingEnabled(True)
 
     def iter_allocated_aircraft(self) -> Iterator[AircraftInventoryData]:
-        for package in self.game_model.game.blue_ato.packages:
+        for package in self.game_model.game.blue.ato.packages:
             for flight in package.flights:
                 yield from AircraftInventoryData.from_flight(flight)
 

qt_ui/windows/basemenu/DepartingConvoysMenu.py

@@ -73,11 +73,15 @@ class DepartingConvoysList(QFrame):
         task_box_layout = QGridLayout()
         scroll_content.setLayout(task_box_layout)
 
-        for convoy in game_model.game.transfers.convoys.departing_from(cp):
+        for convoy in game_model.game.coalition_for(
+            cp.captured
+        ).transfers.convoys.departing_from(cp):
             group_info = DepartingConvoyInfo(convoy)
             task_box_layout.addWidget(group_info)
 
-        for cargo_ship in game_model.game.transfers.cargo_ships.departing_from(cp):
+        for cargo_ship in game_model.game.coalition_for(
+            cp.captured
+        ).transfers.cargo_ships.departing_from(cp):
             group_info = DepartingConvoyInfo(cargo_ship)
             task_box_layout.addWidget(group_info)
 

qt_ui/windows/basemenu/QBaseMenu2.py

@@ -195,7 +195,9 @@ class QBaseMenu2(QDialog):
         ground_unit_limit = self.cp.frontline_unit_count_limit
         deployable_unit_info = ""
 
-        allocated = self.cp.allocated_ground_units(self.game_model.game.transfers)
+        allocated = self.cp.allocated_ground_units(
+            self.game_model.game.coalition_for(self.cp.captured).transfers
+        )
         unit_overage = max(
             allocated.total_present - self.cp.frontline_unit_count_limit, 0
         )

qt_ui/windows/basemenu/airfield/QAircraftRecruitmentMenu.py

@@ -45,7 +45,7 @@ class QAircraftRecruitmentMenu(QFrame, QRecruitBehaviour):
         row = 0
 
         unit_types: Set[AircraftType] = set()
-        for unit_type in self.game_model.game.player_faction.aircrafts:
+        for unit_type in self.game_model.game.blue.faction.aircrafts:
             if self.cp.is_carrier and not unit_type.carrier_capable:
                 continue
             if self.cp.is_lha and not unit_type.lha_capable:

qt_ui/windows/groundobject/QGroundObjectMenu.py

@@ -1,7 +1,6 @@
 import logging
 from typing import List, Optional
 
-from PySide2 import QtCore
 from PySide2.QtGui import Qt
 from PySide2.QtWidgets import (
     QComboBox,
@@ -307,7 +306,7 @@ class QBuyGroupForGroundObjectDialog(QDialog):
         self.buySamBox = QGroupBox("Buy SAM site :")
         self.buyArmorBox = QGroupBox("Buy defensive position :")
 
-        faction = self.game.player_faction
+        faction = self.game.blue.faction
 
         # Sams
 

qt_ui/windows/mission/flight/QFlightCreator.py

@@ -38,7 +38,7 @@ class QFlightCreator(QDialog):
         self.game = game
         self.package = package
         self.custom_name_text = None
-        self.country = self.game.player_country
+        self.country = self.game.blue.country_name
 
         self.setWindowTitle("Create flight")
         self.setWindowIcon(EVENT_ICONS["strike"])
@@ -52,7 +52,6 @@ class QFlightCreator(QDialog):
 
         self.aircraft_selector = QAircraftTypeSelector(
             self.game.aircraft_inventory.available_types_for_player,
-            self.game.player_country,
             self.task_selector.currentData(),
         )
         self.aircraft_selector.setCurrentIndex(0)