added voltage cpu and clock speed

This commit is contained in:
Hristo Jelev
2020-04-27 19:02:22 +03:00
parent b05165999e
commit 713662ce42

View File

@@ -6,63 +6,70 @@
from __future__ import division from __future__ import division
import subprocess, time, socket, os import subprocess, time, socket, os
import paho.mqtt.client as paho import paho.mqtt.client as paho
import config
#get device host name - used in mqtt topic #get device host name - used in mqtt topic
hostname = socket.gethostname() hostname = socket.gethostname()
#mqtt server configuration #mqtt server configuration
mqtt_host = "192.168.0.13" mqtt_host = config.mqtt_host
mqtt_user = "homeassistant" mqtt_user = config.mqtt_user
mqtt_password = "xyMaSokO123" mqtt_password = config.mqtt_password
mqtt_port = "1883" mqtt_port = config.mqtt_port
mqtt_topic_prefix = "masoko" mqtt_topic_prefix = config.mqtt_topic_prefix
def check_used_space(path): def check_used_space(path):
st = os.statvfs(path) st = os.statvfs(path)
free_space = st.f_bavail * st.f_frsize free_space = st.f_bavail * st.f_frsize
total_space = st.f_blocks * st.f_frsize total_space = st.f_blocks * st.f_frsize
used_space = int(100 - ((free_space / total_space) * 100)) used_space = int(100 - ((free_space / total_space) * 100))
return used_space return used_space
def check_cpu_load(): def check_cpu_load():
#bash command to get cpu load from uptime command #bash command to get cpu load from uptime command
p = subprocess.Popen("uptime", shell=True, stdout=subprocess.PIPE).communicate()[0] p = subprocess.Popen("uptime", shell=True, stdout=subprocess.PIPE).communicate()[0]
cpu_load = p.split("average:")[1].split(",")[0].replace(' ', '') cpu_load = p.split("average:")[1].split(",")[0].replace(' ', '')
return cpu_load return cpu_load
def check_voltage():
full_cmd = "vcgencmd measure_volts | cut -f2 -d= | sed 's/000//'"
return subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
def check_cpu_temp(): def check_cpu_temp():
#bash command to get rpi cpu temp full_cmd = "/opt/vc/bin/vcgencmd measure_temp"
full_cmd = "/opt/vc/bin/vcgencmd measure_temp" p = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
p = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE).communicate()[0] cpu_temp = p.replace('\n', ' ').replace('\r', '').split("=")[1].split("'")[0]
cpu_temp = p.replace('\n', ' ').replace('\r', '').split("=")[1].split("'")[0] return cpu_temp
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]
def publish_to_mqtt (cpu_load, cpu_temp, used_space, voltage, sys_clock_speed):
#connect to mqtt server
client = paho.Client()
client.username_pw_set(mqtt_user, mqtt_password)
client.connect(mqtt_host, mqtt_port)
def publish_to_mqtt (cpu_load, cpu_temp, used_space): #publish monitored values to MQTT
#connect to mqtt server client.publish(mqtt_topic_prefix+"/"+hostname+"/cpuload", cpu_load, qos=1)
client = paho.Client() time.sleep(1)
client.username_pw_set(mqtt_user, mqtt_password) client.publish(mqtt_topic_prefix+"/"+hostname+"/cputemp", cpu_temp, qos=1)
client.connect(mqtt_host, mqtt_port) time.sleep(1)
client.publish(mqtt_topic_prefix+"/"+hostname+"/diskusage", used_space, qos=1)
#publish cpu load mqtt message time.sleep(1)
client.publish(mqtt_topic_prefix+"/"+hostname+"/cpuload", cpu_load, qos=1) client.publish(mqtt_topic_prefix+"/"+hostname+"/voltage", voltage, qos=1)
time.sleep(1) time.sleep(1)
#disconect from mqtt server
#publish cpu temperature mqtt message client.disconnect()
client.publish(mqtt_topic_prefix+"/"+hostname+"/cputemp", cpu_temp, qos=1)
time.sleep(1)
#publish used space mqtt message
client.publish(mqtt_topic_prefix+"/"+hostname+"/diskusage", used_space, qos=1)
#disconect from mqtt server
client.disconnect()
if __name__ == '__main__': if __name__ == '__main__':
#check cpu load #collect the monitored values
cpu_load = check_cpu_load() cpu_load = check_cpu_load()
#check cpu temp cpu_temp = check_cpu_temp()
cpu_temp = check_cpu_temp() used_space = check_used_space('/')
#check used space voltage = check_voltage()
used_space = check_used_space('/') sys_clock_speed = check_sys_clock_speed()
#Publish messages
publish_to_mqtt(cpu_load, cpu_temp, used_space) #Publish messages to MQTT
publish_to_mqtt(cpu_load, cpu_temp, used_space, voltage, sys_clock_speed)