lightBox.py

import time
import urllib
from flask import Flask, request, render_template, jsonify, make_response
from gpiozero import Button
from neopixel import *
import signal
import sys
import os

# Turn off pixels at ctrl-c
def signal_handler(signal, frame):
        colorWipe(strip, Color(0,0,0))
        sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)

# LED strip configuration:
LED_COUNT      = 48
LED_PIN        = 18
LED_FREQ_HZ    = 800000
LED_DMA        = 5
LED_BRIGHTNESS = 255
LED_INVERT     = False

# Initialise Flask api
app=Flask(__name__)

# Button configuration
button1 = Button(19)
button2 = Button(16)

# Initial status (off)
status = 0

# Create NeoPixel object, initialise library
strip = Adafruit_NeoPixel(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT)
strip.begin()

# Define functions
def allOn(strip, color):
    	"""Turn all pixels on"""
	for i in range(strip.numPixels()):
		strip.setPixelColor(i, color)
		strip.show()

def allOff():
	"""Turn all pixels off"""
	return allOn(strip, Color(0, 0, 0))

def colorWipe(strip, color, wait_ms=50):
	"""Wipe color across display a pixel at a time."""
	for i in range(strip.numPixels()):
		strip.setPixelColor(i, color)
		strip.show()
		time.sleep(wait_ms/1000.0)

def wheel(pos):
	"""Generate rainbow colors across 0-255 positions."""
	if pos < 85:
		return Color(pos * 3, 255 - pos * 3, 0)
	elif pos < 170:
		pos -= 85
		return Color(255 - pos * 3, 0, pos * 3)
	else:
		pos -= 170
		return Color(0, pos * 3, 255 - pos * 3)

def rainbow(strip, wait_ms=20, iterations=1):
	"""Draw rainbow that fades across all pixels at once."""
	for j in range(256*iterations):
		for i in range(strip.numPixels()):
			strip.setPixelColor(i, wheel((i+j) & 255))
			strip.show()
			time.sleep(wait_ms/1000.0)

def rainbowCycle(strip, wait_ms=20, iterations=5):
	"""Draw rainbow that uniformly distributes itself across all pixels."""
	for j in range(256*iterations):
		for i in range(strip.numPixels()):
			strip.setPixelColor(i, wheel((int(i * 256 / strip.numPixels()) + j) & 255))
			strip.show()
			time.sleep(wait_ms/1000.0)

def get_status():
	global status
	for pixel in range(strip.numPixels()):
        	if strip.getPixelColor(pixel) != (0, 0, 0):
                	status = 1
        	else:
                	status = 0
    		return status

def buttons():
	if button1.is_pressed == True:
        	req = urllib.urlopen('http://192.168.1.108/lightBox/api/v1.0/on')
                time.sleep(0.2)
        elif button2.is_pressed == True:
                req = urllib.urlopen('http://192.168.1.108/lightBox/api/v1.0/off')
                time.sleep(0.2)

# Function to shutdown Pi
def shutdown():
	os.system("sudo shutdown -h now")

# Function to reboot Pi
def reboot():
	os.system("sudo reboot -h now")

# Direct url to front end template
@app.route("/lightBox/api/v1.0/")
def index():
	return render_template('index.html')

# Set up API endpoints
@app.route('/lightBox/api/v1.0/<string:st>', methods=['GET'])
def set_status(st):
	global status, color
    	if st == 'on':
        	status = 1
		return index(), allOn(strip, Color(255, 255, 255))
        elif st == 'blue':
        	status = 1
        	return index(), colorWipe(strip, Color(255, 0, 0))
	elif st == 'red':
		status = 1
		return index(), colorWipe(strip, Color(0, 255, 0))
        elif st == 'green':
		status = 1
		return index(), colorWipe(strip, Color(0, 0, 255))
        elif st == 'cyan':
                status = 1
                return index(), colorWipe(strip, Color(255, 0, 255))
        elif st == 'yellow':
                status = 1
                return index(), colorWipe(strip, Color(0, 255, 255))
        elif st == 'magenta':
                status = 1
                return index(), colorWipe(strip, Color(255, 255, 0))
	elif st == 'orange':
		status = 1
		return index(), colorWipe(strip, Color(0, 255, 125))
	elif st == 'purple':
		status = 1
		return index(), colorWipe(strip, Color(255, 125, 0))
	elif st == 'off':
        	status = 0
		return index(), allOff()
	elif st == 'shutdown':
		status = 0
		return shutdown()
	elif st == 'reboot':
		status = 0
		return reboot()
    	elif st == 'status':
        	status = get_status()
    		return jsonify({'status': status, 'color': color})

# 404 response
@app.errorhandler(404)
def not_found(error):
	return make_response(jsonify({'error': 'Not found'}), 404)

# Main programme logic
if __name__ == "__main__":
	colorWipe(strip, Color(0, 0, 255))
	colorWipe(strip, Color(0, 0, 0))
	app.run(host='192.168.1.108', port=80, debug=True)

	if button1.is_pressed == True:
		req = urllib.urlopen('http://192.168.1.108/lightBox/api/v1.0/on')
		time.sleep(0.2)
	elif button2.is_pressed == True:
		req = urllib.urlopen('http://192.168.1.108/lightBox/api/v1.0/off')
		time.sleep(0.2)