CANNode/tools/shutter2can.py

#!/usr/bin/env python3 

from __future__ import absolute_import, print_function

import sys
import argparse
import socket
from datetime import datetime

import can
from can import Bus, BusState, Logger

import paho.mqtt.client as mqtt

sub_topic = "shutter"
aliases = { 
  'Esszimmer_Garten': 'EZ_G',
  'Esszimmer_Nachbar': 'EZ_N',
  'Wohnzimmer1': 'WZ1',
  'Wohnzimmer2': 'WZ2',
  'Wohnzimmer3': 'WZ3',
  'Wohnzimmer4': 'WZ4',
}
event_map = {
  'stop': 0,
  'off': 0,
  'up': 1,
  'down': 2
}
coil_map = {
  # up , down
  'EZ_G':  [ 0x03, 0x04 , 0x03, 0x05 ],
  'EZ_N':  [ 0x03, 0x08 , 0x03, 0x06 ],
  'WZ1': [ 0x00, 0x01 , 0x00, 0x01 ],
  'WZ2': [ 0x00, 0x01 , 0x00, 0x01 ],
  'WZ3': [ 0x00, 0x01 , 0x00, 0x01 ],
  'WZ4': [ 0x00, 0x01 , 0x00, 0x01 ]
}
cmd_map = {
  'off': 0,
  'on': 1,
  'value': 2,
  'toggle': 3
}
  

def decon(msg_id):
   msg_prio=msg_id>>26
   msg_type=(msg_id>>24) & 0x03
   msg_dst=(msg_id>>16) & 0xFF
   msg_src=(msg_id>>8) & 0xFF
   msg_cmd=msg_id & 0xFF
   print("msg_id: ", hex(msg_id), "prio: ", hex(msg_prio),"type: ", hex(msg_type)," dst: ", hex(msg_dst)," src: ", hex(msg_src)," cmd: ",hex(msg_cmd))
   return [msg_prio, msg_type, msg_dst, msg_src, msg_cmd]

def con(msg_dst,msg_cmd,msg_prio=3,msg_type=0,msg_src=11, ):
   msg_id=(msg_prio<<26) + \
   ((msg_type&0x03)<<24) +\
   ((msg_dst&0xFF)<<16) +\
   ((msg_src&0xFF)<<8) +\
   (msg_cmd&0xFF)
   print("Constructed ID: "+hex(msg_id))
   return msg_id

def on_connect(mcp_mqtt, userdata, flags, rc):
    print("Connected with result code "+str(rc))

    # Subscribing in on_connect() means that if we lose the connection and
    # reconnect then subscriptions will be renewed.
    mcp_mqtt.subscribe(sub_topic+"/+")
    print("Subscribed to:"+sub_topic+"/+")

def on_message(mcp_mqtt, userdata, msg):
    local_bus=userdata
    print("data Received topic: ", msg.topic)
    m_decode=str(msg.payload.decode("utf-8","ignore"))
    print("data Received",m_decode)
    data=m_decode
    sub=msg.topic[len(sub_topic)+1:]
    event=event_map[data]
    coils={
      'up': [ coil_map[sub][0], coil_map[sub][1] ],
      'down': [ coil_map[sub][2], coil_map[sub][3] ]
    }
    if event is not None: # it's 0,1,2
      for key, value in coils.items():
        msg_id=con(msg_dst=value[0],msg_cmd=0)
        msg_data=[value[1]]
        print("turning off: ",sub,value[0],value[1])
        m = can.Message(arbitration_id=msg_id,
            data=msg_data,
            extended_id=True)
        print("going to sent to CAN",m)
        try:
          local_bus.send(m)
        except BaseException as e:
          logging.error("Error sending can message {%s}: %s" % (m, e))
        print("data sent to CAN",m)
      if event is not 0:
        addr = coils[data]
        print("turning on: ",addr[0],addr[1])
        msg_id=con(msg_dst=addr[0],msg_cmd=1)
        msg_data=[addr[1]]
        print(msg_data)
        m = can.Message(arbitration_id=msg_id,
            data=msg_data,
            extended_id=True)
        print("going to sent to CAN",m)
  
        try:
            local_bus.send(m)
        except BaseException as e:
            logging.error("Error sending can message {%s}: %s" % (m, e))
        print("data sent to CAN",m)




def main():
    verbosity = 2

    logging_level_name = ['critical', 'error', 'warning', 'info', 'debug', 'subdebug'][min(5, verbosity)]
    can.set_logging_level(logging_level_name)

    can_filters = []
    config = {"can_filters": can_filters, "single_handle": True}
    config["interface"] = "socketcan"
    config["bitrate"] = 125000
    bus = Bus("can1", **config)

    print('Connected to {}: {}'.format(bus.__class__.__name__, bus.channel_info))
    print('Can Logger (Started on {})\n'.format(datetime.now()))

    mcp_mqtt = mqtt.Client()
    mcp_mqtt.on_connect = on_connect
    mcp_mqtt.on_message = on_message
    mcp_mqtt.user_data_set(bus)
    mcp_mqtt.connect("mcp", 1883, 60)

    try:
      while True:
        mcp_mqtt.loop()
    except KeyboardInterrupt:
        pass
    finally:
        bus.shutdown()

if __name__ == "__main__":
    main()