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

174 lines
4.6 KiB
Python

#!/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
topic0 = {
"topic" : "openWB/bat/get/soc",
"line" : "0",
"value" : ""
}
topic1 = {
"topic" : "openWB/bat/get/power",
"line" : "16",
"value" : ""
}
topic2 = {
"topic" : "openWB/counter/7/get/power",
"line" : "32",
"value" : ""
}
topic3 = {
"topic" : "openWB/pv/8/get/power",
"line" : "48",
"value" : ""
}
topic4 = {
"topic" : "openWB/graph/lastlivevaluesJson", #house-power
"line" : "64",
"value" : ""
}
topic5 = {
"topic" : "openWB/vehicle/4/get/soc",
"line" : "80",
"value" : ""
}
topic6 = {
"topic" : "openWB/internal_chargepoint/0/get/powers",
"line" : "80",
"value" : ""
}
topics = {0:topic0,1:topic1,2:topic2,3:topic3,4:topic4,5:topic5,6:topic6}
#Define Display
epd = epd2in13b_V3.EPD()
epd.init()
#Set fonts
font16 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 16)
font18 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 18)
#Init 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)
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 init_monitor(topics):
logging.info("epd2in13b_V3 Demo")
#epd = epd2in13b_V3.EPD()
#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...")
drawblack.text((4, 0), 'PV Production', font = font16, fill = 0)
drawblack.text((4, 16), 'Hausverbrauch', font = font16, fill = 0)
drawblack.text((4, 32), 'Strom Import', font = font16, fill = 0)
drawblack.text((4, 48), 'Batterie', font = font16, fill = 0)
drawblack.text((4, 64), 'Batterie Entladen', font = font16, fill = 0)
drawblack.text((4, 80), 'Auto Laden', font = font16, fill = 0)
epd.display(epd.getbuffer(HBlackimage), epd.getbuffer(HRYimage))
#logging.info("Clear...")
#epd.init()
#epd.Clear()
print(drawblack)
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():
try:
init_monitor(topics)
except IOError as e:
logging.info(e)
except KeyboardInterrupt:
logging.info("ctrl + c:")
epd2in13b_V3.epdconfig.module_exit()
exit()
client = connect_mqtt()
for topic_id, topic_info in topics.items():
print("\nTopic ID:", topic_id)
for key in topic_info:
print(key + ':', topic_info[key])
topic_info["value"] = subscribe(client, topic_info["topic"])
print(topic_info["line"])
print(topic_info["value"])
drawblack.text((160, int(topic_info["line"])), str(topic_info["value"]), font = font16, fill = 0)
if __name__ == '__main__':
run()