devlolwebstatus

#!/usr/bin/python3

from http.server import BaseHTTPRequestHandler,HTTPServer

import paho.mqtt.client as paho
import logging,sys

import argparse


spaceOpen = False
brokerHost = "mqtt.devlol.org"
mqttTopic = "devlol/h19/door/lockstatus"


class MyHandler(BaseHTTPRequestHandler):

     def do_HEAD(client):
        client.send_response(200)
        client.send_header("Content-type", "text/html")
        client.end_headers()

     def do_GET(self):
        global spaceOpen
        if self.path == "/":
            self.send_response(200)
            self.send_header("Content-type", "image/png")
            self.end_headers()
            if spaceOpen:
                logging.info("return OPEN")
                try:
                    self.wfile.write(load_binary('assets/open.png'))
                except BrokenPipeError:
                    pass
            else:
                logging.info("return CLOSED")
                try:
                    self.wfile.write(load_binary('assets/closed.png'))
                except BrokenPipeError:
                    pass


def load_binary(file):
    with open(file, 'rb') as file:
        return file.read()

def load(file):
    with open(file, 'r') as file:
        return encode(str(file.read()))

def encode(file):
    return bytes(file, 'UTF-8')


def on_message(client, userdata, msg):
    global spaceOpen
    payload = msg.payload.decode("utf-8")
    logging.debug(payload)
    """ Callback for mqtt message."""
    if payload == "UNLOCKED":
        spaceOpen = True
        logging.info("received mqtt: UNLOCKED")
    elif payload == "LOCKED":
        spaceOpen = False
        logging.info("received mqtt: LOCKED")


def on_disconnect(client, userdata, foo):
    connected = False
    while not connected:
        try:
            client.reconnect()
            connected = True
            # resubscribe to the topics
            client.subscribe(mqttTopic)
        except:
            logging.error("Failed to reconnect mqtt...")
            time.sleep(1)


def run(server_class=HTTPServer,
        handler_class=MyHandler):

    parser = argparse.ArgumentParser()
    parser.add_argument("-p", default=8321, help="port to bind to", type=int)
    parser.add_argument("-v", "--verbose", help="increase output verbosity", action="store_true")
    args = parser.parse_args()
    port = args.p

    if args.verbose:
        logging.basicConfig(level=logging.DEBUG)
    else:
        logging.basicConfig(level=logging.ERROR)

    ## setup MQTT client
    client = paho.Client()
    client.on_message = on_message
    client.on_disconnect = on_disconnect

    try:
        client.connect(brokerHost)
        logging.info ("connected to mqtt")
    except:
        logging.error("failed to connect to mqtt")
        on_disconnect(client, None, None)

    client.subscribe(mqttTopic)


    server_address = ('', port)
    httpd = server_class(server_address, handler_class)

    while True:
        httpd.handle_request()
        client.loop()

if __name__ == "__main__":
    run()