强制实施严格的字段.使用棉花糖反序列化的日期格式



我正在将Marshmallow 2.15.3用于Flask项目,并希望强制执行严格的Date和DateTime格式。严格的意思是我只想接受与以下格式相同的字符串。我正在经历的是日期和日期时间的处理方式存在一些差异。格式:

DATE_FORMAT = '%Y-%m-%d'
DATETIME_FORMAT = '%Y-%m-%dT%H:%M:%S'

例如,使用日期时间,我可以执行以下操作:

dt = fields.DateTime(format=DATETIME_FORMAT)
dt.deserialize('2018-01-01')  # fails, as desired
dt.deserialize('2018-01-01T05:06:08.012312+02:00')  # fails, as desired
dt.deserialize('2018-01-01T05:06:08')  # works, as desired and according to format

使用日期我可以执行以下操作:

d = fields.Date() # does not accept format argument
d.deserialize('2018-01')  # fails, as desired
d.deserialize('2018-01-01T05:06:08.012312+02:00')  # works, NOT as desired
d.deserialize('2018-01-01')  # works, as desired and according to format

虽然日期时间不允许额外的信息,但日期允许。据我了解,日期字段没有format参数。有什么方法可以解决此问题以实现类似的功能,并对太短和太长的输入值强制执行我的严格格式?

对于面向未来的代码,我发现 Marshmallow 3.0.0b17 中的Date类现在是DateTime的子类而不是Field,使其继承了formatkwarg(相关提交(。

对于版本 2.15.3 和(通常是 2.X.X(,我找不到任何内置函数。解决方法是猴子修补fields.Date类。经过修改,它看起来像这样:

class Date(Field):
"""ISO8601-formatted date string.
:param kwargs: The same keyword arguments that :class:`Field` receives.
"""
default_error_messages = {
'invalid': 'Not a valid date.',
'format': '"{input}" cannot be formatted as a date.',
}
def __init__(self, format=None, **kwargs):
super(Date, self).__init__(**kwargs)
self.dateformat = format
def _serialize(self, value, attr, obj):
if value is None:
return None
try:
return value.isoformat()
except AttributeError:
self.fail('format', input=value)
return value
def _deserialize(self, value, attr, data):
"""Deserialize an ISO8601-formatted date string to a
:class:`datetime.date` object.
"""
if not value:  # falsy values are invalid
self.fail('invalid')
elif self.dateformat:
try:
return dt.datetime.strptime(value, self.dateformat).date()
except (TypeError, AttributeError, ValueError):
raise self.fail('invalid')
try:
return utils.from_iso_date(value)
except (AttributeError, TypeError, ValueError):
self.fail('invalid')

此处的修改是对__init__定义的添加,_deserialize下添加了整个elif self.dateformat-子句。这允许我使用提交的格式反序列化,例如:

d = fields.Date('%Y-%m-%d') # now accepts a format
d.deserialize('2018-01-01T05:06:08.012312+02:00')  # fails, as desired

最新更新