forms.py
79 lines
| 2.9 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r2 | from django import forms | |
|
Juan C. Espinoza
|
r6 | from django.utils.safestring import mark_safe | |
Miguel Urco
|
r41 | from .models import DeviceType, Device, Experiment, Campaign, Configuration, Location | |
|
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) | |||
|
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
|
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) | |||
|
Miguel Urco
|
r13 | class Meta: | |
| model = Campaign | |||
|
Miguel Urco
|
r47 | exclude = [''] | |
|
Miguel Urco
|
r13 | ||
| class ExperimentForm(forms.ModelForm): | |||
|
Miguel Urco
|
r53 | ||
|
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) | |||
| self.fields['campaign'].widget.attrs['readonly'] = True | |||
|
Juan C. Espinoza
|
r2 | class Meta: | |
| model = Experiment | |||
|
Miguel Urco
|
r47 | exclude = [''] | |
|
Juan C. Espinoza
|
r2 | ||
|
Miguel Urco
|
r41 | class LocationForm(forms.ModelForm): | |
| class Meta: | |||
| model = Location | |||
| exclude = [''] | |||
|
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): | |
| class Meta: | |||
| model = Configuration | |||
|
Miguel Urco
|
r47 | exclude = ['type', 'created_date', 'programmed_date', 'parameters'] | |
|
Miguel Urco
|
r13 | ||
|
Juan C. Espinoza
|
r6 | class DeviceTypeForm(forms.Form): | |
| device_type = forms.ChoiceField(choices=add_empty_choice(DeviceType.objects.all().order_by('name').values_list('id', 'name'))) | |||
Fiorella Quino
|
r50 | ||
|
Miguel Urco
|
r53 | ||
| class UploadFileForm(forms.Form): | |||
| file = forms.FileField() | |||
| class DownloadFileForm(forms.Form): | |||
| format = forms.ComboField() | |||
|
Fiorella Quino
|
r50 | class OperationForm(forms.Form): | |
|
Miguel Urco
|
r53 | # today = datetime.today() | |
|
Fiorella Quino
|
r50 | # -----Campaigns from this month------ | |
|
Miguel Urco
|
r53 | campaign = forms.ChoiceField(choices=Campaign.objects.all().order_by('-start_date').values_list('id', 'name')[:5], label="Campaign") | |
|
Fiorella Quino
|
r50 |