46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import datetime
|
|
|
|
from django import forms
|
|
from django.utils.timezone import now
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from common.literals import TIME_DELTA_UNIT_CHOICES
|
|
|
|
|
|
class SplitTimeDeltaWidget(forms.widgets.MultiWidget):
|
|
"""
|
|
A Widget that splits a timedelta input into three <input type="text">
|
|
boxes.
|
|
"""
|
|
def __init__(self, attrs=None):
|
|
widgets = (
|
|
forms.widgets.NumberInput(
|
|
attrs={
|
|
'maxlength': 4, 'style': 'width: 8em;',
|
|
'placeholder': _('Period')
|
|
}
|
|
),
|
|
forms.widgets.Select(
|
|
attrs={'style': 'width: 8em;'}, choices=TIME_DELTA_UNIT_CHOICES
|
|
),
|
|
|
|
)
|
|
super(SplitTimeDeltaWidget, self).__init__(widgets, attrs)
|
|
|
|
def decompress(self, value):
|
|
return (None, None)
|
|
|
|
def value_from_datadict(self, querydict, files, name):
|
|
unit = querydict.get('{}_1'.format(name))
|
|
period = querydict.get('{}_0'.format(name))
|
|
|
|
if not unit or not period:
|
|
return now()
|
|
|
|
period = int(period)
|
|
|
|
timedelta = datetime.timedelta(**{unit: period})
|
|
return now() + timedelta
|