Added GPU sensors to output (#20)

home-assistant-ecosystem · Oct 20, 2023 · a01ad9d · a01ad9d
1 parent 9fab51c
commit a01ad9d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/glances_api/__init__.py b/glances_api/__init__.py
@@ -170,4 +170,14 @@ async def get_ha_sensor_data(self) -> dict[str, Any]:
             sensor_data["raid"] = data
         if data := self.data.get("uptime"):
             sensor_data["uptime"] = data
+        if data := self.data.get("gpu"):
+            sensor_data["gpu"] = {}
+            for sensor in data:
+                sensor_data["gpu"][f"GPU_{sensor['gpu_id']}__{sensor['name']}"] = {
+                    "name": sensor["name"],
+                    "temperature": sensor["temperature"],
+                    "mem": sensor["mem"],
+                    "proc": sensor["proc"],
+                    "fan_speed": sensor["fan_speed"] if "fan_speed" in sensor else 0,
+                }
         return sensor_data