使用dateutil.parser.parse将ValueError抛出两位数的年份日期

Mihai Todor 发表于 Dev

米海·托多（Mihai Todor）

在进行一些数据清理时，我注意到它dateutil.parser.parse未能拒绝某个格式错误的日期，以为其中的第一个数字是两位数的年份。可以强制将此库视为两位数的年份无效吗？

例子：

from dateutil.parser import parse
parse('22-23 February')

输出：

datetime.datetime(2022, 2, 23, 0, 0)

米海·托多（Mihai Todor）

我设法通过将自定义dateutil.parser.parserinfo对象通过parserinfo参数传递给来解决此问题dateutil.parser.parse。幸运的是，dateutil.parser.parserinfo有一个convertyear方法可以在派生类中重载，以便在年度中执行额外的验证。

from dateutil.parser import parse, parserinfo

class NoTwoDigitYearParserInfo(parserinfo):
    def convertyear(self, year, century_specified=False):
        if year < 100 and not century_specified:
            raise ValueError('Two digit years are not supported.')
        return parserinfo.convertyear(self, year, century_specified)

parse('22-23 February', parserinfo = NoTwoDigitYearParserInfo())

输出：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.5/site-packages/dateutil/parser.py", line 1162, in parse
    return parser(parserinfo).parse(timestr, **kwargs)
  File "/usr/local/lib/python3.5/site-packages/dateutil/parser.py", line 552, in parse
    res, skipped_tokens = self._parse(timestr, **kwargs)
  File "/usr/local/lib/python3.5/site-packages/dateutil/parser.py", line 1055, in _parse
    if not info.validate(res):
  File "/usr/local/lib/python3.5/site-packages/dateutil/parser.py", line 360, in validate
    res.year = self.convertyear(res.year, res.century_specified)
  File "<stdin>", line 4, in convertyear
ValueError: Two digit years are not supported.

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。