diff --git a/game/ato/flight.py b/game/ato/flight.py index 014c40a4..26b4b00f 100644 --- a/game/ato/flight.py +++ b/game/ato/flight.py @@ -285,7 +285,9 @@ class Flight( return self.__str__() def __str__(self) -> str: - string = f"[{self.flight_type}] {self.count} x {self.unit_type}" + string = ( + f"[{self.flight_type}] {self.count} x {self.unit_type} - {self.start_type}" + ) if self.custom_name: return f"{self.custom_name} - {string}" return string diff --git a/game/ato/starttype.py b/game/ato/starttype.py index e42e6402..e3b3b89c 100644 --- a/game/ato/starttype.py +++ b/game/ato/starttype.py @@ -13,3 +13,6 @@ class StartType(Enum): WARM = "Warm" RUNWAY = "Runway" IN_FLIGHT = "In Flight" + + def __str__(self) -> str: + return f"{self.value}"