From b62fa8e47c753dc1cf7457a2eecb3acc5b0e05ef Mon Sep 17 00:00:00 2001
From: Hristo Jelev <jeleff@gmail.com>
Date: Tue, 16 Nov 2021 23:25:26 +0200
Subject: [PATCH] formatting

---
 src/rpi-cpu2mqtt.py | 378 +++++++++++++++++++++++---------------------
 1 file changed, 200 insertions(+), 178 deletions(-)

diff --git a/src/rpi-cpu2mqtt.py b/src/rpi-cpu2mqtt.py
index 6f1178a..9da9a90 100644
--- a/src/rpi-cpu2mqtt.py
+++ b/src/rpi-cpu2mqtt.py
@@ -15,207 +15,229 @@ hostname = socket.gethostname()
 
 
 def check_used_space(path):
-		st = os.statvfs(path)
-		free_space = st.f_bavail * st.f_frsize
-		total_space = st.f_blocks * st.f_frsize
-		used_space = int(100 - ((free_space / total_space) * 100))
-		return used_space
+    st = os.statvfs(path)
+    free_space = st.f_bavail * st.f_frsize
+    total_space = st.f_blocks * st.f_frsize
+    used_space = int(100 - ((free_space / total_space) * 100))
+    return used_space
+
 
 def check_cpu_load():
-		# bash command to get cpu load from uptime command
-		p = subprocess.Popen("uptime", shell=True, stdout=subprocess.PIPE).communicate()[0]
-		cores = subprocess.Popen("nproc", shell=True, stdout=subprocess.PIPE).communicate()[0]
-		cpu_load = str(p).split("average:")[1].split(", ")[0].replace(' ', '').replace(',', '.')
-		cpu_load = float(cpu_load)/int(cores)*100
-		cpu_load = round(float(cpu_load), 1)
-		return cpu_load
+    # bash command to get cpu load from uptime command
+    p = subprocess.Popen("uptime", shell=True, stdout=subprocess.PIPE).communicate()[0]
+    cores = subprocess.Popen("nproc", shell=True, stdout=subprocess.PIPE).communicate()[0]
+    cpu_load = str(p).split("average:")[1].split(", ")[0].replace(' ', '').replace(',', '.')
+    cpu_load = float(cpu_load) / int(cores) * 100
+    cpu_load = round(float(cpu_load), 1)
+    return cpu_load
+
 
 def check_voltage():
-        try:
-            full_cmd = "vcgencmd measure_volts | cut -f2 -d= | sed 's/000//'"
-            voltage = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
-            voltage = voltage.strip()[:-1]
-        except:
-            voltage = 0
-        return voltage
+    try:
+        full_cmd = "vcgencmd measure_volts | cut -f2 -d= | sed 's/000//'"
+        voltage = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
+        voltage = voltage.strip()[:-1]
+    except:
+        voltage = 0
+    return voltage
+
 
 def check_swap():
-		full_cmd = "free -t | awk 'NR == 3 {print $3/$2*100}'"
-		swap = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
-		swap = round(float(swap.decode("utf-8").replace(",",".")), 1)
-		return swap
+    full_cmd = "free -t | awk 'NR == 3 {print $3/$2*100}'"
+    swap = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
+    swap = round(float(swap.decode("utf-8").replace(",", ".")), 1)
+    return swap
+
 
 def check_memory():
-		full_cmd = "free -t | awk 'NR == 2 {print $3/$2*100}'"
-		memory = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
-		memory = round(float(memory.decode("utf-8").replace(",",".")))
-		return memory
+    full_cmd = "free -t | awk 'NR == 2 {print $3/$2*100}'"
+    memory = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
+    memory = round(float(memory.decode("utf-8").replace(",", ".")))
+    return memory
+
 
 def check_cpu_temp():
-		full_cmd = "cat /sys/class/thermal/thermal_zone*/temp 2> /dev/null | sed 's/\(.\)..$//' | tail -n 1"
-		try:
-			p = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
-			cpu_temp = p.decode("utf-8").replace('\n', ' ').replace('\r', '')
-		except:
-			cpu_temp = 0	
-		return cpu_temp
+    full_cmd = "cat /sys/class/thermal/thermal_zone*/temp 2> /dev/null | sed 's/\(.\)..$//' | tail -n 1"
+    try:
+        p = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
+        cpu_temp = p.decode("utf-8").replace('\n', ' ').replace('\r', '')
+    except:
+        cpu_temp = 0
+    return cpu_temp
+
 
 def check_sys_clock_speed():
