Update __init__.py

ignore lesson if lesson.subjects is empty
JonasJoKuJonas · Nov 6, 2022 · 43df74c · 43df74c
1 parent e8a8491
commit 43df74c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/custom_components/webuntis/__init__.py b/custom_components/webuntis/__init__.py
@@ -228,7 +228,7 @@ def _is_class(self):
 
         for lesson in table:
             # pylint: disable=maybe-no-member
-            if lesson.start < now < lesson.end and lesson.code != "cancelled":
+            if lesson.start < now < lesson.end and lesson.code != "cancelled" and lesson.subjects != "":
                 return True
         return False
 
@@ -245,7 +245,7 @@ def _next_class(self):
 
         time_list = []
         for lesson in table:
-            if lesson.start > now and lesson.code != "cancelled":
+            if lesson.start > now and lesson.code != "cancelled" and lesson.subjects != "":
                 time_list.append(lesson.start)
 
         return sorted(time_list)[0].astimezone()
@@ -260,7 +260,7 @@ def _first_class(self):
 
         time_list = []
         for lesson in table:
-            if lesson.code != "cancelled":
+            if lesson.code != "cancelled" and lesson.subjects != "":
                 time_list.append(lesson.start)
 
         if len(time_list) > 1:
@@ -280,7 +280,7 @@ def _next_lesson_to_wake_up(self):
 
         time_list = []
         for lesson in table:
-            if lesson.code != "cancelled":
+            if lesson.code != "cancelled" and lesson.subjects != "":
                 time_list.append(lesson.start)
 
         day = now