diff options
Diffstat (limited to 'servo/forms')
| -rw-r--r-- | servo/forms/repairs.py | 19 |
1 files changed, 16 insertions, 3 deletions
diff --git a/servo/forms/repairs.py b/servo/forms/repairs.py index 8459e52..b821ea9 100644 --- a/servo/forms/repairs.py +++ b/servo/forms/repairs.py @@ -67,12 +67,13 @@ class GsxRepairForm(forms.ModelForm): model = Repair exclude = [] widgets = { - 'device' : forms.HiddenInput(), + 'device': forms.HiddenInput(), 'parts': forms.CheckboxSelectMultiple(), 'unit_received_at': DateTimePickerInput(attrs={'readonly': 'readonly'}) } def __init__(self, *args, **kwargs): + from servo.lib.utils import empty super(GsxRepairForm, self).__init__(*args, **kwargs) repair = kwargs['instance'] techs = User.techies.filter(location=repair.order.location) @@ -84,13 +85,25 @@ class GsxRepairForm(forms.ModelForm): self.fields['parts'].initial = repair.order.get_parts() if not repair.can_mark_complete: - del self.fields['mark_complete'] - del self.fields['replacement_sn'] + del(self.fields['mark_complete']) + del(self.fields['replacement_sn']) choices = Template.templates() for f in ('notes', 'symptom', 'diagnosis'): self.fields[f].widget = AutocompleteTextarea(choices=choices) + symptom_codes = self.instance.get_symptom_code_choices() + self.fields['symptom_code'] = forms.ChoiceField(choices=symptom_codes, + label=_('Symptom group')) + + if empty(self.instance.symptom_code): + # default to the first choice + self.instance.symptom_code = symptom_codes[0][0] + + issue_codes = self.instance.get_issue_code_choices() + self.fields['issue_code'] = forms.ChoiceField(choices=issue_codes, + label=_('Issue code')) + def clean(self, *args, **kwargs): cd = super(GsxRepairForm, self).clean(*args, **kwargs) if self.instance.has_serialized_parts(): |