More next slot bugfixes

dan-r · Sep 18, 2024 · 05758ee · 05758ee
1 parent 29f5ddf
commit 05758ee
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/ohme/utils.py b/custom_components/ohme/utils.py
@@ -17,11 +17,11 @@ def next_slot(hass, data):
     # Loop through slots
     for slot in slots:
         # Only take the first slot start/end that matches. These are in order.
+        if end is None and slot['end'] > datetime.now().astimezone():
+            end = slot['end']
+
         if start is None and slot['start'] > datetime.now().astimezone():
             start = slot['start']
-            end = slot['end']
-        elif end is None and slot['end'] > datetime.now().astimezone():
-            end = slot['end']
         elif collapse_slots and slot['start'] == end:
             end = slot['end']
         elif start is not None and end is not None: