forms.py
202 lines
| 7.3 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r2 | from django import forms | ||
|
Juan C. Espinoza
|
r6 | from django.utils.safestring import mark_safe | ||
Fiorella Quino
|
r266 | from apps.main.models import Device, Experiment, Campaign, Location, Configuration | ||
|
Juan C. Espinoza
|
r219 | from django.template.defaultfilters import default | ||
|
Juan C. Espinoza
|
r2 | |||
Miguel Urco
|
r57 | FILE_FORMAT = ( | ||
| ('json', 'json'), | ||||
| ) | ||||
| DDS_FILE_FORMAT = ( | ||||
| ('json', 'json'), | ||||
| ('text', 'dds') | ||||
| ) | ||||
| RC_FILE_FORMAT = ( | ||||
| ('json', 'json'), | ||||
|
Juan C. Espinoza
|
r79 | ('text', 'racp'), | ||
| ('binary', 'dat'), | ||||
|
Miguel Urco
|
r57 | ) | ||
|
Fiorella Quino
|
r276 | JARS_FILE_FORMAT = ( | ||
| ('json', 'json'), | ||||
| ('racp', 'racp'), | ||||
| ('text', 'jars'), | ||||
| ) | ||||
|
Juan C. Espinoza
|
r2 | def add_empty_choice(choices, pos=0, label='-----'): | ||
| if len(choices)>0: | ||||
| choices = list(choices) | ||||
| choices.insert(0, (0, label)) | ||||
| return choices | ||||
| else: | ||||
| return [(0, label)] | ||||
|
Juan C. Espinoza
|
r6 | class DatepickerWidget(forms.widgets.TextInput): | ||
| def render(self, name, value, attrs=None): | ||||
| input_html = super(DatepickerWidget, self).render(name, value, attrs) | ||||
| html = '<div class="input-group date">'+input_html+'<span class="input-group-addon"><i class="glyphicon glyphicon-calendar"></i></span></div>' | ||||
| return mark_safe(html) | ||||
|
Juan C. Espinoza
|
r138 | class DateRangepickerWidget(forms.widgets.TextInput): | ||
| def render(self, name, value, attrs=None): | ||||
| start = attrs['start_date'] | ||||
|
Juan C. Espinoza
|
r172 | end = attrs['end_date'] | ||
|
Juan C. Espinoza
|
r196 | html = '''<div class="col-md-6 input-group date" style="float:inherit"> | ||
|
Juan C. Espinoza
|
r138 | <input class="form-control" id="id_start_date" name="start_date" placeholder="Start" title="" type="text" value="{}"> | ||
| <span class="input-group-addon"><i class="glyphicon glyphicon-calendar"></i></span> | ||||
| </div> | ||||
|
Juan C. Espinoza
|
r196 | <div class="col-md-6 input-group date" style="float:inherit"> | ||
|
Juan C. Espinoza
|
r138 | <input class="form-control" id="id_end_date" name="end_date" placeholder="End" title="" type="text" value="{}"> | ||
| <span class="input-group-addon"><i class="glyphicon glyphicon-calendar"></i></span> | ||||
| </div>'''.format(start, end) | ||||
| return mark_safe(html) | ||||
|
Miguel Urco
|
r13 | class TimepickerWidget(forms.widgets.TextInput): | ||
| def render(self, name, value, attrs=None): | ||||
| input_html = super(TimepickerWidget, self).render(name, value, attrs) | ||||
| html = '<div class="input-group time">'+input_html+'<span class="input-group-addon"><i class="glyphicon glyphicon-time"></i></span></div>' | ||||
| return mark_safe(html) | ||||
| class CampaignForm(forms.ModelForm): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | experiments = forms.ModelMultipleChoiceField(widget=forms.CheckboxSelectMultiple(), | ||
| queryset=Experiment.objects.filter(template=True), | ||||
| required=False) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r6 | def __init__(self, *args, **kwargs): | ||
|
Miguel Urco
|
r13 | super(CampaignForm, self).__init__(*args, **kwargs) | ||
|
Juan C. Espinoza
|
r6 | self.fields['start_date'].widget = DatepickerWidget(self.fields['start_date'].widget.attrs) | ||
| self.fields['end_date'].widget = DatepickerWidget(self.fields['end_date'].widget.attrs) | ||||
|
Juan C. Espinoza
|
r85 | self.fields['description'].widget.attrs = {'rows': 2} | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r106 | if self.instance.pk: | ||
|
Juan C. Espinoza
|
r91 | self.fields['experiments'].queryset |= self.instance.experiments.all() | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r13 | class Meta: | ||
| model = Campaign | ||||
|
Miguel Urco
|
r47 | exclude = [''] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r13 | class ExperimentForm(forms.ModelForm): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r13 | def __init__(self, *args, **kwargs): | ||
| super(ExperimentForm, self).__init__(*args, **kwargs) | ||||
| self.fields['start_time'].widget = TimepickerWidget(self.fields['start_time'].widget.attrs) | ||||
| self.fields['end_time'].widget = TimepickerWidget(self.fields['end_time'].widget.attrs) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r266 | def save(self): | ||
| exp = super(ExperimentForm, self).save() | ||||
| exp.name = exp.name.replace(' ', '') | ||||
| exp.save() | ||||
| return exp | ||||
|
Juan C. Espinoza
|
r2 | class Meta: | ||
| model = Experiment | ||||
| r311 | exclude = ['task', 'status'] | |||
|
Juan C. Espinoza
|
r2 | |||
|
Miguel Urco
|
r41 | class LocationForm(forms.ModelForm): | ||
| class Meta: | ||||
| model = Location | ||||
| exclude = [''] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r13 | class DeviceForm(forms.ModelForm): | ||
|
Juan C. Espinoza
|
r6 | class Meta: | ||
| model = Device | ||||
| exclude = ['status'] | ||||
|
Juan C. Espinoza
|
r2 | |||
|
Miguel Urco
|
r13 | class ConfigurationForm(forms.ModelForm): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | def __init__(self, *args, **kwargs): | ||
| super(ConfigurationForm, self).__init__(*args, **kwargs) | ||||
|
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'): | ||
| self.fields['experiment'].widget.attrs['disabled'] = 'disabled' | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r13 | class Meta: | ||
| model = Configuration | ||||
|
Miguel Urco
|
r47 | exclude = ['type', 'created_date', 'programmed_date', 'parameters'] | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r53 | class UploadFileForm(forms.Form): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r53 | file = forms.FileField() | ||
|
Miguel Urco
|
r57 | |||
|
Miguel Urco
|
r53 | class DownloadFileForm(forms.Form): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r57 | format = forms.ChoiceField(choices= ((0, 'json'),) ) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r57 | def __init__(self, device_type, *args, **kwargs): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r57 | super(DownloadFileForm, self).__init__(*args, **kwargs) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r57 | self.fields['format'].choices = FILE_FORMAT | ||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r57 | if device_type == 'dds': | ||
| self.fields['format'].choices = DDS_FILE_FORMAT | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Miguel Urco
|
r57 | if device_type == 'rc': | ||
| self.fields['format'].choices = RC_FILE_FORMAT | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r276 | if device_type == 'jars': | ||
| self.fields['format'].choices = JARS_FILE_FORMAT | ||||
|
Fiorella Quino
|
r50 | class OperationForm(forms.Form): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r69 | campaign = forms.ChoiceField(label="Campaign") | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r69 | def __init__(self, *args, **kwargs): | ||
|
Juan C. Espinoza
|
r172 | |||
| campaigns = kwargs.pop('campaigns') | ||||
|
Fiorella Quino
|
r69 | super(OperationForm, self).__init__(*args, **kwargs) | ||
|
Juan C. Espinoza
|
r196 | self.fields['campaign'].label = 'Current Campaigns' | ||
| self.fields['campaign'].choices = add_empty_choice(campaigns.values_list('id', 'name')) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r138 | |||
|
Fiorella Quino
|
r69 | class OperationSearchForm(forms.Form): | ||
| # -----ALL Campaigns------ | ||||
| campaign = forms.ChoiceField(label="Campaign") | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r69 | def __init__(self, *args, **kwargs): | ||
| super(OperationSearchForm, self).__init__(*args, **kwargs) | ||||
| self.fields['campaign'].choices=Campaign.objects.all().order_by('-start_date').values_list('id', 'name') | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r138 | |||
|
Juan C. Espinoza
|
r85 | class NewForm(forms.Form): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | create_from = forms.ChoiceField(choices=((0, '-----'), | ||
| (1, 'Empty (blank)'), | ||||
| (2, 'Template'))) | ||||
| choose_template = forms.ChoiceField() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | def __init__(self, *args, **kwargs): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r85 | template_choices = kwargs.pop('template_choices', []) | ||
| super(NewForm, self).__init__(*args, **kwargs) | ||||
| self.fields['choose_template'].choices = add_empty_choice(template_choices) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r138 | |||
| class FilterForm(forms.Form): | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r138 | def __init__(self, *args, **kwargs): | ||
| extra_fields = kwargs.pop('extra_fields', []) | ||||
| super(FilterForm, self).__init__(*args, **kwargs) | ||||
|
Juan C. Espinoza
|
r172 | |||
| for field in extra_fields: | ||||
|
Juan C. Espinoza
|
r138 | if 'range_date' in field: | ||
| self.fields[field] = forms.CharField(required=False) | ||||
| self.fields[field].widget = DateRangepickerWidget() | ||||
| if 'initial' in kwargs: | ||||
| self.fields[field].widget.attrs = {'start_date':kwargs['initial'].get('start_date', ''), | ||||
|
Juan C. Espinoza
|
r172 | 'end_date':kwargs['initial'].get('end_date', '')} | ||
Juan C. Espinoza
|
r306 | elif field in ('template', 'historical'): | ||
| self.fields[field] = forms.BooleanField(required=False) | ||||
|
Juan C. Espinoza
|
r138 | else: | ||
| self.fields[field] = forms.CharField(required=False) | ||||
|
Juan C. Espinoza
|
r219 | |||
| class ChangeIpForm(forms.Form): | ||||
|
Fiorella Quino
|
r266 | |||
|
Juan C. Espinoza
|
r219 | ip_address = forms.GenericIPAddressField() | ||
| mask = forms.GenericIPAddressField(initial='255.255.255.0') | ||||
| gateway = forms.GenericIPAddressField(initial='0.0.0.0') | ||||
|
Fiorella Quino
|
r266 | dns = forms.GenericIPAddressField(initial='0.0.0.0') | ||
|
Juan C. Espinoza
|
r219 | |||