from django.forms import DateTimeField, RadioSelect, TypedChoiceField, ValidationError from django.utils.translation import gettext_lazy as _ from misago.core.utils import parse_iso8601_string class IsoDateTimeField(DateTimeField): input_formats = ['iso8601'] def prepare_value(self, value): try: return value.isoformat() except AttributeError: return value def to_python(self, value): """ Validates that the input can be converted to a datetime. Returns a Python datetime.datetime object. """ if value in self.empty_values: return None try: return parse_iso8601_string(value) except ValueError: raise ValidationError(self.error_messages['invalid'], code='invalid') class YesNoSwitchBase(TypedChoiceField): def prepare_value(self, value): """normalize bools to binary 1/0 so field works on them too""" if value in (True, 'True', 'true', 1, '1'): return 1 else: return 0 def clean(self, value): return self.prepare_value(value) def YesNoSwitch(**kwargs): yes_label = kwargs.pop('yes_label', _("Yes")) no_label = kwargs.pop('no_label', _("No")) return YesNoSwitchBase( coerce=int, choices=[ (1, yes_label), (0, no_label), ], widget=RadioSelect(attrs={'class': 'yesno-switch'}), **kwargs )