|
|
import json
|
|
|
|
|
|
from django import forms
|
|
|
from .models import RCConfiguration, RCLine, RCLineType, RCLineCode
|
|
|
|
|
|
def create_choices_from_model(model, conf_id):
|
|
|
|
|
|
if model=='RCLine':
|
|
|
instance = RCConfiguration.objects.get(pk=conf_id)
|
|
|
return instance.get_refs_lines()
|
|
|
else:
|
|
|
instance = globals()[model]
|
|
|
return instance.objects.all().values_list('pk', 'name')
|
|
|
|
|
|
class RCConfigurationForm(forms.ModelForm):
|
|
|
|
|
|
class Meta:
|
|
|
model = RCConfiguration
|
|
|
exclude = ('clock', 'ipp', 'ntx', 'clock_divider')
|
|
|
|
|
|
|
|
|
class RCLineForm(forms.ModelForm):
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
self.extra_fields = kwargs.pop('extra_fields', [])
|
|
|
super(RCLineForm, self).__init__(*args, **kwargs)
|
|
|
if 'initial'in kwargs:
|
|
|
for item in self.extra_fields:
|
|
|
if 'model' in item:
|
|
|
self.fields[item['name']] = forms.ChoiceField(choices=create_choices_from_model(item['model'],
|
|
|
kwargs['initial']['rc_configuration']))
|
|
|
else:
|
|
|
self.fields[item['name']] = forms.CharField(initial=item['value'])
|
|
|
|
|
|
class Meta:
|
|
|
model = RCLine
|
|
|
fields = ('rc_configuration', 'line_type', 'channel')
|
|
|
widgets = {
|
|
|
'channel': forms.HiddenInput(),
|
|
|
}
|
|
|
|
|
|
def save(self):
|
|
|
line = super(RCLineForm, self).save()
|
|
|
|
|
|
#auto add channel
|
|
|
line.channel = RCLine.objects.filter(rc_configuration=line.rc_configuration).count()-1
|
|
|
|
|
|
#auto add position for TX, TR & CODE
|
|
|
if line.line_type.name in ('tx', 'tr', 'codes', 'windows'):
|
|
|
line.position = RCLine.objects.filter(rc_configuration=line.rc_configuration, line_type=line.line_type).count()-1
|
|
|
|
|
|
#save extra fields in params
|
|
|
params = {}
|
|
|
for item in self.extra_fields:
|
|
|
params[item['name']] = self.data[item['name']]
|
|
|
line.params = json.dumps(params)
|
|
|
line.save()
|
|
|
return
|
|
|
|
|
|
class RCLineViewForm(forms.Form):
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
extra_fields = kwargs.pop('extra_fields')
|
|
|
super(RCLineViewForm, self).__init__(*args, **kwargs)
|
|
|
for label, value in extra_fields.items():
|
|
|
if 'ref' in label:
|
|
|
value = RCLine.objects.get(pk=value).get_name()
|
|
|
elif 'code' in label:
|
|
|
value = RCLineCode.objects.get(pk=value).name
|
|
|
self.fields[label] = forms.CharField(initial=value)
|
|
|
|
|
|
class RCLineEditForm(forms.Form):
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
self.extra_fields = kwargs.pop('extra_fields', [])
|
|
|
super(RCLineEditForm, self).__init__(*args, **kwargs)
|
|
|
if 'initial'in kwargs:
|
|
|
for item in self.extra_fields:
|
|
|
if 'model' in item:
|
|
|
self.fields[item['name']] = forms.ChoiceField(choices=create_choices_from_model(item['model'],
|
|
|
kwargs['initial']['rc_configuration']),
|
|
|
initial=item['value'],
|
|
|
widget=forms.Select(attrs={'name':'%s|%s' % (kwargs['initial']['line'], item['name'])}))
|
|
|
else:
|
|
|
self.fields[item['name']] = forms.CharField(initial=item['value'],
|
|
|
widget=forms.TextInput(attrs={'name':'%s|%s' % (kwargs['initial']['line'], item['name'])}))
|
|
|
|
