From d6d3df65df90d591e73a6c93b007f0614b531f17 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Geoffrey=20=E2=80=9CFrogeye=E2=80=9D=20Preud=27homme?=
 <geoffrey@frogeye.fr>
Date: Mon, 17 Jun 2024 19:30:48 +0200
Subject: [PATCH] frobar: Display temperature for other CPU types

---
 hm/desktop/frobar/frobar/providers.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/hm/desktop/frobar/frobar/providers.py b/hm/desktop/frobar/frobar/providers.py
index 87d5f57..9568936 100644
--- a/hm/desktop/frobar/frobar/providers.py
+++ b/hm/desktop/frobar/frobar/providers.py
@@ -144,16 +144,20 @@ class RamProvider(AlertingSection, PeriodicUpdater):
 class TemperatureProvider(AlertingSection, PeriodicUpdater):
     NUMBER_STATES = 2
     RAMP = ""
+    MAIN_TEMPS = ["coretemp", "amdgpu", "cpu_thermal"]
+    # For Intel, AMD and ARM respectively.
 
     def fetcher(self):
         allTemp = psutil.sensors_temperatures()
-        if "coretemp" not in allTemp:
-            # TODO Opti Remove interval
-            return ""
-        temp = allTemp["coretemp"][0]
+        for main in self.MAIN_TEMPS:
+            if main in allTemp:
+                break
+        else:
+            return "?"
+        temp = allTemp[main][0]
 
-        self.warningThresold = temp.high
-        self.dangerThresold = temp.critical
+        self.warningThresold = temp.high or 90.0
+        self.dangerThresold = temp.critical or 100.0
         self.updateLevel(temp.current)
 
         self.icon = Section.ramp(temp.current / temp.high, self.RAMP)