from enum import Enum

from stale import *


class ZawartoscPola(Enum):
    PUSTE = 0
    SCIANA = 1
    SCIANA2 = 2
    SYGNALIZATOR_CZERWONY = 3
    SYGNALIZATOR_ZOLTY = 4
    SYGNALIZATOR_ZIELONY = 5


class Swiatlo(Enum):
    CZERWONE = 0
    CZERWONE_Z_ZOLTYM = 1
    ZIELONE = 2
    ZOLTE = 3


ZawartoscPolaNaKolorPola = {
    ZawartoscPola.PUSTE: CZARNY,
    ZawartoscPola.SCIANA: JASNOSZARY,
    ZawartoscPola.SCIANA2: SZARY,
    ZawartoscPola.SYGNALIZATOR_CZERWONY: CZERWONY,
    ZawartoscPola.SYGNALIZATOR_ZOLTY: ZOLTY,
    ZawartoscPola.SYGNALIZATOR_ZIELONY: ZIELONY
}