-		full_cmd = "awk '{printf (\"%0.0f\",$1/1000); }' </sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq"
-		return subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
+    full_cmd = "awk '{printf (\"%0.0f\",$1/1000); }' </sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq"
+    return subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
+
 
 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])
+    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": "",
-			"unit_of_measurement": "",
-		}
-		data["state_topic"] = config.mqtt_topic_prefix+"/"+hostname+"/"+what_config
-		data["unique_id"] = hostname+"_"+what_config
-		if what_config == "cpuload":
-			data["icon"] = "mdi:speedometer"
-			data["name"] = hostname + " CPU Usage"
-			data["unit_of_measurement"] = "%"
-		elif what_config == "cputemp":
-			data["icon"] = "hass:thermometer"
-			data["name"] = hostname + " CPU Temperature"
-			data["unit_of_measurement"] = "°C"
-		elif what_config == "diskusage":
-			data["icon"] = "mdi:harddisk"
-			data["name"] = hostname + " Disk Usage"
-			data["unit_of_measurement"] = "%"
-		elif what_config == "voltage":
-			data["icon"] = "mdi:speedometer"
-			data["name"] = hostname + " CPU Voltage"
-			data["unit_of_measurement"] = "V"
-		elif what_config == "swap":
-			data["icon"] = "mdi:harddisk"
-			data["name"] = hostname + " Disk Swap"
-			data["unit_of_measurement"] = "%"
-		elif what_config == "memory":
-			data["icon"] = "mdi:memory"
-			data["name"] = hostname + " Memory Usage"
-			data["unit_of_measurement"] = "%"
-		elif what_config == "sys_clock_speed":
-			data["icon"] = "mdi:speedometer"
-			data["name"] = hostname + " CPU Clock Speed"
-			data["unit_of_measurement"] = "MHz"
-		elif what_config == "uptime_days":
-			data["icon"] = "mdi:timer"
-			data["name"] = hostname + " Uptime"
-			data["unit_of_measurement"] = "days"
-		else:
-			return ""
-		# Return our built discovery config
-		return json.dumps(data)
-	
-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()
-		client.username_pw_set(config.mqtt_user, config.mqtt_password)
-		client.connect(config.mqtt_host, int(config.mqtt_port))
+    data = {
+        "state_topic": "",
+        "icon": "",
+        "name": "",
+        "unique_id": "",
+        "unit_of_measurement": "",
+    }
+    data["state_topic"] = config.mqtt_topic_prefix + "/" + hostname + "/" + what_config
+    data["unique_id"] = hostname + "_" + what_config
+    if what_config == "cpuload":
+        data["icon"] = "mdi:speedometer"
+        data["name"] = hostname + " CPU Usage"
+        data["unit_of_measurement"] = "%"
+    elif what_config == "cputemp":
+        data["icon"] = "hass:thermometer"
+        data["name"] = hostname + " CPU Temperature"
+        data["unit_of_measurement"] = "°C"
+    elif what_config == "diskusage":
+        data["icon"] = "mdi:harddisk"
+        data["name"] = hostname + " Disk Usage"
+        data["unit_of_measurement"] = "%"
+    elif what_config == "voltage":
+        data["icon"] = "mdi:speedometer"
+        data["name"] = hostname + " CPU Voltage"
+        data["unit_of_measurement"] = "V"
+    elif what_config == "swap":
+        data["icon"] = "mdi:harddisk"
+        data["name"] = hostname + " Disk Swap"
+        data["unit_of_measurement"] = "%"
+    elif what_config == "memory":
+        data["icon"] = "mdi:memory"
+        data["name"] = hostname + " Memory Usage"
+        data["unit_of_measurement"] = "%"
+    elif what_config == "sys_clock_speed":
+        data["icon"] = "mdi:speedometer"
+        data["name"] = hostname + " CPU Clock Speed"
+        data["unit_of_measurement"] = "MHz"
+    elif what_config == "uptime_days":
+        data["icon"] = "mdi:timer"
+        data["name"] = hostname + " Uptime"
+        data["unit_of_measurement"] = "days"
+    else:
+        return ""
+    # Return our built discovery config
+    return json.dumps(data)
 
