From 98a7494b6f930490e62cfc134f23d69b32203977 Mon Sep 17 00:00:00 2001 From: Alban Gruin Date: Thu, 26 Jan 2017 13:45:24 +0100 Subject: Fonction pour récupérer l'année et la semaine d'un cours directement depuis le manager Utilise la méthode la plus adaptée en fonction du SGBD (pgsql: utilisation de extract, autre: rawsql) --- models.py | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/models.py b/models.py index 62274d2..64fb675 100644 --- a/models.py +++ b/models.py @@ -1,6 +1,7 @@ -from django.db import models +from django.db import connection, models from django.db.models import Q -from django.db.models.functions import Extract +from django.db.models.expressions import RawSQL +from django.db.models.functions import Extract, ExtractYear from django.utils.text import slugify import hashlib @@ -111,9 +112,23 @@ class Room(models.Model): class CourseManager(models.Manager): + def __get_weeks(self, qs): + extractYear = ExtractYear("begin") + + if connection.vendor == "postgresql": + return qs.annotate(week=ExtractWeek("begin"), year=extractYear) + else: + return qs.annotate(week=RawSQL("""cast(strftime("%%W", "begin") as integer)""", []), year=extractYear) + def get_courses_for_group(self, group, **filters): return self.get_queryset().filter(Q(groups__td__isnull=True) | Q(groups__td=group.td), Q(groups__tp__isnull=True) | Q(groups__tp=group.tp), groups__mention=group.mention, groups__subgroup=group.subgroup, **filters).order_by("begin") + def get_weeks(self): + return self.__get_weeks(self.get_queryset()) + + def get_courses_for_group_and_weeks(self, group, **filters): + qs = self.get_courses_for_group(group, **filters) + return self.__get_weeks(qs) class Course(models.Model): objects = CourseManager() -- cgit v1.2.1