From d581af3b8a69db418aa3a7e0b5158c6362d32519 Mon Sep 17 00:00:00 2001 From: Raffson Date: Sun, 24 Sep 2023 22:14:26 +0200 Subject: [PATCH] Remove flights with package not in ATO in migrator --- game/migrator.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/game/migrator.py b/game/migrator.py index b603a258..eaadd02a 100644 --- a/game/migrator.py +++ b/game/migrator.py @@ -99,12 +99,18 @@ class Migrator: try_set_attr(layout, "nav_from", []) def _update_flights(self) -> None: + to_remove = [] for f in self.game.db.flights.objects.values(): try_set_attr(f, "frequency") try_set_attr(f, "tacan") try_set_attr(f, "tcn_name") try_set_attr(f, "fuel", f.unit_type.max_fuel) - self._update_flight_plan(f) + if f.package in f.squadron.coalition.ato.packages: + self._update_flight_plan(f) + else: + to_remove.append(f.id) + for fid in to_remove: + self.game.db.flights.remove(fid) def _release_untasked_flights(self) -> None: for cp in self.game.theater.controlpoints: