Added some error checking

jinglemansweep · Feb 25, 2024 · f9960c5 · f9960c5
1 parent 9dd56b1
commit f9960c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wideboy/systems/scene/entity_tiles.py b/wideboy/systems/scene/entity_tiles.py
@@ -621,7 +621,7 @@ class CellDateDogsFleaTreatment(GridCell):
 
     @property
     def label(self):
-        if "state" not in self.entity_state:
+        if "day" not in self.entity_state or "month" not in self.entity_state:
             return "N/A"
         return f"{self.entity_state['day']}/{self.entity_state['month']}"
 
@@ -632,7 +632,7 @@ class CellDateDogsWormTreatment(GridCell):
 
     @property
     def label(self):
-        if "state" not in self.entity_state:
+        if "day" not in self.entity_state or "month" not in self.entity_state:
             return "N/A"
         return f"{self.entity_state['day']}/{self.entity_state['month']}"