From 4ad87aef3e4a055048cca47efc47db122e854ab1 Mon Sep 17 00:00:00 2001 From: MetalStormGhost Date: Fri, 15 Sep 2023 22:24:17 +0300 Subject: [PATCH] Add CJTF factions to the coalitions in Pretense, if they're not being used in the campaign. --- game/pretense/pretensemissiongenerator.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/game/pretense/pretensemissiongenerator.py b/game/pretense/pretensemissiongenerator.py index 033faca4..79f18fd4 100644 --- a/game/pretense/pretensemissiongenerator.py +++ b/game/pretense/pretensemissiongenerator.py @@ -9,7 +9,11 @@ import dcs.lua from dataclasses import field from dcs import Mission, Point from dcs.coalition import Coalition -from dcs.countries import country_dict +from dcs.countries import ( + country_dict, + CombinedJointTaskForcesBlue, + CombinedJointTaskForcesRed, +) from dcs.task import OptReactOnThreat from game.atcdata import AtcData @@ -143,6 +147,12 @@ class PretenseMissionGenerator(MissionGenerator): self.mission.coalition["blue"].add_country(self.p_country) self.mission.coalition["red"].add_country(self.e_country) + # Add CJTF factions to the coalitions, if they're not being used in the campaign + # if CombinedJointTaskForcesBlue not in {self.p_country, self.e_country}: + # self.mission.coalition["blue"].add_country(CombinedJointTaskForcesBlue()) + # if CombinedJointTaskForcesRed not in {self.p_country, self.e_country}: + # self.mission.coalition["red"].add_country(CombinedJointTaskForcesRed()) + belligerents = {self.p_country.id, self.e_country.id} for country_id in country_dict.keys(): if country_id not in belligerents: