This commit is contained in:
2024-04-14 07:34:53 +02:00
parent c14ea74287
commit e4396418c8
3 changed files with 49 additions and 42 deletions

70
main.py
View File

@@ -1,6 +1,7 @@
# ajouter cette ligne en bas du fichier : sudo visudo
# gilles ALL=(ALL) NOPASSWD: /sbin/shutdown
# ajouter le hostname du computer
import os
import time
@@ -10,6 +11,9 @@ import subprocess
import threading
import psutil
import pynvml
import socket
hostname = socket.gethostname()
# Initialisation de pynvml
pynvml.nvmlInit()
@@ -61,9 +65,9 @@ def get_cpu_usage():
device_info = {
"identifiers": ["aorus"],
"name": "Aorus",
"manufacturer": "GigaByte",
"identifiers": [f"{hostname}"],
"name": "Mqtt_pilot",
"manufacturer": "Black",
"model": "desktop",
"sw_version": "1.0.0",
"suggested_area": "salon",
@@ -72,12 +76,12 @@ device_info = {
# Configuration des entités
shutdown_entity = {
"name": "shutdown_aorus",
"name": f"shutdown_{hostname}",
"type": "switch",
"unique_id": "shutdown_aorus_18:c0:4d:b5:65:74",
"command_topic": "pilot/aorus/shutdown/set",
"state_topic": "pilot/aorus/shutdown",
"availability_topic": "pilot/aorus/shutdown/available",
"unique_id": f"shutdown_{hostname}_18:c0:4d:b5:65:74",
"command_topic": f"pilot/{hostname}/shutdown/available",
"state_topic": f"pilot/{hostname}/shutdown",
"availability_topic": f"pilot/{hostname}/shutdown/available",
"device_class": "switch",
"payload_on": "ON",
"payload_off": "OFF",
@@ -88,13 +92,13 @@ shutdown_entity = {
}
# battery_entity = {
# "name": "battery_aorus",
# "name": f"battery_{hostname}",
# "type": "sensor",
# "unique_id": "battery_aorus_18:c0:4d:b5:65:74",
# "state_topic": "pilot/aorus/battery",
# "unique_id": f"battery_{hostname}_18:c0:4d:b5:65:74",
# "state_topic": f"pilot/{hostname}/battery",
# "unit_of_measurement": "%",
# "device_class": "battery",
# "availability_topic": "pilot/aorus/battery/available",
# "availability_topic": f"pilot/{hostname}/battery/available",
# "payload_available": "online",
# "payload_not_available": "offline",
# "icon": "mdi:battery",
@@ -102,11 +106,11 @@ shutdown_entity = {
# }
cpu_temp_entity = {
"name": "cpu_temp_aorus",
"name": f"cpu_temp_{hostname}",
"type": "sensor",
"unique_id": "cpu_temp_aorus_18:c0:4d:b5:65:74",
"state_topic": "pilot/aorus/cpu_temp",
"availability_topic": "pilot/aorus/cpu_temp/available",
"unique_id": f"cpu_temp_{hostname}_18:c0:4d:b5:65:74",
"state_topic": f"pilot/{hostname}/cpu_temp",
"availability_topic": f"pilot/{hostname}/cpu_temp/available",
"device_class": "temperature",
"unit_of_measurement": "°C",
"payload_available": "online",
@@ -116,11 +120,11 @@ cpu_temp_entity = {
}
gpu_temp_entity = {
"name": "gpu_temp_aorus",
"name": f"gpu_temp_{hostname}",
"type": "sensor",
"unique_id": "gpu_temp_aorus_18:c0:4d:b5:65:74",
"state_topic": "pilot/aorus/gpu_temp",
"availability_topic": "pilot/aorus/gpu_temp/available",
"unique_id": f"gpu_temp_{hostname}_18:c0:4d:b5:65:74",
"state_topic": f"pilot/{hostname}/gpu_temp",
"availability_topic": f"pilot/{hostname}/gpu_temp/available",
"device_class": "temperature",
"unit_of_measurement": "°C",
"payload_available": "online",
@@ -130,11 +134,11 @@ gpu_temp_entity = {
}
# Définition de l'entité memory_used
memory_used_entity = {
"name": "memory_used_aorus",
"name": f"memory_used_{hostname}",
"type": "sensor",
"unique_id": "memory_used_aorus_18:c0:4d:b5:65:74",
"state_topic": "pilot/aorus/memory_used",
"availability_topic": "pilot/aorus/memory_used/available",
"unique_id": f"memory_used_{hostname}_18:c0:4d:b5:65:74",
"state_topic": f"pilot/{hostname}/memory_used",
"availability_topic": f"pilot/{hostname}/memory_used/available",
# "device_class": "memory",
"unit_of_measurement": "MB",
"payload_available": "online",
@@ -144,11 +148,11 @@ memory_used_entity = {
}
gpu_memory_usage_entity = {
"name": "gpu_memory_usage",
"name": f"gpu_memory_usage_{hostname}",
"type": "sensor",
"unique_id": "gpu_memory_used_aorus_18:c0:4d:b5:65:74",
"state_topic": "pilot/aorus/gpu_memory_usage/state",
"availability_topic": "pilot/aorus/gpu_memory_usage/available",
"unique_id": f"gpu_memory_used_{hostname}_18:c0:4d:b5:65:74",
"state_topic": f"pilot/{hostname}/gpu_memory_usage/state",
"availability_topic": f"pilot/{hostname}/gpu_memory_usage/available",
"unit_of_measurement": "MB",
# "device_class": "memory",
"unit_of_measurement": "MB",
@@ -159,11 +163,11 @@ gpu_memory_usage_entity = {
}
cpu_usage_entity = {
"name": "cpu_usage_aorus",
"name": f"cpu_usage_{hostname}",
"type": "sensor",
"unique_id": "cpu_usage_aorus_18:c0:4d:b5:65:74",
"state_topic": "pilot/aorus/cpu_usage",
"availability_topic": "pilot/aorus/cpu_usage/available",
"unique_id": f"cpu_usage_{hostname}_18:c0:4d:b5:65:74",
"state_topic": f"pilot/{hostname}/cpu_usage",
"availability_topic": f"pilot/{hostname}/cpu_usage/available",
"unit_of_measurement": "%",
"payload_available": "online",
"payload_not_available": "offline",
@@ -206,7 +210,7 @@ def publish_discovery_messages(client):
)
# Publication de la configuration du capteur gpu_memory_usage
client.publish(
f"homeassistant/sensor/{gpu_memory_usage_entity['name']}/config",
f"{discovery_prefix}/sensor/{gpu_memory_usage_entity['name']}/config",
json.dumps(gpu_memory_usage_entity),
retain=True,
)