From 9116b9efe1037168f60003e22ebf4631d20a73c3 Mon Sep 17 00:00:00 2001 From: SDE Date: Wed, 15 Jan 2025 19:55:09 +0300 Subject: [PATCH] 2.1.21 fix cargo_type choices --- RoutesApp/forms.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/RoutesApp/forms.py b/RoutesApp/forms.py index 2a47be6..ff83157 100644 --- a/RoutesApp/forms.py +++ b/RoutesApp/forms.py @@ -35,12 +35,22 @@ class RouteForm(forms.ModelForm): self.fields['to_city'].required = True self.fields['type_transport'].required = True + data = None + if kwargs and 'data' in kwargs: + data= kwargs['data'] + if owner_type == 'mover': self.fields['departure_DT'].required = True self.fields['type_transport'].choices = type_transport_choices[:-1] else: self.fields['type_transport'].choices = type_transport_choices + cargo_types = copy.deepcopy(cargo_type_choices) + if data and 'type_transport' in data and data['type_transport'] == 'avia': + cargo_types = cargo_types[:2] + cargo_types[3:] + self.fields['cargo_type'].choices = cargo_types + + def clean(self): # print('check') cleaned_data = super(RouteForm, self).clean()