from machine import Pin
from utime import sleep
from SEVEN import Display

display_pins = (0, 4, 5, 17, 16, 2, 15) 
display7 = Display(display_pins)

def main():
    while True:
        # CASO 0
        print("--- EJECUTANDO CASO 0 ---")
        for _ in range(5):
            for i in range(6):
                display7.display[i].on()
                sleep(1)
                display7.display[i].off()
        
        for _ in range(3):
            for i in range(5, -1, -1):
                display7.display[i].on()
                sleep(1)
                display7.display[i].off()
        # CASO 1
        print("--- EJECUTANDO CASO 1 ---")
        for _ in range(4):
            for i in range(7):
                display7.display[i].off()
            
            for i in range(6):
                display7.display[i].on()
                sleep(1)
            
            for i in range(5, -1, -1):
                display7.display[i].off()
                sleep(1)

        # CASO 2
        print("--- EJECUTANDO CASO 2 ---")
        for _ in range(3):
            for numero in range(10):
                display7.show(numero)
                sleep(1)
            
            for numero in range(9, -1, -1):
                display7.show(numero)
                sleep(1)
        
        for _ in range(4):
            display7.display[6].on()
            sleep(1)
            display7.display[6].off()
            sleep(0.5)

        # CASO 3
        print("--- EJECUTANDO CASO 3 ---")
        for _ in range(3):
            for numero in range(0, 9, 2):
                display7.show(numero)
                sleep(1)
            
            for numero in range(9, 0, -1):
                display7.show(numero)
                sleep(1)

        
        for i in range(7):
            display7.display[i].off()
        sleep(2)


if __name__ == "__main__":
    main()