from machine import Pin
from time import sleep
import ssd1306
import display_oled


def inicio():
    for fila in range(0,4):
        pines_Filas[fila].value(0)

def escanear(fila, columna):

    pines_Filas[fila].value(1)
    key = 0
    if (pines_Columnas[columna].value()==0):
        key = 0
    if (pines_Columnas[columna].value() == 1):
        key = 1
        while (pines_Columnas[columna].value() == 1):
          pass
    pines_Filas[fila].value(0)
    return key

def obtener(oled, pos, val):
  ingreso=""
  while True:
    for fila in range(4):
        for columna in range(4):
            tecla = escanear(fila, columna)
            if tecla == 1:
              if teclas[fila][columna]=="D":
                return ingreso
                break
              else:
                ingreso=ingreso+teclas[fila][columna]
                if pos==0 and int(ingreso)<=100:
                  oled.text(ingreso, 60, 0)
                elif pos==1 and int(ingreso)<=5 and int(ingreso)!=0:
                  oled.text(ingreso, 60, 15)
                elif pos==2 and int(ingreso)<=80 and int(ingreso)!=0:
                  oled.text(ingreso, 60, 30)
                else:
                  ingreso=ingreso[:-1]
                  display_oled.no_available(oled)
                  display_oled.principal_screen(oled)
                  display_oled.rewrite(oled, val)
                  return 0

                oled.show()


                
            

teclas = [['1', '2', '3', 'A'], ['4', '5', '6', 'B'], ['7', '8', '9', 'C'], ['*', '0', '#', 'D']]
filas = [1,2,42,41]
columnas = [39,38,37,36]
pines_Filas = [Pin(pin_nombre, mode=Pin.OUT) for pin_nombre in filas]
pines_Columnas = [Pin(pin_nombre, mode=Pin.IN, pull=Pin.PULL_DOWN) for pin_nombre in columnas]
inicio()