From 4fdb7f6263d6463ac2099c383b1221789d28e42e Mon Sep 17 00:00:00 2001
From: Alban Gruin
Date: Thu, 16 Nov 2017 13:34:30 +0100
Subject: Correction des erreurs lorsqu’un groupe n’a pas de sous-groupe

---
 models.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'models.py')

diff --git a/models.py b/models.py
index b9f1ba7..a90e9da 100644
--- a/models.py
+++ b/models.py
@@ -72,10 +72,12 @@ class Timetable(SlugModel):
 
 class GroupManager(Manager):
     def get_parents(self, group):
-        groups_criteria = Q(subgroup="") | Q(subgroup__startswith=group.subgroup) | \
-                          reduce(lambda x, y: x | y,
-                                 [Q(subgroup=group.subgroup[:i])
-                                  for i in range(1, len(group.subgroup) + 1)])
+        groups_criteria = Q(subgroup="") | Q(subgroup__startswith=group.subgroup)
+
+        if len(group.subgroup) != 0:
+            groups_criteria |= reduce(lambda x, y: x | y,
+                                      [Q(subgroup=group.subgroup[:i])
+                                       for i in range(1, len(group.subgroup + 1))])
 
         return self.get_queryset().filter(groups_criteria, mention=group.mention,
                                           timetable=group.timetable)
-- 
cgit v1.2.1