e-paper-display/examples/solar-monitoring.py

#!/usr/bin/python
# -*- coding:utf-8 -*-
import sys
import os
picdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'pic')
libdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'lib')
if os.path.exists(libdir):
    sys.path.append(libdir)

import logging
from waveshare_epd import epd2in13b_V3
#import waveshare-epaper
import time
from PIL import Image,ImageDraw,ImageFont
import traceback

# Import MQTT client.
from paho.mqtt import client as mqtt_client


logging.basicConfig(level=logging.DEBUG)

# MQTT credentials
#mqtt_username  = 'homeassistant' # MQTT client username
#mqtt_password = '3355' # MQTT client password
client_id = 'solarmonitor'  # unique client_id
mqtt_broker = "openwb-2"  # broker address, usually your HASS IP address
port = 1883

#set topics
topic_house_battery_soc = "openWB/bat/get/soc"
topic_house_battery_power = "openWB/bat/get/power"

topic_power_import_export = "openWB/counter/7/get/power"
topic_pv_power = "openWB/pv/8/get/power"

topic_house_power = "openWB/graph/lastlivevaluesJson" #house-power

topic_car_soc = "openWB/vehicle/4/get/soc"
topic_charge_point_power = "openWB/internal_chargepoint/0/get/powers"


def connect_mqtt():
    def on_connect(client, userdata, flags, rc):
        if rc == 0:
            logging.info("Connected to MQTT Broker!")
        else:
            logging.info("Failed to connect, return code %d\n", rc)

    client = mqtt_client.Client(client_id)
    # client.username_pw_set(username, password)
    client.on_connect = on_connect
    client.connect(mqtt_broker, port)
    return client


def subscribe(client: mqtt_client, topic):
    def on_message(client, userdata, msg):
        print(f"Received `{msg.payload.decode()}` from `{msg.topic}` topic")

    client.loop_start()
    client.subscribe(topic)
    client.on_message = on_message
    client.loop_stop()

def monitor(charge_point_power):
    
    logging.info("epd2in13b_V3 Demo")
    logging.info(libdir)
    logging.info(picdir)
    
    epd = epd2in13b_V3.EPD()
    #epd = epaper.epaper('epd2in13b_V3').EPD()
    #logging.info("init and Clear")
    #epd.init()
    #epd.Clear()
    
    # Drawing on the image
    logging.info("Drawing")    
    font16 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 16)
    font18 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 18)
    
    # Drawing on the Horizontal image
    logging.info("1.Drawing on the Horizontal image...") 
    HBlackimage = Image.new('1', (epd.height, epd.width), 255)  # 298*126
    HRYimage = Image.new('1', (epd.height, epd.width), 255)  # 298*126  ryimage: red or yellow image  
    drawblack = ImageDraw.Draw(HBlackimage)
    drawry = ImageDraw.Draw(HRYimage)
    drawblack.text((2, 0), 'PV Production', font = font16, fill = 0)
    drawblack.text((2, 16), 'Hausverbrauch', font = font16, fill = 0)
    drawblack.text((2, 32), 'Strom Import', font = font16, fill = 0)
    drawblack.text((2, 48), 'Batterie', font = font16, fill = 0)
    drawblack.text((2, 64), 'Batterie Entladen', font = font16, fill = 0)
    drawblack.text((2, 80), 'Auto Laden', font = font16, fill = 0)

    #drawblack.text((100, 80), str(charge_point_power), font = font16, fill = 0)

    
    epd.display(epd.getbuffer(HBlackimage), epd.getbuffer(HRYimage))

    #logging.info("Clear...")
    #epd.init()
    #epd.Clear()
    
    logging.info("Goto Sleep...")
    #epd.sleep()

    #except IOError as e:
    #    logging.info(e)

    #except KeyboardInterrupt:    
    #    logging.info("ctrl + c:")
    #    epd2in13b_V3.epdconfig.module_exit()
    #    exit()
    
    #client.loop_forever()


def run():
    client = connect_mqtt()
    house_battery_soc = subscribe(client, topic_house_battery_soc)
    house_battery_power = subscribe(client, topic_house_battery_power)
    power_import_export = subscribe(client, topic_power_import_export)
    pv_power = subscribe(client, topic_pv_power)
    house_power = subscribe(client, topic_house_power)
    car_soc = subscribe(client, topic_car_soc)
    charge_point_power = subscribe(client, topic_charge_point_power)

    logging.info("house battery soc: " + str(house_battery_soc))
    logging.info("house battery power: " + str(house_battery_power))
    logging.info("power import export: " + str(power_import_export))
    logging.info("pv power: " + str(pv_power))
    logging.info("house power: " + str(house_power))
    logging.info("car soc: " + str(car_soc))
    logging.info("charge point power: " + str(charge_point_power))
    #try:
    monitor(charge_point_power)
    #except IOError as e:
    #    logging.info(e)

    #except KeyboardInterrupt:    
    #    logging.info("ctrl + c:")
    #    epd2in13b_V3.epdconfig.module_exit()
    #    exit()


if __name__ == '__main__':
    run()