# Egg Hunt Monthly Project
# Sample Code
from codex import *
from eggs import *
import random
import ascii_art

egg_list = [egg1, egg2, egg3, egg4, egg5, egg6,
            egg7, egg8, egg9, egg10]
easter_eggs = [surprise1, surprise2,surprise3, surprise4]
hides_an_egg = []

# ---- Main Program ----
# -- select eggs to hide the surprise images
for i in range(len(easter_eggs)):
    egg = random.choice(egg_list)
    hides_an_egg.append(egg)

#-- main loop    
while True:
    if buttons.was_pressed(BTN_A):
        found_egg = random.choice(egg_list)
        draw_egg = ascii_art.get_images(found_egg)
        display.draw_image(draw_egg[0], scale=10)

    if buttons.was_pressed(BTN_B):
        if len(hides_an_egg) == 0:
            display.clear()
            display.print("All eggs", scale=4)
            display.print("found!", scale=4)
        elif found_egg in hides_an_egg:
            reveal = random.choice(easter_eggs)
            reveal_egg = ascii_art.get_images(reveal)
            display.draw_image(reveal_egg[0], scale=10)
            easter_eggs.remove(reveal)
            hides_an_egg.remove(found_egg)