refactor date forms in formutils

larigira/formutils.py

@@ -1,6 +1,13 @@
+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):
@@ -21,3 +28,47 @@ 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)
+        else:
+            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))

larigira/timeform_base.py

@@ -1,57 +1,14 @@
 import logging
-log = logging.getLogger('timeform_base')
 
 from datetime import datetime
 from pytimeparse.timeparse import timeparse
 
 from flask_wtf import Form
-from wtforms import StringField, Field, validators, SubmitField, \
+from wtforms import StringField, validators, SubmitField, \
         SelectMultipleField, ValidationError
-import wtforms.widgets
 
-
-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)
-        else:
-            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))
+from larigira.formutils import EasyDateTimeField
+log = logging.getLogger(__name__)
 
 
 class SingleAlarmForm(Form):