forms.py
102 lines
| 3.9 KiB
| text/x-python
|
PythonLexer
|
r157 | import os | ||
|
r6 | from django import forms | ||
|
r123 | from apps.main.models import Device, Experiment | ||
|
r316 | from .models import JARSConfiguration, JARSFilter | ||
|
r128 | from .widgets import SpectralWidget | ||
|
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 | ||||
|
r6 | |||
class JARSConfigurationForm(forms.ModelForm): | ||||
def __init__(self, *args, **kwargs): | ||||
super(JARSConfigurationForm, self).__init__(*args, **kwargs) | ||||
instance = getattr(self, 'instance', None) | ||||
|
r263 | |||
|
r6 | if instance and instance.pk: | ||
|
r26 | devices = Device.objects.filter(device_type__name='jars') | ||
|
r123 | self.fields['device'].widget.choices = [(device.id, device) for device in devices] | ||
|
r139 | self.fields['spectral_number'].widget.attrs['readonly'] = True | ||
|
r131 | self.fields['spectral'].widget = SpectralWidget() | ||
|
r263 | |||
|
r123 | class Meta: | ||
model = JARSConfiguration | ||||
|
r316 | exclude = ('type', 'parameters', 'status', 'filter_parms', 'author', 'hash', 'filter') | ||
|
r263 | |||
|
r316 | class JARSFilterForm(forms.ModelForm): | ||
|
r135 | def __init__(self, *args, **kwargs): | ||
|
r316 | super(JARSFilterForm, self).__init__(*args, **kwargs) | ||
|
r135 | instance = getattr(self, 'instance', None) | ||
|
r263 | |||
|
r316 | self.fields['f_decimal'].widget.attrs['readonly'] = True | ||
|
r293 | |||
|
r263 | if 'initial' in kwargs: | ||
|
r316 | self.fields['filter_template'] = forms.ChoiceField( | ||
choices=create_choices_from_model(JARSFilter), | ||||
initial = kwargs['initial']['id'] | ||||
) | ||||
# self.fields['name'].initial = kwargs['initial']['id'] | ||||
# filter_id = kwargs['initial']['filter_id'] | ||||
# 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'])) | ||||
# jars_filter = JARSFilter.objects.get(pk=kwargs['initial']['filter_id']) | ||||
# labels = [f.name for f in jars_filter._meta.get_fields()] | ||||
# for label in ['id']: | ||||
# 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" | ||||
|
r263 | |||
|
r135 | class Meta: | ||
|
r316 | model = JARSFilter | ||
exclude = ('name', ) | ||||
|
r263 | |||
|
r269 | |||
|
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) | ||||
|
r263 | |||
|
r157 | |||
class JARSImportForm(forms.Form): | ||||
|
r263 | |||
|
r275 | file_name = ExtFileField(extensions=['.racp','.json']) | ||