main.py

from weioLib.weio import *
import weioLib.weioParser as wp
from weioLib.weioSerial import Serial
import json

import time

import paho.mqtt.client as mqtt

def setup():
    attach.process(electrolink)

MQTT_CLIENT = None
ser = None

def iCb(event, cbName):
    print ">>> INTERRUPT"
    if (MQTT_CLIENT is not None):
        rsp = {}
        rsp["id"] = "INTERRUPT"
        rsp["data"] = cbName
        MQTT_CLIENT.publish('/electrolink/rsp', json.dumps(rsp))

def electrolink():
    # The callback for when the client receives a CONNACK response from the server.
    def on_connect(client, userdata, flags, rc):
        print("Connected with result code "+str(rc))

    # The callback for when a PUBLISH message is received from the server.
    def on_message(client, userdata, msg):
        print(msg.topic+" "+str(msg.payload))
        
        data = json.loads(msg.payload)
        print data
        
        # Call the HW function on the board
        r = None
        if (data["method"] == "attachInterrupt"):
            attachInterrupt(data["params"][0],
                data["params"][1], iCb, data["params"][2])
        # UART uses standard pySerial functions, but need a firmware init
        # to put pin0 and pin1 in their alternate function and init the
        # CDC interface with the correct baudrate.
        elif (data["method"] == "uartStart"):
            global ser
            if ser == None:
                ser = Serial(data["params"][0])
        elif (data["method"] == "uartSend"):
            ret = ser.write(data["params"][0])
        elif (data["method"] == "uartReceive"):
            n = ser.inWaiting()
            if n:
                r = {}
                r["data"] = ser.read(n)
        elif(data["method"] == "uartStop"):
            ser.close()
            ser = None
        else:
            r = wp.weioSpells[data["method"]](data["params"])
        
        # send back result
        rsp = {}
        
        if (r is not None):
            rsp["result"] = r["data"]
        else:
            rsp["result"] = "none"
        
        rsp["id"] = "NORMAL"
        
        client.publish('/electrolink/rsp', json.dumps(rsp))
    
    def on_publish(client, userdata, mid):
        print("mid: "+str(mid))
    
    global MQTT_CLIENT
    MQTT_CLIENT = client = mqtt.Client()
    client.on_connect = on_connect
    client.on_message = on_message
    client.on_publish = on_publish

    client.connect("localhost", 1883, 60)

    wp.weioSpells["digitalWrite"]([18, 1])

    # Start looping
    client.loop_start()
    
    client.subscribe('/electrolink/req')
    
    while True:
        print "tick"
        time.sleep(3)