forms.py
115 lines
| 4.5 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r157 | import os | ||
|
Juan C. Espinoza
|
r6 | from django import forms | ||
Fiorella Quino
|
r123 | from apps.main.models import Device, Experiment | ||
|
Fiorella Quino
|
r135 | from .models import JARSConfiguration, JARSfilter | ||
|
Fiorella Quino
|
r128 | from .widgets import SpectralWidget | ||
|
Fiorella Quino
|
r269 | from apps.main.forms import add_empty_choice | ||
| def create_choices_from_model(model, filter_id=None): | ||||
| #instance = globals()[model] | ||||
| choices = model.objects.all().values_list('pk', 'name') | ||||
| choices = add_empty_choice(choices) | ||||
| return choices | ||||
|
Juan C. Espinoza
|
r6 | |||
| class JARSConfigurationForm(forms.ModelForm): | ||||
| def __init__(self, *args, **kwargs): | ||||
| super(JARSConfigurationForm, self).__init__(*args, **kwargs) | ||||
| instance = getattr(self, 'instance', None) | ||||
|
Fiorella Quino
|
r263 | |||
|
Juan C. Espinoza
|
r6 | if instance and instance.pk: | ||
Miguel Urco
|
r26 | devices = Device.objects.filter(device_type__name='jars') | ||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r123 | if instance.experiment: | ||
| experiments = Experiment.objects.filter(pk=instance.experiment.id) | ||||
| self.fields['experiment'].widget.choices = [(experiment.id, experiment) for experiment in experiments] | ||||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r123 | self.fields['device'].widget.choices = [(device.id, device) for device in devices] | ||
|
Fiorella Quino
|
r131 | #self.fields['spectral'].widget = SpectralWidget() | ||
|
Fiorella Quino
|
r139 | self.fields['spectral_number'].widget.attrs['readonly'] = True | ||
|
Fiorella Quino
|
r131 | self.fields['spectral'].widget = SpectralWidget() | ||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r123 | #-------------JARS Configuration needs an Experiment----------------- | ||
| def clean(self): | ||||
| cleaned_data = super(JARSConfigurationForm, self).clean() | ||||
| experiment = cleaned_data.get('experiment') | ||||
|
Fiorella Quino
|
r276 | if not experiment: | ||
|
Fiorella Quino
|
r123 | msg = "Error: Jars Configuration needs an Experiment" | ||
| self.add_error('experiment', msg) | ||||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r123 | class Meta: | ||
| model = JARSConfiguration | ||||
|
Fiorella Quino
|
r136 | exclude = ('type', 'parameters', 'status', 'filter_parms') | ||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r135 | class JARSfilterForm(forms.ModelForm): | ||
| def __init__(self, *args, **kwargs): | ||||
| super(JARSfilterForm, self).__init__(*args, **kwargs) | ||||
| instance = getattr(self, 'instance', None) | ||||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r139 | self.fields['fch_decimal'].widget.attrs['readonly'] = True | ||
|
Fiorella Quino
|
r293 | |||
|
Fiorella Quino
|
r263 | if 'initial' in kwargs: | ||
|
Fiorella Quino
|
r269 | if 'filter_id' not in kwargs['initial']: | ||
| self.fields.pop('name') | ||||
| else: | ||||
| self.fields['name'] = forms.ChoiceField(choices=create_choices_from_model(JARSfilter)) | ||||
| filter_id = kwargs['initial']['filter_id'] | ||||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r269 | if filter_id == 0: | ||
| for value in self.fields: | ||||
| if value != 'name': | ||||
| self.fields.pop(value) | ||||
| self.fields['name'].label = "Filter Template Name" | ||||
| else: | ||||
| self.fields['name'] = forms.ChoiceField(choices=create_choices_from_model(JARSfilter, kwargs['initial']['filter_id'])) | ||||
|
Fiorella Quino
|
r293 | jars_filter = JARSfilter.objects.get(pk=kwargs['initial']['filter_id']) | ||
|
Fiorella Quino
|
r269 | labels = [f.name for f in jars_filter._meta.get_fields()] | ||
|
Fiorella Quino
|
r293 | |||
|
Fiorella Quino
|
r272 | for label in ['id']: | ||
|
Fiorella Quino
|
r269 | labels.remove(label) | ||
| for label in labels: | ||||
| self.fields['name'].initial = kwargs['initial']['filter_id'] | ||||
| self.fields[label].initial = getattr(jars_filter,label) | ||||
| self.fields['name'].label = "Filter Template Name" | ||||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r135 | class Meta: | ||
| model = JARSfilter | ||||
|
Juan C. Espinoza
|
r157 | exclude = ('type', 'parameters', 'status') | ||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r269 | |||
|
Juan C. Espinoza
|
r157 | 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) | ||||
|
Fiorella Quino
|
r263 | |||
|
Juan C. Espinoza
|
r157 | |||
| class JARSImportForm(forms.Form): | ||||
|
Fiorella Quino
|
r263 | |||
|
Fiorella Quino
|
r275 | file_name = ExtFileField(extensions=['.racp','.json']) | ||