from machine import Pin
from time import sleep
import random

relay_pin = 28
relay_pin2 = 0
relay_pin3 = 1


relay = Pin(relay_pin, Pin.OUT)
relay2 = Pin(relay_pin2, Pin.OUT)
relay3 = Pin(relay_pin3, Pin.OUT)

def relay_on(relayv):  #Function relay on
   relayv.value(1)


def relay_off(relayv): #Function relay off
   relayv.value(0)



while True:
    light = random.randint(2,3)
    relayuse = relay
    print(light)
    for x in range(light):
        relay_on(relay)
        sleep(0.4) 
        relay_off(relay)
        sleep(0.4)
        
    if light == 2:
        relayuse = relay2
    else:
        relayuse = relay3

    for x in range(light):
        relay_on(relayuse)
        sleep(0.4) 
        relay_off(relayuse)
        sleep(0.4)

    sleep(2)