diff --git a/game/game.py b/game/game.py index 770c849a..233c9e20 100644 --- a/game/game.py +++ b/game/game.py @@ -205,6 +205,11 @@ class Game: return self.player_faction return self.enemy_faction + def country_for(self, player: bool) -> str: + if player: + return self.player_country + return self.enemy_country + def bullseye_for(self, player: bool) -> Bullseye: if player: return self.blue_bullseye diff --git a/game/transfers.py b/game/transfers.py index 4bdff773..3ffb3cf2 100644 --- a/game/transfers.py +++ b/game/transfers.py @@ -253,7 +253,7 @@ class AirliftPlanner: flight = Flight( self.package, - self.game.player_country, + self.game.country_for(inventory.control_point.captured), unit_type, flight_size, FlightType.TRANSPORT,