From b7f053a7899d706b0a2f872211cbe31d3796bc02 Mon Sep 17 00:00:00 2001
From: leaskovski <tristan.leask@googlemail.com>
Date: Tue, 6 Apr 2021 21:47:37 +0100
Subject: [PATCH] Started adding in MQTT discovery broadcasts

---
 src/rpi-cpu2mqtt.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/rpi-cpu2mqtt.py b/src/rpi-cpu2mqtt.py
index f772298..8741b0a 100644
--- a/src/rpi-cpu2mqtt.py
+++ b/src/rpi-cpu2mqtt.py
@@ -61,6 +61,19 @@ def check_uptime():
 		full_cmd = "awk '{print int($1/3600/24)}' /proc/uptime"
 		return int(subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0])
 
+def config_json(what_config):
+		data = {
+			"state_topic": "",
+			"icon": "",
+			"name": "",
+			"unique_id": "",
+		}
+		data["state_topic"] = config.mqtt_topic_prefix+"/"+hostname+"/"+what_config
+		data["unique_id"] = hostname+"_"+what_config
+		if what_config == "cpu_load":
+			data["icon"] = "mdi:speedometer"
+			data["name"] = hostname + " CPU Usage"
+	
 def publish_to_mqtt (cpu_load = 0, cpu_temp = 0, used_space = 0, voltage = 0, sys_clock_speed = 0, swap = 0, memory = 0, uptime_days = 0):
 		# connect to mqtt server
 		client = paho.Client()
@@ -69,6 +82,9 @@ def publish_to_mqtt (cpu_load = 0, cpu_temp = 0, used_space = 0, voltage = 0, sy
 
 		# publish monitored values to MQTT
 		if config.cpu_load:
+			if config.descovery_messages:
+				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"/cpuload/config", config_json('cpu_load'), qos=0)
+				time.sleep(config.sleep_time)
 			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/cpuload", cpu_load, qos=1)
 			time.sleep(config.sleep_time)
 		if config.cpu_temp: