forms.py
149 lines
| 5.5 KiB
| text/x-python
|
PythonLexer
| r348 | import os | ||
| import json | |||
| r381 | from queue import Empty | ||
| r348 | |||
| from django import forms | |||
| from django.utils.safestring import mark_safe | |||
| from apps.main.models import Device | |||
| from apps.main.forms import add_empty_choice | |||
| from .models import USRPTXConfiguration | |||
| def create_choices_from_model(model, conf_id, all_choice=False): | |||
| instance = globals()[model] | |||
| choices = instance.objects.all().values_list('pk', 'name') | |||
| return choices | |||
| r381 | |||
| def validators_pulse_2(data): | |||
| info = {} | |||
| if data['enable_2'] is True and None in data.values(): | |||
| for key, value in data.items(): | |||
| if value is None: | |||
| info[key] = "Ensure this value isn't empty." | |||
| raise forms.ValidationError(info) | |||
| def validators_textarea(data): | |||
| info = {} | |||
| if data['code_type_1'].pk == 19 and data['code_1'] == '': | |||
| info['code_1'] = "Ensure this value isn't empty." | |||
| try: | |||
| if data['code_type_2'].pk == 19 and data['code_2'] == '': | |||
| info['code_2'] = "Ensure this value isn't empty." | |||
| except: | |||
| pass | |||
| if info != {}: | |||
| raise forms.ValidationError(info) | |||
| r348 | class USRPTXConfigurationForm(forms.ModelForm): | ||
| def __init__(self, *args, **kwargs): | |||
| super(USRPTXConfigurationForm, self).__init__(*args, **kwargs) | |||
| instance = getattr(self, 'instance', None) | |||
| if instance and instance.pk: | |||
| devices = Device.objects.filter(device_type__name='usrp_tx') | |||
| r360 | #if instance.experiment: | ||
| #self.fields['experiment'].widget.attrs['read_only'] = True | |||
| r348 | #self.fields['experiment'].widget.choices = [(instance.experiment.id, instance.experiment)] | ||
| r360 | #self.fields['device'].widget.choices = [(device.id, device) for device in devices] | ||
| r348 | |||
| if 'initial' in kwargs and 'experiment' in kwargs['initial'] and kwargs['initial']['experiment'] not in (0, '0'): | |||
| self.fields['experiment'].widget.attrs['readonly'] = True | |||
| class Meta: | |||
| model = USRPTXConfiguration | |||
| exclude = ('type', 'parameters', 'status', 'total_units', 'author', 'hash') | |||
| def clean(self): | |||
| form_data = super(USRPTXConfigurationForm, self).clean() | |||
| r381 | |||
| kwargs_check_pulse2 = {} | |||
| kwargs_check_pulse2['enable_2'] = form_data.get('enable_2') | |||
| kwargs_check_pulse2['pulse_2'] = form_data.get('pulse_2') | |||
| kwargs_check_pulse2['code_type_2'] = form_data.get('code_type_2') | |||
| kwargs_check_pulse2['repetitions_2'] = form_data.get('repetitions_2') | |||
| validators_pulse_2(kwargs_check_pulse2) | |||
| kwargs_check_textarea = {} | |||
| kwargs_check_textarea['code_type_1'] = form_data.get('code_type_1') | |||
| kwargs_check_textarea['code_1'] = form_data.get('code_1') | |||
| kwargs_check_textarea['code_type_2'] = form_data.get('code_type_2') | |||
| kwargs_check_textarea['code_2'] = form_data.get('code_2') | |||
| validators_textarea(kwargs_check_textarea) | |||
| r348 | return form_data | ||
| def save(self, *args, **kwargs): | |||
| conf = super(USRPTXConfigurationForm, self).save(*args, **kwargs) | |||
| conf.save() | |||
| return conf | |||
| r360 | class USRPTXEditionForm(forms.ModelForm): | ||
| def __init__(self, *args, **kwargs): | |||
| super(USRPTXEditionForm, self).__init__(*args, **kwargs) | |||
| instance = getattr(self, 'instance', None) | |||
| if instance and instance.pk: | |||
| devices = Device.objects.filter(device_type__name='usrp_tx') | |||
| #if instance.experiment: | |||
| #self.fields['experiment'].widget.attrs['read_only'] = True | |||
| #self.fields['experiment'].widget.choices = [(instance.experiment.id, instance.experiment)] | |||
| #self.fields['device'].widget.choices = [(device.id, device) for device in devices] | |||
| if 'initial' in kwargs and 'experiment' in kwargs['initial'] and kwargs['initial']['experiment'] not in (0, '0'): | |||
| self.fields['experiment'].widget.attrs['readonly'] = True | |||
| class Meta: | |||
| model = USRPTXConfiguration | |||
| exclude = ('template', 'device', 'label', 'ip_address', 'daughterboard', 'antenna', 'type', 'parameters', 'status', 'total_units', 'author', 'hash') | |||
| def clean(self): | |||
| form_data = super(USRPTXEditionForm, self).clean() | |||
| return form_data | |||
| def save(self, *args, **kwargs): | |||
| conf = super(USRPTXEditionForm, self).save(*args, **kwargs) | |||
| conf.save() | |||
| return conf | |||
| r348 | class ExtFileField(forms.FileField): | ||
| """ | |||
| Same as forms.FileField, but you can specify a file extension whitelist. | |||
| >>> from django.core.files.uploadedfile import SimpleUploadedFile | |||
| >>> | |||
| >>> t = ExtFileField(ext_whitelist=(".pdf", ".txt")) | |||
| >>> | |||
| >>> t.clean(SimpleUploadedFile('filename.pdf', 'Some File Content')) | |||
| >>> t.clean(SimpleUploadedFile('filename.txt', 'Some File Content')) | |||
| >>> | |||
| >>> t.clean(SimpleUploadedFile('filename.exe', 'Some File Content')) | |||
| Traceback (most recent call last): | |||
| ... | |||
| ValidationError: [u'Not allowed filetype!'] | |||
| """ | |||
| def __init__(self, *args, **kwargs): | |||
| extensions = kwargs.pop("extensions") | |||
| self.extensions = [i.lower() for i in extensions] | |||
| super(ExtFileField, self).__init__(*args, **kwargs) | |||
| def clean(self, *args, **kwargs): | |||
| data = super(ExtFileField, self).clean(*args, **kwargs) | |||
| filename = data.name | |||
| ext = os.path.splitext(filename)[1] | |||
| ext = ext.lower() | |||
| if ext not in self.extensions: | |||
| raise forms.ValidationError('Not allowed file type: %s' % ext) | |||
| class USRPTXImportForm(forms.Form): | |||
| file_name = ExtFileField(extensions=['.racp', '.json', '.dat']) |
