51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from wtforms.fields.html5 import DateField
|
|
from wtforms.fields import Field, SelectField as SelectField_
|
|
from wtforms.widgets import TextArea
|
|
|
|
from atst.domain.date import parse_date
|
|
|
|
|
|
class DateField(DateField):
|
|
def _value(self):
|
|
if self.data:
|
|
return parse_date(self.data)
|
|
else:
|
|
return None
|
|
|
|
def process_formdata(self, values):
|
|
if values:
|
|
self.data = values[0]
|
|
else:
|
|
self.data = []
|
|
|
|
|
|
class NewlineListField(Field):
|
|
widget = TextArea()
|
|
|
|
def _value(self):
|
|
if isinstance(self.data, list):
|
|
return "\n".join(self.data)
|
|
elif self.data:
|
|
return self.data
|
|
else:
|
|
return ""
|
|
|
|
def process_formdata(self, valuelist):
|
|
if valuelist:
|
|
self.data = [l.strip() for l in valuelist[0].split("\n") if l]
|
|
else:
|
|
self.data = []
|
|
|
|
def process_data(self, value):
|
|
if isinstance(value, list):
|
|
self.data = "\n".join(value)
|
|
else:
|
|
self.data = value
|
|
|
|
|
|
class SelectField(SelectField_):
|
|
def __init__(self, *args, **kwargs):
|
|
render_kw = kwargs.get("render_kw", {})
|
|
kwargs["render_kw"] = {**render_kw, "required": False}
|
|
super().__init__(*args, **kwargs)
|