import board

from kmk.extensions.LED import LED
from kmk.kmk_keyboard import KMKKeyboard as _KMKKeyboard
from kmk.scanners import DiodeOrientation


class KMKKeyboard(_KMKKeyboard):
    col_pins = (
        board.GP2,
        board.GP3,
        board.GP29,
        board.GP27,
        board.GP6,
        board.GP5,
    )
    row_pins = (
        board.GP0,
        board.GP1,
        board.GP7,
        board.GP16,
        board.GP28,
    )

    diode_orientation = DiodeOrientation.COLUMNS

    leds = LED(led_pin=[board.GP27, board.GP28])
    _KMKKeyboard.extensions.append(leds)