From 955471836e4fc7584e84a3cbd1cadb125b85509b Mon Sep 17 00:00:00 2001
From: Dominik George <dominik.george@teckids.org>
Date: Tue, 16 Jul 2019 23:15:13 +0200
Subject: [PATCH] Import class groups from Untis.

---
 biscuit/apps/untis/util.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/biscuit/apps/untis/util.py b/biscuit/apps/untis/util.py
index cedce2b..da0341e 100644
--- a/biscuit/apps/untis/util.py
+++ b/biscuit/apps/untis/util.py
@@ -61,6 +61,14 @@ def untis_import_xml(request, untis_xml):
         room, created = Room.objects.get_or_create(short_name=short_name, defaults={
             'name': name})
 
+    classes = dom.getElementsByTagName('class')
+    for class_node in classes:
+        short_name = class_node.attributes['id'].value[3:]
+        name = _('Class %s') % short_name
+
+        class_, created = Group.objects.get_or_create(short_name=short_name, defaults={
+            'name': name})
+
     lessons = dom.getElementsByTagName('lesson')
     for lesson_node in lessons:
         subject_abbrev = get_child_node_id(lesson_node, 'lesson_subject')[3:]
-- 
GitLab