diff --git a/rowers/models.py b/rowers/models.py index 7a1d12a9..3fea4dbb 100644 --- a/rowers/models.py +++ b/rowers/models.py @@ -3086,31 +3086,31 @@ class IndoorVirtualRaceResultForm(ModelForm): class Meta: model = IndoorVirtualRaceResult fields = ['teamname','weightcategory','boatclass','age','adaptiveclass', - #'entrycategory' + 'entrycategory' ] def __init__(self, *args, **kwargs): -# categories = kwargs.pop('categories',None) + categories = kwargs.pop('categories',None) super(IndoorVirtualRaceResultForm, self).__init__(*args, **kwargs) -# if categories is not None: -# self.fields['entrycategory'].queryset = categories -# self.fields['entrycategory'].empty_label = None -# else: -# self.fields.pop('entrycategory') + if categories is not None: + self.fields['entrycategory'].queryset = categories + self.fields['entrycategory'].empty_label = None + else: + self.fields.pop('entrycategory') class VirtualRaceResultForm(ModelForm): class Meta: model = VirtualRaceResult fields = ['teamname','weightcategory','boatclass','boattype', 'age','adaptiveclass', - #'entrycategory' + 'entrycategory' ] def __init__(self, *args, **kwargs): boattypes = kwargs.pop('boattypes',None) -# categories = kwargs.pop('categories',None) + categories = kwargs.pop('categories',None) super(VirtualRaceResultForm, self).__init__(*args, **kwargs) if boattypes: @@ -3120,11 +3120,11 @@ class VirtualRaceResultForm(ModelForm): required=False, label='Mixed Gender') -# if categories is not None: -# self.fields['entrycategory'].queryset = categories -# self.fields['entrycategory'].empty_label = None -# else: -# self.fields.pop('entrycategory') + if categories is not None: + self.fields['entrycategory'].queryset = categories + self.fields['entrycategory'].empty_label = None + else: + self.fields.pop('entrycategory') from rowers.metrics import rowingmetrics