qcg/fields.py

   1 # coding=utf-8
   2 from datetime import timedelta
   3 from django import forms
   4 from django.core.exceptions import ValidationError
   5
   6
   7 class TimeRangeWidget(forms.MultiWidget):
   8     def decompress(self, value):
   9         if not value:
  10             return [None, None]
  11
  12         seconds = int(value.total_seconds())
  13
  14         if seconds % 60 != 0:
  15             return [seconds, TimeRangeField.SECONDS]
  16         elif seconds % 3600 != 0:
  17             return [seconds / 60, TimeRangeField.MINUTES]
  18         elif seconds % 86400 != 0:
  19             return [seconds / 3600, TimeRangeField.HOURS]
  20         else:
  21             return [seconds / 86400, TimeRangeField.DAYS]
  22
  23
  24 class TimeRangeField(forms.MultiValueField):
  25     SECONDS, MINUTES, HOURS, DAYS = range(4)
  26     UNIT_CHOICES = (
  27         (SECONDS, u"Sekund"),
  28         (MINUTES, u"Minut"),
  29         (HOURS, u"Godzin"),
  30         (DAYS, u"Dni"),
  31     )
  32
  33     value = forms.IntegerField(min_value=0, required=False)
  34     unit = forms.TypedChoiceField(choices=UNIT_CHOICES, coerce=int, required=False)
  35
  36     widget = TimeRangeWidget(widgets=(value.widget, unit.widget))
  37
  38     def __init__(self, *args, **kwargs):
  39         super(TimeRangeField, self).__init__((self.value, self.unit), *args, **kwargs)
  40
  41     def compress(self, data_list):
  42         value, unit = data_list
  43
  44         if value is None:
  45             return None
  46
  47         if unit == self.SECONDS:
  48             return timedelta(seconds=value)
  49         elif unit == self.MINUTES:
  50             return timedelta(minutes=value)
  51         elif unit == self.HOURS:
  52             return timedelta(hours=value)
  53         elif unit == self.DAYS:
  54             return timedelta(days=value)
  55
  56         raise ValidationError(u"Nieprawidłowa jednostka")