-		# publish monitored values to MQTT
-		if config.cpu_load:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_cpuload/config", config_json('cpuload'), 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:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_cputemp/config", config_json('cputemp'), qos=0)
-				time.sleep(config.sleep_time)
-			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/cputemp", cpu_temp, qos=1)
-			time.sleep(config.sleep_time)
-		if config.used_space:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_diskusage/config", config_json('diskusage'), qos=0)
-				time.sleep(config.sleep_time)
-			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/diskusage", used_space, qos=1)
-			time.sleep(config.sleep_time)
-		if config.voltage:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_voltage/config", config_json('voltage'), qos=0)
-				time.sleep(config.sleep_time)
-			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/voltage", voltage, qos=1)
-			time.sleep(config.sleep_time)
-		if config.swap:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_swap/config", config_json('swap'), qos=0)
-				time.sleep(config.sleep_time)
-			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/swap", swap, qos=1)
-			time.sleep(config.sleep_time)
-		if config.memory:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_memory/config", config_json('memory'), qos=0)
-				time.sleep(config.sleep_time)
-			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/memory", memory, qos=1)
-			time.sleep(config.sleep_time)
-		if config.sys_clock_speed:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_sys_clock_speed/config", config_json('sys_clock_speed'), qos=0)
-				time.sleep(config.sleep_time)
-			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/sys_clock_speed", sys_clock_speed, qos=1)
-			time.sleep(config.sleep_time)
-		if config.uptime:
-			if config.discovery_messages:
-				client.publish("homeassistant/sensor/"+config.mqtt_topic_prefix+"/"+hostname+"_uptime_days/config", config_json('uptime_days'), qos=0)
-				time.sleep(config.sleep_time)
-			client.publish(config.mqtt_topic_prefix+"/"+hostname+"/uptime_days", uptime_days, qos=1)
-			time.sleep(config.sleep_time)
-		# disconect from mqtt server
-		client.disconnect()
 
-def bulk_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):
-		# compose the CSV message containing the measured values
+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()
+    client.username_pw_set(config.mqtt_user, config.mqtt_password)
+    client.connect(config.mqtt_host, int(config.mqtt_port))
 
-		values = cpu_load, float(cpu_temp), used_space, float(voltage), int(sys_clock_speed), swap, memory, uptime_days
-		values = str(values)[1:-1]
+    # publish monitored values to MQTT
+    if config.cpu_load:
+        if config.discovery_messages:
+            client.publish("homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_cpuload/config",
+                           config_json('cpuload'), 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:
+        if config.discovery_messages:
+            client.publish("homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_cputemp/config",
+                           config_json('cputemp'), qos=0)
+            time.sleep(config.sleep_time)
+        client.publish(config.mqtt_topic_prefix + "/" + hostname + "/cputemp", cpu_temp, qos=1)
+        time.sleep(config.sleep_time)
+    if config.used_space:
+        if config.discovery_messages:
+            client.publish("homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_diskusage/config",
+                           config_json('diskusage'), qos=0)
+            time.sleep(config.sleep_time)
+        client.publish(config.mqtt_topic_prefix + "/" + hostname + "/diskusage", used_space, qos=1)
+        time.sleep(config.sleep_time)
+    if config.voltage:
+        if config.discovery_messages:
+            client.publish("homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_voltage/config",
+                           config_json('voltage'), qos=0)
+            time.sleep(config.sleep_time)
+        client.publish(config.mqtt_topic_prefix + "/" + hostname + "/voltage", voltage, qos=1)
+        time.sleep(config.sleep_time)
+    if config.swap:
+        if config.discovery_messages:
+            client.publish("homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_swap/config",
+                           config_json('swap'), qos=0)
+            time.sleep(config.sleep_time)
+        client.publish(config.mqtt_topic_prefix + "/" + hostname + "/swap", swap, qos=1)
+        time.sleep(config.sleep_time)
+    if config.memory:
+        if config.discovery_messages:
+            client.publish("homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_memory/config",
+                           config_json('memory'), qos=0)
+            time.sleep(config.sleep_time)
+        client.publish(config.mqtt_topic_prefix + "/" + hostname + "/memory", memory, qos=1)
+        time.sleep(config.sleep_time)
+    if config.sys_clock_speed:
+        if config.discovery_messages:
+            client.publish(
+                "homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_sys_clock_speed/config",
+                config_json('sys_clock_speed'), qos=0)
+            time.sleep(config.sleep_time)
+        client.publish(config.mqtt_topic_prefix + "/" + hostname + "/sys_clock_speed", sys_clock_speed, qos=1)
+        time.sleep(config.sleep_time)
+    if config.uptime:
+        if config.discovery_messages:
+            client.publish("homeassistant/sensor/" + config.mqtt_topic_prefix + "/" + hostname + "_uptime_days/config",
+                           config_json('uptime_days'), qos=0)
+            time.sleep(config.sleep_time)
+        client.publish(config.mqtt_topic_prefix + "/" + hostname + "/uptime_days", uptime_days, qos=1)
+        time.sleep(config.sleep_time)
+    # disconect from mqtt server
+    client.disconnect()
 
-		# connect to mqtt server
-		client = paho.Client()
-		client.username_pw_set(config.mqtt_user, config.mqtt_password)
-		client.connect(config.mqtt_host, int(config.mqtt_port))
 
-		# publish monitored values to MQTT
-		client.publish(config.mqtt_topic_prefix+"/"+hostname, values, qos=1)
+def bulk_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):
+    # compose the CSV message containing the measured values
+
+    values = cpu_load, float(cpu_temp), used_space, float(voltage), int(sys_clock_speed), swap, memory, uptime_days
+    values = str(values)[1:-1]
+
+    # connect to mqtt server
+    client = paho.Client()
+    client.username_pw_set(config.mqtt_user, config.mqtt_password)
+    client.connect(config.mqtt_host, int(config.mqtt_port))
+
+    # publish monitored values to MQTT
+    client.publish(config.mqtt_topic_prefix + "/" + hostname, values, qos=1)
+
+    # disconect from mqtt server
+    client.disconnect()
 
