from enum import Enum


class STROKE_TYPES(Enum):
    FREESTYLE = "Freestyle"
    BREASTSTROKE = "Breaststroke"
    BUTTERFLY = "Butterfly"
    BACKSTROKE = "Backstroke"
    PARA_MEDLEY = "Para-Medley"
    MEDLEY = "Medley"
    INDIVIDUAL_MEDLEY = "Individual Medley"


STROKE_TYPES_FOR_150IM = (
    STROKE_TYPES.BACKSTROKE.value,
    STROKE_TYPES.BREASTSTROKE.value,
    STROKE_TYPES.FREESTYLE.value,
)

STROKE_TYPES_FOR_200_400IM = (
    STROKE_TYPES.BUTTERFLY.value,
    STROKE_TYPES.BACKSTROKE.value,
    STROKE_TYPES.BREASTSTROKE.value,
    STROKE_TYPES.FREESTYLE.value,
)

STROKE_TYPES_FOR_MEDLEY = (
    STROKE_TYPES.BACKSTROKE.value,
    STROKE_TYPES.BREASTSTROKE.value,
    STROKE_TYPES.BUTTERFLY.value,
    STROKE_TYPES.FREESTYLE.value,
)
