import time
time.sleep(0.1) # Wait for USB to become ready

from machine import Pin
from utime import sleep

BUTTON_PIN = 0

button = Pin(BUTTON_PIN, Pin.IN, Pin.PULL_UP)

# G = 3
# F = 4
# A = 5
# B = 6
# C = 14
# D = 13
# E = 12

G_ = Pin(3, Pin.OUT)
F_ = Pin(4, Pin.OUT)
A_ = Pin(5, Pin.OUT)
B_ = Pin(6, Pin.OUT)
C_ = Pin(14, Pin.OUT)
D_ = Pin(13, Pin.OUT)
E_ = Pin(12, Pin.OUT)

LED_PIN = [3, 4, 5, 6, 14, 13, 12]

def getButtonState():
  if button.value() == 0:
    time.sleep(0.5)
    return True
  else:
    return False

def setLED(pin_n, toggle):
  led = Pin(pin_n, Pin.OUT)
  if toggle:
    led.on()
  else:
    led.off()

def allOff():
  for x in LED_PIN:
    setLED(x, False)


# G F A B C D E

# G = 3
# F = 4
# A = 5
# B = 6
# C = 14
# D = 13
# E = 12
number = ["0111111", "0001100", "1011011", "1011110", "1101100", "1110110","1110111","0011100","1111111","1111100"]
def segmentDisplay(number_):
  setLED(G_, int(number[number_][0]))
  setLED(F_, int(number[number_][1]))
  setLED(A_, int(number[number_][2]))
  setLED(B_, int(number[number_][3]))
  setLED(C_, int(number[number_][4]))
  setLED(D_, int(number[number_][5]))
  setLED(E_, int(number[number_][6]))

n = 0
while True:
  if getButtonState():
    segmentDisplay(n)
    n += 1
    if n == 10:
      n = 0