check if attribute is present

rine77 · Dec 16, 2024 · 05c3f55 · 05c3f55
1 parent b1eff2c
commit 05c3f55
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/homeassistantedupage/sensor.py b/custom_components/homeassistantedupage/sensor.py
@@ -98,8 +98,10 @@ def extra_state_attributes(self):
         for i, grade in enumerate(self._grades):
             attributes[f"grade_{i+1}_title"] = grade.title
             attributes[f"grade_{i+1}_grade_n"] = grade.grade_n
-            attributes[f"grade_{i+1}_max_points"] = grade.max_points
-            attributes[f"grade_{i+1}_percent"] = grade.percent
+            if grade.max_points:
+                attributes[f"grade_{i+1}_max_points"] = grade.max_points
+            if grade.percent:
+                attributes[f"grade_{i+1}_percent"] = grade.percent
             attributes[f"grade_{i+1}_date"] = grade.date.strftime("%Y-%m-%d %H:%M:%S")
 
             teacher_name = grade.teacher.name if grade.teacher else "unknown"