| @@ -9,12 +9,13 from apps.main.forms import add_empty_choice | |||||
| 9 | from .models import RCConfiguration, RCLine, RCLineType, RCLineCode |
|
9 | from .models import RCConfiguration, RCLine, RCLineType, RCLineCode | |
| 10 | from .widgets import KmUnitWidget, KmUnitHzWidget, KmUnitDcWidget, UnitKmWidget, DefaultWidget, CodesWidget, HiddenWidget, HCheckboxSelectMultiple |
|
10 | from .widgets import KmUnitWidget, KmUnitHzWidget, KmUnitDcWidget, UnitKmWidget, DefaultWidget, CodesWidget, HiddenWidget, HCheckboxSelectMultiple | |
| 11 |
|
11 | |||
| 12 | def create_choices_from_model(model, conf_id, all=False): |
|
12 | def create_choices_from_model(model, conf_id, all_choice=False): | |
| 13 |
|
13 | |||
| 14 | if model=='RCLine': |
|
14 | if model=='RCLine': | |
| 15 | instance = RCConfiguration.objects.get(pk=conf_id) |
|
15 | instance = RCConfiguration.objects.get(pk=conf_id) | |
| 16 | choices = [(line.pk, line.get_name()) for line in instance.get_lines(line_type__name='tx')] |
|
16 | choices = [(line.pk, line.get_name()) for line in instance.get_lines(line_type__name='tx')] | |
| 17 | choices = add_empty_choice(choices, label='All') |
|
17 | if all_choice: | |
|
|
18 | choices = add_empty_choice(choices, label='All') | |||
| 18 | else: |
|
19 | else: | |
| 19 | instance = globals()[model] |
|
20 | instance = globals()[model] | |
| 20 | choices = instance.objects.all().values_list('pk', 'name') |
|
21 | choices = instance.objects.all().values_list('pk', 'name') | |
| @@ -310,8 +311,12 class RCLineEditForm(forms.ModelForm): | |||||
| 310 | else: |
|
311 | else: | |
| 311 | help_text = '' |
|
312 | help_text = '' | |
| 312 |
|
313 | |||
| 313 |
if 'model' in params[label]: |
|
314 | if 'model' in params[label]: | |
| 314 | self.fields[label] = forms.ChoiceField(choices=create_choices_from_model(params[label]['model'], conf.id), |
|
315 | if line.line_type.name=='tr': | |
|
|
316 | all_choice = True | |||
|
|
317 | else: | |||
|
|
318 | all_choice = False | |||
|
|
319 | self.fields[label] = forms.ChoiceField(choices=create_choices_from_model(params[label]['model'], conf.id, all_choice=all_choice), | |||
| 315 | initial=value, |
|
320 | initial=value, | |
| 316 | widget=forms.Select(attrs={'name':'%s|%s|%s' % (count, line.id, label)}), |
|
321 | widget=forms.Select(attrs={'name':'%s|%s|%s' % (count, line.id, label)}), | |
| 317 | help_text=help_text) |
|
322 | help_text=help_text) | |
General Comments 0
You need to be logged in to leave comments.
Login now