mqtt-halt-pack/client/client.py

import paho.mqtt.client as mqtt
import paho.mqtt.publish as publish
import traceback
import os
import datetime
import time

monitor_state = 1

FIRST_RECONNECT_DELAY = 1
RECONNECT_RATE = 2
MAX_RECONNECT_COUNT = 12
MAX_RECONNECT_DELAY = 60

def on_connect(client, userdata, flags, reason_code, properties):
    now = datetime.datetime.now()
    print(str(now) + ": " +"Connected to %s:%s" % (mqttc._host, mqttc._port))
    if reason_code == 0:
        print("Connected to MQTT Broker!")
    else:
        print("Failed to connect, return code %d\n", reason_code)

def on_disconnect(client, userdata, flags, reason_code, properties):
    now = datetime.datetime.now()
    print(str(now) + ": disconnected")
    print("Disconnected with result code: %s", reason_code)
    reconnect_count, reconnect_delay = 0, FIRST_RECONNECT_DELAY
    while reconnect_count < MAX_RECONNECT_COUNT:
        print("Reconnecting in %d seconds...", reconnect_delay)
        time.sleep(reconnect_delay)

        try:
            client.reconnect()
            print("Reconnected successfully!")
            return
        except Exception as err:
            print("%s. Reconnect failed. Retrying...", err)

        reconnect_delay *= RECONNECT_RATE
        reconnect_delay = min(reconnect_delay, MAX_RECONNECT_DELAY)
        reconnect_count += 1
    print("Reconnect failed after %s attempts. Exiting...", reconnect_count)

def on_message(client, userdata, message):
    global monitor_state

    now = datetime.datetime.now()
    print(str(now) + ": " + message.topic + " " + str(message.qos) + " " + str(message.payload))

    if message.payload.decode('UTF-8') == 'halt':
        print('powering off')
        try:
            publish.single("dom/pimirror-ack", "went-halt", hostname="192.168.31.5")
            print("ack sent")
        except Exception as e:
            print("Exception when sending:" + str(e))
        except:
            print("Exception when sending:" + traceback.format_exc())
        
        os.system("sudo shutdown now -h")

    elif message.payload.decode('UTF-8') == 'display_on':
        if monitor_state == 0:
            monitor_state = 1
            os.system("/home/pi/display_on.sh")
            print("display on")

    elif message.payload.decode('UTF-8') == 'display_off':
        if monitor_state == 1:
            monitor_state = 0
            os.system("/home/pi/display_off.sh")
            print("display off")

def on_publish(client, userdata, mid, reason_codes, properties):
    now = datetime.datetime.now()
    print(str(now) + ": " +"mid: "+str(mid))

def on_subscribe(client, userdata, mid, reason_codes, properties):
    now = datetime.datetime.now()
    print(str(now) + ": " +"Subscribed: "+str(mid)+" "+str(reason_codes))

def on_unsubscribe(client, userdata, mid, reason_codes, properties):
    now = datetime.datetime.now()
    print(str(now) + ": unsubscribe")

    # In NEW version, reason_codes is always a list. Empty for MQTTv3
    for unsub_result in reason_codes:
        # Any reason code >= 128 is a failure.
        if reason_codes[0] >= 128:
            print("unsubscribe code: " + str(reason_codes[0]))

def on_log(client, obj, level, string):
    now = datetime.datetime.now()
    print(str(now) + ": " +string)

# If you want to use a specific client id, use
# mqttc = mqtt.Client("client-id")
# but note that the client id must be unique on the broker. Leaving the client
# id parameter empty will generate a random id for you.

client_id = 'mqtt-client-solaria'

mqttc = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2, client_id)
mqttc.on_message = on_message
mqttc.on_connect = on_connect
mqttc.on_disconnect = on_disconnect
mqttc.on_publish = on_publish
mqttc.on_subscribe = on_subscribe
mqttc.on_unsubscribe = on_unsubscribe

# Uncomment to enable debug messages
mqttc.on_log = on_log
mqttc.connect("192.168.31.5")
mqttc.subscribe("dom/pimirror", 0)

mqttc.loop_forever()