forms.py
37 lines
| 1.4 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r2 | from django import forms | ||
|
Juan C. Espinoza
|
r6 | from django.utils.safestring import mark_safe | ||
| from .models import DeviceType, Device, Experiment | ||||
|
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) | ||||
| class NewExperimentForm(forms.ModelForm): | ||||
| def __init__(self, *args, **kwargs): | ||||
| super(NewExperimentForm, self).__init__(*args, **kwargs) | ||||
| 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
|
r2 | class Meta: | ||
| model = Experiment | ||||
| fields = ['name', 'alias', 'start_date', 'end_date'] | ||||
|
Juan C. Espinoza
|
r6 | class NewDeviceForm(forms.ModelForm): | ||
| class Meta: | ||||
| model = Device | ||||
| exclude = ['status'] | ||||
|
Juan C. Espinoza
|
r2 | |||
|
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'))) | ||||