1 files changed, 9 insertions, 10 deletions
diff --git a/views.py b/views.py
index 24473d1..c549cfb 100644
--- a/views.py
+++ b/views.py
@@ -18,18 +18,17 @@ def index(request):
 
     current_year, current_week, _ = timezone.now().isocalendar()
     start, _ = get_week(current_year, current_week)
+    groups_weeks = Course.objects.get_weeks(start)
 
     for group in groups:
-        courses = Course.objects.get_courses_for_group(group, begin__gte=start).values("begin").annotate(count=Count("begin"))
-        weeks = []
-        for course in courses:
-            year, week, _ = course["begin"].isocalendar()
-            date, _ = get_week(year, week)
-
-            if date not in weeks:
-                weeks.append(date)
-
-        group.weeks = weeks
+        for group_week in groups_weeks:
+            if group.corresponds_to(group_week.timetable_id, group_week.mention, group_week.subgroup, group_week.td, group_week.tp):
+                if not hasattr(group, "weeks"):
+                    group.weeks = []
+
+                date, _ = get_week(group_week.year, group_week.week)
+                if date not in group.weeks:
+                    group.weeks.append(date)
 
     return render(request, "index.html", {"timetables": timetables, "groups": groups})