-		# disconect from mqtt server
-		client.disconnect()
 
 if __name__ == '__main__':
-		# set all monitored values to False in case they are turned off in the config
-		cpu_load = cpu_temp = used_space = voltage = sys_clock_speed = swap = memory = uptime_days = False
+    # set all monitored values to False in case they are turned off in the config
+    cpu_load = cpu_temp = used_space = voltage = sys_clock_speed = swap = memory = uptime_days = False
 
-		# delay the execution of the script
-		time.sleep(config.random_delay)
+    # delay the execution of the script
+    time.sleep(config.random_delay)
 
-		# collect the monitored values
-		if config.cpu_load:
-			cpu_load = check_cpu_load()
-		if config.cpu_temp:
-			cpu_temp = check_cpu_temp()
-		if config.used_space:
-			used_space = check_used_space('/')
-		if config.voltage:
-			voltage = check_voltage()
-		if config.sys_clock_speed:
-			sys_clock_speed = check_sys_clock_speed()
-		if config.swap:
-			swap = check_swap()
-		if config.memory:
-			memory = check_memory()
-		if config.uptime:
-			uptime_days = check_uptime()
-		# Publish messages to MQTT
-		if config.group_messages:
-			bulk_publish_to_mqtt(cpu_load, cpu_temp, used_space, voltage, sys_clock_speed, swap, memory, uptime_days)
-		else:
-			publish_to_mqtt(cpu_load, cpu_temp, used_space, voltage, sys_clock_speed, swap, memory, uptime_days)
+    # collect the monitored values
+    if config.cpu_load:
+        cpu_load = check_cpu_load()
+    if config.cpu_temp:
+        cpu_temp = check_cpu_temp()
+    if config.used_space:
+        used_space = check_used_space('/')
+    if config.voltage:
+        voltage = check_voltage()
+    if config.sys_clock_speed:
+        sys_clock_speed = check_sys_clock_speed()
+    if config.swap:
+        swap = check_swap()
+    if config.memory:
+        memory = check_memory()
+    if config.uptime:
+        uptime_days = check_uptime()
+    # Publish messages to MQTT
+    if config.group_messages:
+        bulk_publish_to_mqtt(cpu_load, cpu_temp, used_space, voltage, sys_clock_speed, swap, memory, uptime_days)
+    else:
+        publish_to_mqtt(cpu_load, cpu_temp, used_space, voltage, sys_clock_speed, swap, memory, uptime_days)