from machine import Pin
from random import randint
from time import sleep

numeros_display= [[1,1,1,1,1,1,0], [0,0,0,0,1,1,0],
                  [1,1,0,1,1,0,1], [1,1,1,1,0,0,1], 
                  [0,1,1,0,0,1,1], [1,0,1,1,0,1,1],
                  [1,0,1,1,1,1,1], [1,1,1,0,0,0,0],
                  [1,1,1,1,1,1,1], [1,1,1,1,0,1,1]]

display= [Pin(pin, Pin.OUT) for pin in (5,6,13,14,15,4,3)]
Button= Pin(0, Pin.IN, Pin.PULL_DOWN)
led= Pin(25, Pin.OUT)

def sortear():

    if Button.value() == 1:
        numero_sorteado= randint(0, 9)
        sleep(1)
        print(f"numero sorteado: {numero_sorteado}")
        sleep(1)
        return numero_sorteado

    else:
        return None

def numero_imprimido():
    numero_sorteado2= sortear()

    if numero_sorteado2 == 0:
        for i, x in enumerate(numeros_display[0]):
            display[i].value(x)

    elif numero_sorteado2 == 1:
        for i, x in enumerate(numeros_display[1]):
            display[i].value(x)

    elif numero_sorteado2 == 2:
        for i, x in enumerate(numeros_display[2]):
            display[i].value(x)

    elif numero_sorteado2 == 3:
        for i, x in enumerate(numeros_display[3]):
            display[i].value(x)

    elif numero_sorteado2 == 4:
        for i, x in enumerate(numeros_display[4]):
            display[i].value(x)

    elif numero_sorteado2 == 5:
        for i, x in enumerate(numeros_display[5]):
            display[i].value(x)

    elif numero_sorteado2 == 6:
        for i, x in enumerate(numeros_display[6]):
            display[i].value(x)
    
    elif numero_sorteado2 == 7:
        for i, x in enumerate(numeros_display[7]):
            display[i].value(x)

    elif numero_sorteado2 == 8:
        for i, x in enumerate(numeros_display[8]):
            display[i].value(x)

    elif numero_sorteado2 == 9:
        for i, x in enumerate(numeros_display[9]):
            display[i].value(x)

while True:
    numero_imprimido()