diff options
Diffstat (limited to 'models.py')
-rw-r--r-- | models.py | 8 |
1 files changed, 3 insertions, 5 deletions
@@ -119,6 +119,7 @@ class Room(models.Model): class CourseManager(GroupedManager): def __get_weeks(self, qs): extractYear = ExtractYear("begin") + qs = qs.group_by("groups", "year", "week").order_by("groups__name", "year", "week") if connection.vendor == "postgresql": return qs.annotate(week=ExtractWeek("begin"), year=extractYear) @@ -128,11 +129,8 @@ class CourseManager(GroupedManager): 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) + def get_weeks(self, **criteria): + qs = self.get_queryset().filter(**criteria) return self.__get_weeks(qs) |