from machine import Pin
import utime

# Nastavení pinů pro LED, R 220 ohmů
zelena = Pin(20, Pin.OUT)
oranzova = Pin(19, Pin.OUT)
cervena = Pin(18, Pin.OUT)

def semafor():
    while True:
        # zelena svítí
        zelena.value(1)
        oranzova.value(0)
        cervena.value(0)
        utime.sleep(5)  # zelena svítí 5 sekund
        
        # zelena a oranzova svítí (připravujeme na červenou)
        zelena.value(0)
        oranzova.value(1)
        utime.sleep(2)  # oranzova svítí 2 sekundy
        
        # cervena svítí
        oranzova.value(0)
        cervena.value(1)
        utime.sleep(5)  # cervena svítí 5 sekund
        
        # cervena a oranzova svítí (připravujeme na zelenou)
        oranzova.value(1)
        utime.sleep(2)  # cervena a oranzova svítí 2 sekundy

        # zelená svítí (konec jednoho cyklu)
        cervena.value(0)
        oranzova.value(0)
        zelena.value(1)
        utime.sleep(3)  # cervena a oranzova svítí 2 sekundy
      
        # Vypnutí všech LED před dalším cyklem
        zelena.value(0)
        oranzova.value(0)
        cervena.value(0)
        utime.sleep(1)  # Krátká pauza před dalším cyklem

# Spuštění smyčky semaforu
semafor()