powerctl/power_mqtt1.py

#!/usr/bin/env python3

import smbus
import time
import paho.mqtt.client as mqt
from datetime import datetime
import _thread

bus = smbus.SMBus(1)
mqtc = mqt.Client()

access = {}
# [ code : [last_access, penalty] ]


def allowed(code):
    print (access)
    if code in access:
        if millis_since(access[code][0]) > 3000+access[code][1]:
            access[code] = [datetime.now(),0]
            return True
        else:
            access[code] = [datetime.now(),access[code][1]+3000]
            return False
    else:
        access[code] = [datetime.now(),0]
        return True

def millis_since(start_time):
    dt = datetime.now() - start_time
    ms = (dt.days * 24 * 60 * 60 + dt.seconds) * 1000 + dt.microseconds / 1000.0
    return ms

def on_connect(mosq, obj, foo ,bar ):
    print( "Connected" )

def on_message(mosq, obj, msg):
    #print( "Received on topic: " + msg.topic + " Message: "+str(msg.payload) );
    msgs(msg.payload, msg.topic)

def on_subscribe(mosq, obj, mid, granted_qos):
    print("Subscribed OK")

#Funktion Setze Bit in Variable / Function Set Bit in byte
def set_bit(value, bit):
    return value | (1<<bit)

#Funktion rücksetzte Bit in Variable / Function reset Bit in byte
def clear_bit(value, bit):
    return value & ~(1<<bit)

def i2c_status_thread():
    global state
    #ports = (0xFE,0xFD,0xFB,0xF7,0xEF,0xDF,0xBF,0x7F)
    #         bael loun cant flur
    #         leba e    ina
    #         d
    names = { "flur", "baellebad", "lounge-front", "lounge-back", "baellebad-ein", "lounge-ein", "cantina-ein", "zentral-aus", "cantina"}
    ports = (0xF7,0xFF,0xFF,0xFE,0xFD,0xFF,0xFF,0xFF,0xFB)
    inputs = (0x23,0x21)   
    while True:
        byte = bus.read_byte(inputs[0])
        update = map( lambda x: 1 if (byte | x == x) else 0 , ports)
        byte = bus.read_byte(inputs[1])
        update[:8] = map( lambda x: 1 if (byte | x == x) else 0 , ports[:8])
        for i,j,n in state,update,names:
            if i != j:
                if n in power:
                    mqtc.publish("foobar/oben/" + n + "/strom/status", states[j], qos=0, retain=False)
                else:
                    mqtc.publish("foobar/oben/" + n + "/licht/status", states[j], qos=0, retain=False)

        time.sleep(5)

#Buttonbefehle
def switch(i,speed=0.5):
    if allowed(i):
        o = 0
        if i > 7:
            o = set_bit(o,i-8)
            bus.write_byte(0x21,255-o)
            time.sleep(speed)
            o = clear_bit(o,i-8)
            bus.write_byte(0x21,255-o)
        else:
            o = set_bit(o, i)
            bus.write_byte(0x3f,255-o)
            time.sleep(speed)
            o = clear_bit(o,i)
            bus.write_byte(0x3f,255-o)
    
state = [0,0,0,0,0,0,0,0,0,0]
states = {0: "OFF", 1: "ON"}
commands = { "flur" : 0, "baellebad" : 1, "lounge-front": 2, "lounge-back" : 3,"baellebad-ein" : 4, "lounge-ein" : 5, "cantina-ein" : 6, "zentral-aus" : 7, "cantina" : 8}
power = { "zentral" : 7 , "baellebad" : 4, "lounge" : 5, "cantina" : 6 }
light = { "flur" : 0, "baellebad" : 1, "lounge-front" : 2, "lounge-back" : 3, "cantina" : 8 }



# foobar/oben/lounge   /licht/action
#             cantina  /strom/status
#             flur
#             baellebad
#             zentral 

def switch_state(i, state):
    if state[i] != state: # changed
        switch(i)
        state[i] = state
        return True
    return False

def decode_topic(topic, state):
    #lazy
    print("dummy")
    list = topic.split('/')
    if list[3] == "licht":
        if list[2] in light:
            if switch_state(light[list[2]], state):
                mqtc.publish(topic.replace("action", "status"), states[state], qos=0, retain=False)
    elif list[3] == "strom":
        if list[2] in power:
            if switch_state(power[list[2]], state):
                mqtc.publish(topic.replace("action", "status"), states[state], qos=0, retain=False)

def msgs(inp, topic):
    c = inp.decode("utf-8")
    l = len(c)
    # supporting number commands
    if c == "OFF" or c == "ON":
        if c.find("ON") >= 0:
            decode_topic(topic, 1)
        elif c.find("OFF") >= 0:
            decode_topic(topic, 0)
        return
    elif l < 3:
        try:
            msg = int(inp)
            switch(msg)
        except ValueError:
            print ( "No valid value")
            return

    #supporting string commmands
    else:

        cmds = c.split(",")
        print ( "Command:", cmds[0], "Number of Parameters:", len(cmds) )

        #suporting string commands with an arg separated by ','
        if len(cmds) > 1:
            
            #error checking
            try:
                command = commands[cmds[0]]
            except KeyError:
                return
            try:
                arg = int (cmds[1])
            except ValueError:
                return
            
            #strobo
            if ( command == 100 or command == 99 ):
                if arg < 100:
                    for i in range(arg):
                        switch((100-command)*8, speed=0.05)
                        time.sleep(0.06)
            
            #command with parameter used for dimming
            else:       
                switch(command,speed = 4*arg/100+1 )

        #single string command without parameter
        else:
            try:
                switch(commands[c])
            except KeyError:
                return
        
    return


def init_mqtt():
    mqtc.connect("10.42.0.244", 1883, 60)
    mqtc.subscribe("foobar/oben/licht", 0)
    mqtc.subscribe("foobar/oben/lounge-back/licht/action", 0)
    mqtc.subscribe("foobar/oben/lounge-front/licht/action", 0)
    mqtc.subscribe("foobar/oben/lounge/strom/action", 0)
    mqtc.subscribe("foobar/oben/baellebad/licht/action", 0)
    mqtc.subscribe("foobar/oben/baellebad/strom/action", 0)
    mqtc.subscribe("foobar/oben/cantina/licht/action", 0)
    mqtc.subscribe("foobar/oben/cantina/strom/action", 0)
    mqtc.subscribe("foobar/oben/flur/licht/action", 0)
    mqtc.subscribe("foobar/oben/licht", 0)
    mqtc.on_message = on_message
    mqtc.on_connect = on_connect
    mqtc.on_subscribe = on_subscribe

    _thread.start_new_thread(i2c_status_thread, ())

    mqtc.loop_forever()

init_mqtt()

# vim: noai:ts=4:sw=4