#Modulos
from hcsr04 import HCSR04
from machine import Pin
import time 

T = 2
E = 4

#Objeto HC-SR04 tanque elevado
Sensor = HCSR04(trigger_pin = T, echo_pin = E)

#Leds
Led_VERDE = Pin(5, Pin.OUT)
Led_AMARILLO = Pin(19, Pin.OUT)
Led_ANARANJADO = Pin(21, Pin.OUT)
Led_ROJO = Pin(22, Pin.OUT)

#relay
relay = Pin(18, Pin.OUT)

#definir una funcion para el control de los leds
def leds():
  #LED VERDE
  if DistanciaCM <= 5:
    Led_VERDE.on()
    relay.value(0)
  else:
    Led_VERDE.off()

      #LED AMARILLO
  if 6 < DistanciaCM <= 35:
    Led_AMARILLO.on() 
  else:
    Led_AMARILLO.off()

      #LED ANARANJADO
  if 35 < DistanciaCM <= 45:
    Led_ANARANJADO.on() 
  else:
    Led_ANARANJADO.off()   

      #LED ROJO
  if 45 < DistanciaCM:
    Led_ROJO.on() 
    relay.value(1)
  else:
    Led_ROJO.off()   

#Obtener las distancias 
while True:
  DistanciaCM = Sensor.distance_cm()
  print("Distancia:", DistanciaCM,)


  time.sleep(0.5)
  leds()