kilnController/picoreflowd.py

import threading,time
import json
from oven import Oven

import bottle
from gevent.pywsgi import WSGIServer
from geventwebsocket import WebSocketHandler, WebSocketError

app = bottle.Bottle()
oven = Oven()
oven.start()
wsocks = []
wsocks_control = []

def notifyAll(message):
    message_json = json.dumps(message)
    print "sending to %d clients: %s"%(len(wsocks),message_json)
    for wsock in wsocks:
        if wsock:
            try:
                wsock.send(message_json)
            except:
                print "Could not write to socket!"
                wsocks.remove(wsock)
        else:
            wsocks.remove(wsock)

class OvenWatcher(threading.Thread):
    def __init__(self,oven):
        threading.Thread.__init__(self)
        self.daemon = True
        
        self.oven = oven

    def run(self):
        while True:
            oven_state = self.oven.get_state()
            notifyAll(oven_state)
            time.sleep(1)

ovenWatcher = OvenWatcher(oven)
ovenWatcher.start()

@app.route('/')
def index():
    return bottle.redirect('/picoreflow/index.html')

@app.route('/picoreflow/:filename#.*#')
def send_static(filename):
    return bottle.static_file(filename, root='./public/')

@app.route('/run')
def start_oven():
    oven.run_profile("abc")

    return "Starting"

@app.route('/control')
def handle_control():
    env = bottle.request.environ;
    print env
    wsock = env.get('wsgi.websocket')
    if not wsock:
        abort(400, 'Expected WebSocket request.')

    wsocks_control.append(wsock)
    while True:
        try:
            message = wsock.receive()
            wsock.send("Your message was: %r" % message)
            print message
            if message == "start":
                print "START"
                oven.run_profile("abc")
            elif message == "stop":
                oven.abort_run()
        except WebSocketError:
            break


@app.route('/status')
def handle_websocket():
    env = bottle.request.environ;
    print env
    wsock = env.get('wsgi.websocket')
    if not wsock:
        abort(400, 'Expected WebSocket request.')

    wsocks.append(wsock)
    while True:
        try:
            message = wsock.receive()
            wsock.send("Your message was: %r" % message)
        except WebSocketError:
            break


def main():
    server = WSGIServer(("0.0.0.0", 8080), app,
                    handler_class=WebSocketHandler)
    server.serve_forever()

if __name__ == "__main__":
    main()