From e09d72f12dca73e2faad6e7ed224617c73977c8c Mon Sep 17 00:00:00 2001
From: alfrix <alfredomonclus@gmail.com>
Date: Sat, 2 Apr 2022 11:40:37 -0300
Subject: [PATCH] main and temp: fix same color for heater_generics and sensors

---
 panels/main_menu.py   | 8 ++++----
 panels/temperature.py | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/panels/main_menu.py b/panels/main_menu.py
index 392bc0cb..f1ae2d37 100644
--- a/panels/main_menu.py
+++ b/panels/main_menu.py
@@ -91,12 +91,12 @@ class MainPanel(MenuPanel):
             class_name = "graph_label_heater_bed"
             type = "bed"
         elif device.startswith("heater_generic"):
-            h = 1
+            self.h = 1
             for d in self.devices:
                 if "heater_generic" in d:
-                    h += 1
+                    self.h += 1
             image = "heater"
-            class_name = "graph_label_sensor_%s" % h
+            class_name = "graph_label_sensor_%s" % self.h
             type = "sensor"
         elif device.startswith("temperature_fan"):
             f = 1
@@ -111,7 +111,7 @@ class MainPanel(MenuPanel):
         else:
             s = 1
             try:
-                s += h
+                s += self.h
             except Exception:
                 pass
             for d in self.devices:
diff --git a/panels/temperature.py b/panels/temperature.py
index 5d798366..69200e31 100644
--- a/panels/temperature.py
+++ b/panels/temperature.py
@@ -297,12 +297,12 @@ class TemperaturePanel(ScreenPanel):
             class_name = "graph_label_heater_bed"
             type = "bed"
         elif device.startswith("heater_generic"):
-            h = 1
+            self.h = 1
             for d in self.devices:
                 if "heater_generic" in d:
-                    h += 1
+                    self.h += 1
             image = "heater"
-            class_name = "graph_label_sensor_%s" % h
+            class_name = "graph_label_sensor_%s" % self.h
             type = "sensor"
         elif device.startswith("temperature_fan"):
             f = 1
@@ -317,7 +317,7 @@ class TemperaturePanel(ScreenPanel):
         else:
             s = 1
             try:
-                s += h
+                s += self.h
             except Exception:
                 pass
             for d in self.devices: