diff --git a/rowers/forms.py b/rowers/forms.py index 1ad57343..2bb383e7 100644 --- a/rowers/forms.py +++ b/rowers/forms.py @@ -540,7 +540,7 @@ class TrendFlexModalForm(forms.Form): initial = types.waterboattype) rankingonly = forms.BooleanField(initial=False, label='Only Ranking Pieces', - required=True) + required=False) # This form sets options for the summary stats page diff --git a/rowers/views.py b/rowers/views.py index c77ebef0..09f32763 100644 --- a/rowers/views.py +++ b/rowers/views.py @@ -5112,6 +5112,10 @@ def team_comparison_select(request, else: waterboattype = types.waterboattype + if 'rankingonly' in request.session: + rankingonly = request.session['rankingonly'] + else: + rankingonly = False if 'modalities' in request.session: modalities = request.session['modalities'] @@ -5130,8 +5134,6 @@ def team_comparison_select(request, enddate = dateform.cleaned_data['enddate'] startdatestring = startdate.strftime('%Y-%m-%d') enddatestring = enddate.strftime('%Y-%m-%d') - request.session['startdate'] = startdatestring - request.session['enddate'] = enddatestring else: dateform = DateRangeForm(initial={ 'startdate':startdate, @@ -5152,6 +5154,11 @@ def team_comparison_select(request, if modality != 'water': waterboattype = [b[0] for b in types.boattypes] + + if 'rankingonly' in modalityform.cleaned_data: + rankingonly = modalityform.cleaned_data['rankingonly'] + else: + rankingonly = False request.session['modalities'] = modalities request.session['waterboattype'] = waterboattype @@ -5203,6 +5210,9 @@ def team_comparison_select(request, startdatetime__lte=enddate, workouttype__in=modalities).order_by("-date", "-starttime").exclude(boattype__in=negtypes) + if rankingonly: + workouts = [w for w in workouts if w.rankingpiece] + query = request.GET.get('q') if query: query_list = query.split() @@ -5224,7 +5234,8 @@ def team_comparison_select(request, chartform = ChartParamChoiceForm(initial={'teamid':0}) modalityform = TrendFlexModalForm(initial={ 'modality':modality, - 'waterboattype':waterboattype + 'waterboattype':waterboattype, + 'rankingonly':rankingonly, })