v1
This commit is contained in:
70
main.py
70
main.py
@@ -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,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user