Fix airlifts always using player country.

This commit is contained in:
Dan Albert 2021-05-25 15:08:03 -07:00
parent e49da6afd6
commit 6b30f47588
2 changed files with 6 additions and 1 deletions

View File

@ -205,6 +205,11 @@ class Game:
return self.player_faction return self.player_faction
return self.enemy_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: def bullseye_for(self, player: bool) -> Bullseye:
if player: if player:
return self.blue_bullseye return self.blue_bullseye

View File

@ -253,7 +253,7 @@ class AirliftPlanner:
flight = Flight( flight = Flight(
self.package, self.package,
self.game.player_country, self.game.country_for(inventory.control_point.captured),
unit_type, unit_type,
flight_size, flight_size,
FlightType.TRANSPORT, FlightType.TRANSPORT,