From 51768926049c936ee13ba9b54f379cc7663452c1 Mon Sep 17 00:00:00 2001
From: Giga77 <2777446+Giga77@users.noreply.github.com>
Date: Sat, 23 Nov 2024 10:29:50 +0100
Subject: [PATCH] Update ecole_directe_helper.py

---
 .../ecole_directe/ecole_directe_helper.py     | 21 ++++++-------------
 1 file changed, 6 insertions(+), 15 deletions(-)

diff --git a/custom_components/ecole_directe/ecole_directe_helper.py b/custom_components/ecole_directe/ecole_directe_helper.py
index 0d18ed4..c0682da 100644
--- a/custom_components/ecole_directe/ecole_directe_helper.py
+++ b/custom_components/ecole_directe/ecole_directe_helper.py
@@ -205,6 +205,7 @@ def get_ecoledirecte_session(data, config_path, hass) -> EDSession | None:
             + urllib.parse.quote(data["password"], safe="")
             + '", "isRelogin": false}'
         )
+
         login = get_response(
             None,
             f"{APIURL}/login.awp?v={APIVERSION}",
@@ -513,21 +514,11 @@ def get_grades_evaluations(token, eleve, annee_scolaire, config_path):
             response["disciplines"] = get_disciplines_periode(periode_json)
             if periode_json["ensembleMatieres"]:
                 response["moyenne_generale"] = {
-                    "moyenneGenerale": periode_json.get(
-                        ["ensembleMatieres"]["moyenneGenerale"], ""
-                    ),
-                    "moyenneClasse": periode_json.get(
-                        ["ensembleMatieres"]["moyenneClasse"], ""
-                    ),
-                    "moyenneMin": periode_json.get(
-                        ["ensembleMatieres"]["moyenneMin"], ""
-                    ),
-                    "moyenneMax": periode_json.get(
-                        ["ensembleMatieres"]["moyenneMax"], ""
-                    ),
-                    "dateCalcul": periode_json.get(
-                        ["ensembleMatieres"]["dateCalcul"], ""
-                    ),
+                    "moyenneGenerale": periode_json["ensembleMatieres"].get("moyenneGenerale", ""),
+                    "moyenneClasse": periode_json["ensembleMatieres"].get("moyenneClasse", ""),
+                    "moyenneMin": periode_json["ensembleMatieres"].get("moyenneMin", ""),
+                    "moyenneMax": periode_json["ensembleMatieres"].get("moyenneMax", ""),
+                    "dateCalcul": periode_json["ensembleMatieres"].get("dateCalcul", ""),
                 }
 
             break