forms.py
110 lines
| 4.0 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r45 | import os | |
|
Juan C. Espinoza
|
r23 | import json | |
Miguel Urco
|
r13 | from django import forms | |
|
Juan C. Espinoza
|
r45 | from django.utils.safestring import mark_safe | |
| from apps.main.models import Device | |||
| from apps.main.forms import add_empty_choice | |||
| r346 | from .models import PedestalConfiguration | ||
|
Juan C. Espinoza
|
r25 | ||
|
Juan C. Espinoza
|
r149 | def create_choices_from_model(model, conf_id, all_choice=False): | |
|
Juan C. Espinoza
|
r172 | ||
| r346 | instance = globals()[model] | ||
| choices = instance.objects.all().values_list('pk', 'name') | |||
|
Juan C. Espinoza
|
r172 | ||
|
Juan C. Espinoza
|
r79 | return choices | |
|
Juan C. Espinoza
|
r45 | ||
| r346 | class PedestalConfigurationForm(forms.ModelForm): | ||
|
Juan C. Espinoza
|
r172 | ||
|
Juan C. Espinoza
|
r45 | def __init__(self, *args, **kwargs): | |
| r346 | super(PedestalConfigurationForm, self).__init__(*args, **kwargs) | ||
|
Juan C. Espinoza
|
r172 | ||
|
Juan C. Espinoza
|
r45 | instance = getattr(self, 'instance', None) | |
|
Juan C. Espinoza
|
r172 | ||
|
Juan C. Espinoza
|
r45 | if instance and instance.pk: | |
|
Juan C. Espinoza
|
r172 | ||
| r345 | devices = Device.objects.filter(device_type__name='pedestal') | ||
| r360 | #if instance.experiment: | ||
| #self.fields['experiment'].widget.attrs['read_only'] = True | |||
|
Juan C. Espinoza
|
r172 | #self.fields['experiment'].widget.choices = [(instance.experiment.id, instance.experiment)] | |
| r360 | #self.fields['device'].widget.choices = [(device.id, device) for device in devices] | ||
|
Juan C. Espinoza
|
r172 | ||
|
Juan C. Espinoza
|
r85 | if 'initial' in kwargs and 'experiment' in kwargs['initial'] and kwargs['initial']['experiment'] not in (0, '0'): | |
|
Juan C. Espinoza
|
r116 | self.fields['experiment'].widget.attrs['readonly'] = True | |
|
Juan C. Espinoza
|
r172 | ||
|
Miguel Urco
|
r13 | class Meta: | |
| r346 | model = PedestalConfiguration | ||
| r360 | exclude = ('template', 'type', 'parameters', 'status', 'total_units', 'author', 'hash') | ||
|
Juan C. Espinoza
|
r23 | ||
|
Juan C. Espinoza
|
r79 | def clean(self): | |
| r346 | form_data = super(PedestalConfigurationForm, self).clean() | ||
|
Juan C. Espinoza
|
r79 | return form_data | |
|
Juan C. Espinoza
|
r107 | ||
|
Juan C. Espinoza
|
r316 | def save(self, *args, **kwargs): | |
| r346 | conf = super(PedestalConfigurationForm, self).save(*args, **kwargs) | ||
|
Juan C. Espinoza
|
r116 | conf.save() | |
| return conf | |||
|
Juan C. Espinoza
|
r172 | ||
| r360 | class PedestalEditionForm(forms.ModelForm): | ||
| def __init__(self, *args, **kwargs): | |||
| super(PedestalEditionForm, self).__init__(*args, **kwargs) | |||
| instance = getattr(self, 'instance', None) | |||
| if instance and instance.pk: | |||
| devices = Device.objects.filter(device_type__name='pedestal') | |||
| #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 = PedestalConfiguration | |||
| exclude = ('device', 'label', 'template', 'type', 'parameters', 'status', 'total_units', 'author', 'hash') | |||
| def clean(self): | |||
| form_data = super(PedestalEditionForm, self).clean() | |||
| return form_data | |||
| def save(self, *args, **kwargs): | |||
| conf = super(PedestalEditionForm, self).save(*args, **kwargs) | |||
| conf.save() | |||
| return conf | |||
| r346 | class ExtFileField(forms.FileField): | ||
| """ | |||
| Same as forms.FileField, but you can specify a file extension whitelist. | |||
|
Juan C. Espinoza
|
r107 | ||
| r346 | >>> 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!'] | |||
| """ | |||
|
Juan C. Espinoza
|
r328 | def __init__(self, *args, **kwargs): | |
| r346 | extensions = kwargs.pop("extensions") | ||
| self.extensions = [i.lower() for i in extensions] | |||
|
Juan C. Espinoza
|
r328 | ||
| r346 | super(ExtFileField, self).__init__(*args, **kwargs) | ||
|
Juan C. Espinoza
|
r328 | ||
| r346 | 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) | |||
|
Juan C. Espinoza
|
r328 | ||
| r346 | class PedestalImportForm(forms.Form): | ||
|
Juan C. Espinoza
|
r328 | ||
| r346 | file_name = ExtFileField(extensions=['.racp', '.json', '.dat']) |
