From a6c5bfa4796081747e04ae4047007d5c8ad23164 Mon Sep 17 00:00:00 2001
From: Alban Gruin
Date: Mon, 4 Sep 2017 18:35:26 +0200
Subject: Ajout d’un champ « groupe parent » au modèle Group pour n’afficher
 que les groupes qui n’ont pas d’enfants. Par exemple, le groupe TPA21 aura
 comme parent le groupe TDA2, qui aura le groupe CMA comme parent.

Pour l’instant, le parseur d’emploi du temps ne créée pas de telles relations.
---
 admin.py  | 2 +-
 models.py | 8 ++++++++
 utils.py  | 4 ++--
 views.py  | 4 ++--
 4 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/admin.py b/admin.py
index f818865..1c8606a 100644
--- a/admin.py
+++ b/admin.py
@@ -34,7 +34,7 @@ class LastUpdateAdmin(admin.ModelAdmin):
 class GroupAdmin(admin.ModelAdmin):
     fieldsets = (
         (None, {"fields": ("name", "celcat_name", "timetable",)}),
-        ("Groupes", {"fields": ("mention", "subgroup", "td", "tp",)}),)
+        ("Groupes", {"fields": ("mention", "subgroup", "td", "tp", "parent_group",)}),)
     list_display = ("name", "timetable",)
     list_filter = ("timetable__name",)
     readonly_fields = ("celcat_name", "mention", "subgroup", "td", "tp",)
diff --git a/models.py b/models.py
index 876160b..53bcef7 100644
--- a/models.py
+++ b/models.py
@@ -56,7 +56,14 @@ class LastUpdate(models.Model):
         verbose_name_plural = "dernières mises à jour"
 
 
+class GroupManager(Manager):
+    def get_relevant_groups(self):
+        return self.get_queryset().annotate(children_count=Count("children")).filter(children_count=0)
+
+
 class Group(models.Model):
+    objects = GroupManager()
+
     name = models.CharField(max_length=255, verbose_name="nom")
     celcat_name = models.CharField(max_length=255, verbose_name="nom dans Celcat")
     timetable = models.ForeignKey(Timetable, on_delete=models.CASCADE, verbose_name="emploi du temps")
@@ -65,6 +72,7 @@ class Group(models.Model):
     subgroup = models.CharField(max_length=1, verbose_name="sous-groupe", null=True)
     td = models.IntegerField(verbose_name="groupe de TD", null=True)
     tp = models.IntegerField(verbose_name="groupe de TP", null=True)
+    parent_group = models.ForeignKey("self", verbose_name="groupe parent", null=True, default=None, related_name="children")
 
     slug = models.SlugField(max_length=64, default="")
 
diff --git a/utils.py b/utils.py
index 8dc386f..8630036 100644
--- a/utils.py
+++ b/utils.py
@@ -54,8 +54,8 @@ def parse_group(name):
     #  ([\w ]+?)                                                      correspond à au moins un caractère
     #           (\s*                                                  zéro, un ou plusieurs espaces
     #               (((CM)(\w))|                                      correspond à CM suivi d'une lettre ou...
-    #                           ((TD)(\w)(\d))|                       ... à TD suivi d'une lettre et d'un chiffre ou...
-    #                                          ((TP)(\w)(\d)(\d)))    ... à TP suivi d'une lettre et de deux chiffres
+    #                           ((TD)(\w)(\d))|                       ... à TD suivi d’une lettre et d'un chiffre ou...
+    #                                          ((TP)(\w)(\d)(\d)))    ... à TP suivi d’une lettre et de deux chiffres
     #                                                             )?  groupe optionel
     #                                                               $ fin de la ligne
     group_regex = re.compile("^([\w ]+?)(\s*(((CM)(\w))|((TD)(\w)(\d))|((TP)(\w)(\d)(\d))))?$")
diff --git a/views.py b/views.py
index 9365695..176f174 100644
--- a/views.py
+++ b/views.py
@@ -24,8 +24,8 @@ from .models import Timetable, LastUpdate, Group, Subscription, Course
 from .utils import get_current_week, get_week, group_courses
 
 def index(request):
-    timetables = Timetable.objects.all().order_by("name")
-    groups = Group.objects.filter(tp__isnull=False).order_by("name")
+    timetables = Timetable.objects.order_by("name")
+    groups = Group.objects.get_relevant_groups().order_by("name")
 
     year, week = get_current_week()
     start, _ = get_week(year, week)
-- 
cgit v1.2.1