diff options
Diffstat (limited to 'chimere/forms.py')
| -rw-r--r-- | chimere/forms.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/chimere/forms.py b/chimere/forms.py index 84ae27b..ed6b5d7 100644 --- a/chimere/forms.py +++ b/chimere/forms.py @@ -125,10 +125,14 @@ class MarkerAdminFormBase(forms.ModelForm): keys['initial'].update(property_dct) else: keys['initial'] = property_dct + subcategories = keys.pop('subcategories') \ + if 'subcategories' in keys else [] super(MarkerAdminFormBase, self).__init__(*args, **keys) if settings.CHIMERE_DAYS_BEFORE_EVENT: self.fields['start_date'].widget = DatePickerWidget() self.fields['end_date'].widget = DatePickerWidget() + if subcategories: + self.fields['categories'].choices = subcategories def clean(self): ''' @@ -203,10 +207,14 @@ class RouteAdminForm(forms.ModelForm): keys['initial'].update(property_dct) else: keys['initial'] = property_dct + subcategories = keys.pop('subcategories') \ + if 'subcategories' in keys else [] super(RouteAdminForm, self).__init__(*args, **keys) if settings.CHIMERE_DAYS_BEFORE_EVENT: self.fields['start_date'].widget = DatePickerWidget() self.fields['end_date'].widget = DatePickerWidget() + if subcategories: + self.fields['categories'].choices = subcategories def save(self, *args, **keys): """ |