diff --git a/rowers/models.py b/rowers/models.py index 7a1d12a9..5e0ae86e 100644 --- a/rowers/models.py +++ b/rowers/models.py @@ -3091,13 +3091,13 @@ class IndoorVirtualRaceResultForm(ModelForm): 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: @@ -3110,7 +3110,7 @@ class VirtualRaceResultForm(ModelForm): 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