From f80b948fb1922a1e47a6da3ffa1d8e445a3be4d0 Mon Sep 17 00:00:00 2001 From: Dan Albert Date: Thu, 20 May 2021 21:06:53 -0700 Subject: [PATCH] Fix loadout downgrading bugs. If the pylon had no weapon this would raise, and if no replacement was found we wouldn't clear the pylon. --- gen/flights/loadouts.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/gen/flights/loadouts.py b/gen/flights/loadouts.py index 2c30954b..c9ae20c4 100644 --- a/gen/flights/loadouts.py +++ b/gen/flights/loadouts.py @@ -35,6 +35,9 @@ class Loadout: new_pylons = dict(self.pylons) for pylon_number, weapon in self.pylons.items(): + if weapon is None: + del new_pylons[pylon_number] + continue if not weapon.available_on(date): pylon = Pylon.for_aircraft(unit_type, pylon_number) for fallback in weapon.fallbacks: @@ -44,6 +47,8 @@ class Loadout: continue new_pylons[pylon_number] = fallback break + else: + del new_pylons[pylon_number] return Loadout(f"{self.name} ({date.year})", new_pylons, date) @classmethod