From d7e64ea30bf678044a1f9187853aca5ca7c9ebc4 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Fri, 28 Jan 2022 15:52:59 +0100 Subject: [PATCH] use request.periods only for students --- teleforma/views/core.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/teleforma/views/core.py b/teleforma/views/core.py index a88281fa..7bb4040e 100644 --- a/teleforma/views/core.py +++ b/teleforma/views/core.py @@ -143,21 +143,22 @@ def get_host(request): def get_periods(request): - period_ids = request.session.get('period_ids') user = request.user + periods = [] - if period_ids: - periods = [Period.objects.get(id=period_id) for period_id in period_ids] - else: - periods = [] - student = user.student.all() - if student: + student = user.student.all() + if student: + period_ids = request.session.get('period_ids') + if period_ids: + periods = [Period.objects.get(id=period_id) for period_id in period_ids] + else: student = user.student.get() periods = [training.period for training in student.trainings.all()] for period in periods: for child in period.children.all(): periods.append(child) - + request.session['period_ids'] = [period.id for period in periods] + else: if user.is_superuser or user.is_staff: periods = Period.objects.filter(is_open=True) @@ -172,8 +173,6 @@ def get_periods(request): if not quota.period in periods: periods.append(quota.period) - request.session['period_ids'] = [period.id for period in periods] - return periods -- 2.39.5