diff --git a/biscuit/apps/untis/sub.py b/biscuit/apps/untis/sub.py index 50e59d79fdad8479e9c930d54b0a10b66a741523..c16bafcce016cbe9514ed75113998029fc5c4b9f 100644 --- a/biscuit/apps/untis/sub.py +++ b/biscuit/apps/untis/sub.py @@ -246,7 +246,7 @@ class HeaderInformation: self.affected_teachers) > 0 or len(self.affected_classes) > 0 -def get_header_information(subs, date): +def get_header_information(subs, date, events=[]): """ Get header information like affected teachers/classes and missing teachers/classes for a given date :param date: The date as datetime object @@ -267,6 +267,15 @@ def get_header_information(subs, date): if _class not in info.affected_classes: info.affected_classes.append(_class) + for event in events: + for teacher in event.teachers: + if teacher.id not in [x.id for x in info.affected_teachers]: + info.affected_teachers.append(teacher) + + for _class in event.classes: + if _class.id not in [x.id for x in info.affected_classes]: + info.affected_classes.append(_class) + # Get all absences that are relevant for this day info.absences = get_all_absences_by_date(date)