Atlas-Install/source/Cup.py

26 lines
747 B
Python

from .Track import *
from PIL import Image
from .patch_ct_icon import get_cup_icon
class Cup:
def __init__(self, name: str, id: int,
track1: Track = EMPTY_TRACK,
track2: Track = EMPTY_TRACK,
track3: Track = EMPTY_TRACK,
track4: Track = EMPTY_TRACK,
icon: Image = None):
self.name = name
self.track1 = track1
self.track2 = track2
self.track3 = track3
self.track4 = track4
self.icon = icon if icon else create_cup_icon(id)
self.id = id # cup number
def get_ctfile_cup(self):
pass
def get_tracks(self):
return self.track1, self.track2, self.track3, self.track4