aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAlban Gruin2017-01-26 13:45:24 +0100
committerAlban Gruin2017-01-26 13:58:30 +0100
commit98a7494b6f930490e62cfc134f23d69b32203977 (patch)
treef5d012987109b00e43707de2a3dfd24b433e9319
parent46e4a2d58503dd7c97056a4e543edcbc4e4ca353 (diff)
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)
-rw-r--r--models.py19
1 files 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()