flaskServer.py

#!/usr/bin/env python

import argparse

from flask import Flask, request, jsonify, Response
import json
import gevent
from gevent.wsgi import WSGIServer
from gevent.queue import Queue

import ProgramModules.sharedObjects as app
from SculptureController import SculptureController

flaskApp = Flask(__name__,  static_folder='jsGui', static_url_path='/jsGui')
sculpture = SculptureController()
serverSentEventStreams = []




class ServerSentEvent(object):

	def __init__(self, data):
		self.data = data
		self.event = None
		self.id = None
		self.desc_map = {
			self.data : "data",
			self.event : "event",
			self.id : "id"
		}

	def encode(self):
		if not self.data:
			return ""
		lines = ["%s: %s" % (v, k)
					 for k, v in self.desc_map.iteritems() if k]

		return "%s\n\n" % "\n".join(lines)




def sendNewOutputState(whatChanged):
	data = {whatChanged : app.messenger.getMessages(whatChanged)}
	for sub in serverSentEventStreams[:]:
		sub.put(json.dumps(data), False)


@flaskApp.route('/')
def jsGui():
	return flaskApp.send_static_file('index.htm')


@flaskApp.route('/getData', methods =['GET', 'POST'])
def getData():
	return jsonify(sculpture.getCurrentStateData())

@flaskApp.route('/doCommand', methods =['POST'])
def doCommand():
	requestData = json.loads(request.data)
	command = requestData[0]
	result = sculpture.doCommand(requestData)
	if command == 'loadSculpture':
		app.messenger.addBinding('outputChanged', sendNewOutputState, ('outputChanged',))
		app.messenger.addBinding('log', sendNewOutputState, ('log',))

	return jsonify({'command' : command, 'result' : result})


@flaskApp.route("/dataStream")
def subscribe():
	def gen():
		q = Queue()
		serverSentEventStreams.append(q)
		try:
			while True:
				result = q.get()
				ev = ServerSentEvent(str(result))
				yield ev.encode()
		except GeneratorExit:
			serverSentEventStreams.remove(q)
	return Response(gen(), mimetype="text/event-stream")

# Testing commands
# sculpture.loadSculpture('tympani')
# app.messenger.addBinding('outputChanged', sendNewOutputState, ('outputChanged',))
# app.messenger.addBinding('log', sendNewOutputState, ('log',))
# sculpture.doCommand(['addPattern', 'poofers', 'AllPoof'])
# sculpture.doCommand(['addPattern', 'poofers', 'Chase'])
# sculpture.doCommand(['addGlobalInput', {'type' : 'pulse', 'subType' : 'audio'}])
# sculpture.doCommand(['addGlobalInput', {'type' : 'multi', 'subType' : 'randomPulse'}])
# sculpture.doCommand(['addGlobalInput', {'type' : 'multi', 'subType' : 'basic', 'number' : 5, 'min' : [0, 0, 1, 0,  1], 'max' : [1, 2, 3, 4,  5], 'description' : ['foo', 'bar', 'baz', 'bam', 'boo']}])
# sculpture.doCommand(["changePatternInputBinding","poofers","poofersPattern0","poofButton",2,0])


if __name__ == '__main__':
	parser = argparse.ArgumentParser(description='FireSculptureController')
	parser.add_argument('--port', type=int, default=8080,
			    help='port for webserver to listen on')
	args = parser.parse_args()
	flaskApp.debug = True
	server = WSGIServer(("", args.port), flaskApp)
	try:
		print 'starting server on localhost:%s' % args.port
		server.serve_forever()
	except Exception, err:
		sculpture.doReset()
		print err