from machine import Pin
from time import sleep_ms

vPVermelho = Pin(23, Pin.OUT, value=0)
vPAmarelo = Pin(22, Pin.OUT, value=0)
vPVerde = Pin(21, Pin.OUT, value=0)

botao = Pin(33, Pin.IN, Pin.PULL_UP)

def pare():
    vPVermelho.on()
    sleep_ms(2000)

def siga(tempo=5000):
    vPVermelho.off()
    vPVerde.on()
    sleep_ms(tempo)

def atencao():
    vPAmarelo.on()
    vPVerde.off()
    sleep_ms(500)
    vPAmarelo.off()     

estadoAnterior = estadoAtual = botao.value()
t = 5000

while True:
    estadoAtual = botao.value()
    if estadoAtual != estadoAnterior:
        if estadoAtual == 1:
            print('Pressionou')
            t = 1000         
        
        sleep_ms(200)
        estadoAnterior = estadoAtual
        
    pare()    
    
    siga(t)
    t = 5000

    atencao()