mirror of
https://github.com/Faraphel/Atlas-Install.git
synced 2025-07-03 11:18:26 +02:00
33 lines
968 B
Python
33 lines
968 B
Python
from .Track import *
|
|
from PIL import Image
|
|
from .patch_ct_icon import get_cup_icon
|
|
|
|
|
|
class Cup:
|
|
def __init__(self, name: str,
|
|
track1: Track = EMPTY_TRACK,
|
|
track2: Track = EMPTY_TRACK,
|
|
track3: Track = EMPTY_TRACK,
|
|
track4: Track = EMPTY_TRACK,
|
|
icon: Image = None):
|
|
|
|
self.name = name
|
|
self.tracks = [track1, track2, track3, track4]
|
|
self.icon = icon
|
|
|
|
def get_ctfile_cup(self, race=False):
|
|
"""
|
|
:param race: is it a text used for Race_*.szs ?
|
|
:return: ctfile definition for the cup
|
|
"""
|
|
ctfile_cup = f'\nC "{self.name}"\n'
|
|
for track in self.tracks:
|
|
ctfile_cup += track.get_ctfile_track(race)
|
|
return ctfile_cup
|
|
|
|
def get_icon(self, id: int):
|
|
"""
|
|
:param id: cup number
|
|
:return: icon of the cup
|
|
"""
|
|
return self.icon if self.icon else get_cup_icon(id)
|