Fix bug introduced in improvements to livery-selector

This will fix the bug I missed where the squadron needs to be using the livery-set before it could take effect for flight-members.
This commit is contained in:
Raffson 2024-08-03 20:40:57 +02:00
parent 532515fad7
commit d2547def51
No known key found for this signature in database
GPG Key ID: B0402B2C9B764D99

View File

@ -27,18 +27,17 @@ class AircraftPainter:
def livery_from_squadron(self) -> Optional[str]:
return self.flight.squadron.livery
def livery_from_squadron_set(self) -> Optional[str]:
def livery_from_squadron_set(self, member_uses_livery_set: bool) -> Optional[str]:
if not (
self.flight.squadron.livery_set and self.flight.squadron.use_livery_set
self.flight.squadron.livery_set
and (self.flight.squadron.use_livery_set or member_uses_livery_set)
):
return None
return random.choice(self.flight.squadron.livery_set)
def determine_livery(self, member_uses_livery_set: bool) -> Optional[str]:
if (
member_uses_livery_set
and (livery := self.livery_from_squadron_set()) is not None
):
livery = self.livery_from_squadron_set(member_uses_livery_set)
if livery is not None:
return livery
if (livery := self.livery_from_squadron()) is not None:
return livery