Add additional None check

dan-r · Jul 8, 2024 · 0af1968 · 0af1968
1 parent 5ec6a42
commit 0af1968
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/ohme/utils.py b/custom_components/ohme/utils.py
@@ -35,7 +35,7 @@ def slot_list(data):
     slots = []
     wh_tally = 0
 
-    if 'batterySocBefore' in data and data['batterySocBefore'] is not None:
+    if 'batterySocBefore' in data and data['batterySocBefore'] is not None and data['batterySocBefore']['wh'] is not None:
         wh_tally = data['batterySocBefore']['wh'] # Get the wh value we start from
 
     for slot in session_slots: