larigira/larigira/formutils.py

import logging

from wtforms.fields import StringField
from wtforms import Field
import wtforms.widgets

from datetime import datetime

log = logging.getLogger(__name__)


class AutocompleteTextInput(wtforms.widgets.Input):
    def __init__(self, datalist=None):
        super().__init__('text')
        self.datalist = datalist

    def __call__(self, field, **kwargs):
        # every second can be specified
        if self.datalist is not None:
            return super(AutocompleteTextInput, self).__call__(
                field, list=self.datalist, autocomplete="autocomplete",
                **kwargs)
        return super(AutocompleteTextInput, self).__call__(
            field, **kwargs)


class AutocompleteStringField(StringField):
    def __init__(self, datalist, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.widget = AutocompleteTextInput(datalist)


class DateTimeInput(wtforms.widgets.Input):
    input_type = 'datetime-local'

    def __call__(self, field, **kwargs):
        # every second can be specified
        return super(DateTimeInput, self).__call__(field, step='1', **kwargs)


class EasyDateTimeField(Field):
    '''
    a "fork" of DateTimeField which uses HTML5 datetime-local

    The format is not customizable, because it is imposed by the HTML5
    specification.

    This field does not ensure that browser actually supports datetime-local
    input type, nor does it provide polyfills.
    '''
    widget = DateTimeInput()
    formats = ('%Y-%m-%dT%H:%M:%S', '%Y-%m-%dT%H:%M')

    def __init__(self, label=None, validators=None, **kwargs):
        super(EasyDateTimeField, self).__init__(label, validators, **kwargs)

    def _value(self):
        if self.raw_data:
            return ' '.join(self.raw_data)
        return self.data and self.data.strftime(self.formats[0]) or ''

    def process_formdata(self, valuelist):
        if valuelist:
            date_str = ' '.join(valuelist)
            for fmt in self.formats:
                try:
                    self.data = datetime.strptime(date_str, fmt)
                    return
                except ValueError:
                    log.debug('Format `%s` not valid for `%s`',
                              fmt, date_str)
            raise ValueError(self.gettext(
                'Not a valid datetime value <tt>{}</tt>').format(date_str))