traffic.py

#!/usr/bin/python2
import RPi.GPIO as GPIO
from flask import Flask, render_template, request
from time import sleep
from random import randint

app = Flask(__name__)

GPIO.setmode(GPIO.BCM)

# Create a dictionary called pins to store the pin number, name, and pin state:
pins = {
   2 : {'name' : 'red', 'state' : GPIO.LOW},
   3 : {'name' : 'yellow', 'state' : GPIO.LOW},
   4 : {'name' : 'green', 'state' : GPIO.LOW},
   }


# Set each pin as an output and make it low:
for pin in pins:
   GPIO.setup(pin, GPIO.OUT)
   GPIO.output(pin, GPIO.LOW)

def humanize_pin_state(pins):
    for pin_num, pin_data in pins.items():
        if pin_data['state'] == GPIO.LOW:
            pin_data['state'] = 'off'
        else:
            pin_data['state'] = 'on'
    return pins

@app.route("/")
def main():
   # For each pin, read the pin state and store it in the pins dictionary:
   for pin in pins:
      pins[pin]['state'] = GPIO.input(pin)
   # Put the pin dictionary into the template data dictionary:
   templateData = {
      'pins' : humanize_pin_state(pins)
      }
   # Pass the template data into the template main.html and return it to the user
   return render_template('main.html', **templateData)

# The function below is executed when someone requests a URL with the pin number and action in it:
@app.route("/<changePin>/<action>")
def action(changePin, action):
   # Convert the pin from the URL into an integer:
   changePin = int(changePin)
   # Get the device name for the pin being changed:
   deviceName = pins[changePin]['name']
   # If the action part of the URL is "on," execute the code indented below:
   if action == "on":
      # Set the pin high:
      GPIO.output(changePin, GPIO.HIGH)
      # Save the status message to be passed into the template:
      message = "Turned " + deviceName + " on."
   if action == "off":
      GPIO.output(changePin, GPIO.LOW)
      message = "Turned " + deviceName + " off."
   if action == "toggle":
      # Read the pin and set it to whatever it isn't (that is, toggle it):
      GPIO.output(changePin, not GPIO.input(changePin))
      message = "Toggled " + deviceName + "."
   if action == "party":
	for i in range(1,20):
      		GPIO.output(changePin, GPIO.HIGH)
      		message = "Turned " + deviceName + " on."

   # For each pin, read the pin state and store it in the pins dictionary:
   for pin in pins:
      pins[pin]['state'] = GPIO.input(pin)

   # Along with the pin dictionary, put the message into the template data dictionary:
   templateData = {
      'message' : message,
      'pins' : humanize_pin_state(pins)
   }

   return render_template('main.html', **templateData)

@app.route("/rager")
def partyHard(): #Add untested partyHard mode
    countDown()
    sleep(.25)
    iterations, max_iterations, prev = 0, 5, 0
    while True:
        if iterations < max_iterations: iterations = iterations + 1
        else: break
        prev = rager(randint(2,4), prev)
        sleep(.5)
    return render_template('main.html', **templateData)


def blinkyBlink(pin):
    GPIO.output(pin, GPIO.high)
    GPIO.output(pin, GPIO.low)

def goLow():
    for i in pins: GPIO.output(i, GPIO.low)

def countDown():
    goLow()
    sleep(.25)
    for i in pins:
        for t in range(0,3):
        GPIO.output(i, GPIO.high)
        sleep(.75)
        GPIO.output(i, GPIO.low)

def rager(curr, prev):
    for i in range(1,randint(4,16)):
        curr = curr if curr != prev else randint(2,4) #give it a second shot of being something else for a little more fun.
        blinkyBlink(curr)
    return curr

if __name__ == "__main__":
   app.run(host='0.0.0.0', port=80, debug=True)