avoid error if session doesn't exist

hacf-fr · Mar 28, 2024 · 8db01d5 · 8db01d5
1 parent 135f3fe
commit 8db01d5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/custom_components/ecole_directe/ecole_directe_helper.py b/custom_components/ecole_directe/ecole_directe_helper.py
@@ -8,7 +8,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 APIURL = "https://api.ecoledirecte.com/v3"
-APIVERSION = "4.53.0"
+APIVERSION = "4.53.4"
 
 
 def encode_body(dictionnary, is_recursive=False):

diff --git a/custom_components/ecole_directe/sensor.py b/custom_components/ecole_directe/sensor.py
@@ -42,15 +42,15 @@ async def async_setup_entry(
     ]
 
     sensors = []
-
-    for eleve in coordinator.data["session"].eleves:
-        sensors.append(EDChildSensor(coordinator, eleve))
-        if "CAHIER_DE_TEXTES" in eleve.modules:
-            sensors.append(EDHomeworksSensor(coordinator, eleve))
-        if "NOTES" in eleve.modules:
-            sensors.append(EDGradesSensor(coordinator, eleve))
-
-    async_add_entities(sensors, False)
+    if "session" in coordinator.data and "eleves" in coordinator.data["session"]:
+        for eleve in coordinator.data["session"].eleves:
+            sensors.append(EDChildSensor(coordinator, eleve))
+            if "CAHIER_DE_TEXTES" in eleve.modules:
+                sensors.append(EDHomeworksSensor(coordinator, eleve))
+            if "NOTES" in eleve.modules:
+                sensors.append(EDGradesSensor(coordinator, eleve))
+
+        async_add_entities(sensors, False)
 
 
 class EDGenericSensor(CoordinatorEntity, SensorEntity):