From 912f3cca24da9d870aa191348fef901adcd55b7d Mon Sep 17 00:00:00 2001
From: Alban Gruin
Date: Wed, 1 Nov 2017 15:17:35 +0100
Subject: get_parents() sélectionne aussi les groupes enfants dont le
 sous-groupe commence par la même chaîne que celui du groupe passé en
 paramètre. Correction cosmétique du code.

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

(limited to 'models.py')

diff --git a/models.py b/models.py
index c546737..c3118a3 100644
--- a/models.py
+++ b/models.py
@@ -71,7 +71,7 @@ class Timetable(SlugModel):
 
 class GroupManager(Manager):
     def get_parents(self, group):
-        groups_criteria = Q(subgroup__isnull=True) | \
+        groups_criteria = Q(subgroup__isnull=True) | 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)])
@@ -80,7 +80,7 @@ class GroupManager(Manager):
                                           timetable=group.timetable)
 
     def get_relevant_groups(self, timetable, *args, **criteria):
-        sub = Group.objects.filter(timetable=timetable,mention=OuterRef("mention"),
+        sub = Group.objects.filter(timetable=timetable, mention=OuterRef("mention"),
                                    subgroup__startswith=OuterRef("subgroup")) \
                            .order_by().values("mention").annotate(c=Count("*")).values("c")
 
-- 
cgit v1.2.1