Complete code listing

pymon.py

from neopixel import *
from gpiozero import Button
from time import sleep
import random

green = Button(4)
pink = Button(17)
yellow = Button(27)
blue = Button(22)


LED_COUNT      = 150      # Number of LED pixels.
LED_PIN        = 18      # GPIO pin connected to the pixels (must support PWM!).
LED_FREQ_HZ    = 800000  # LED signal frequency in hertz (usually 800khz)
LED_DMA        = 5       # DMA channel to use for generating signal (try 5)
LED_BRIGHTNESS = 128     # Set to 0 for darkest and 255 for brightest
LED_INVERT     = False   # True to invert the signal (when using NPN transistor level shift)


def neo(color,n):
    if color == "pink":
        for i in range(150):
            strip.setPixelColor(i,Color(20,255,147))
        strip.show()
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(0,0,0))
        strip.show()
    elif color == "green":
        for i in range(150):
            strip.setPixelColor(i,Color(255,0,0))
        strip.show()
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(0,0,0))
        strip.show()
    elif color == "blue":
        for i in range(150):
            strip.setPixelColor(i,Color(0,0,255))
        strip.show()
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(0,0,0))
        strip.show()
    elif color == "yellow":
        for i in range(150):
            strip.setPixelColor(i,Color(255,255,0))
        strip.show()
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(0,0,0))
        strip.show()
def correct(n):
    for i in range(9):
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(255,0,0))
        strip.show()
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(0,0,0))
        strip.show()
        sleep(n)

def wrong(n):
    for i in range(9):
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(0,255,0))
        strip.show()
        sleep(n)
        for i in range(150):
            strip.setPixelColor(i,Color(0,0,0))
        strip.show()
        sleep(n)



if __name__ == '__main__':
    strip = Adafruit_NeoPixel(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS)
    strip.begin()
    try:
        lights = ['pink','yellow','blue','green']

        while True:
            green.wait_for_press()
            for i in range(16):
                for light in lights:
                    neo(light,0.1)
            random.shuffle(lights)
            print(lights)
            sleep(0.5)
            player = []
            for light in lights:
                neo(light,0.3)    
            for i in range(150):
                strip.setPixelColor(i,Color(0,0,0))
                strip.show()
            while len(player) <4:
                sleep(0.3)
                if green.is_pressed:
                    player.append("green")
                    neo("green",0.1)
                    print(player)
                elif pink.is_pressed:
                    player.append("pink")
                    neo("pink",0.1)
                    print(player)
                elif yellow.is_pressed:
                    player.append("yellow")
                    neo("yellow",0.1)
                    print(player)
                elif blue.is_pressed:
                    player.append("blue")
                    neo("blue",0.1)
                    print(player)        
            for choice in player:
                neo(choice,0.5)
            if lights == player:
                print("CORRECT")
                correct(0.1)
                for i in range(150):
                    strip.setPixelColor(i,Color(0,0,0))
                strip.show()
            else:
                print("WRONG ANSWER McFLY")
                wrong(0.1)
                for i in range(150):
                    strip.setPixelColor(i,Color(0,0,0))
                strip.show()            
    except KeyboardInterrupt:
        for i in range(360):
            strip.setPixelColor(i,Color(0,0,0))
        strip.show()