dateutils 默认为最后出现的识别部分，而不是下一个

Question

我正在使用dateutils.parser.parse 来解析可能包含部分信息的日期字符串。如果某些信息不存在，parse 可以采用 default 关键字参数，从中填充任何缺失的字段。此默认默认为datetime.datetime.today()。

对于像dateutil.parser.parse("Thursday") 这样的情况，这意味着它将返回下一个星期四的日期。但是，我需要它返回上一个星期四的日期（包括今天，如果今天恰好是星期四）。

所以，假设today == datetime.datetime(2018, 2, 20)（星期二），我希望所有这些asserts 都是真的：

from dateutil import parser
from datetime import datetime

def parse(date_str, default=None):
    # this needs to be modified
    return parser.parse(date_str, default=default)

today = datetime(2018, 2, 20)

assert parse("Tuesday", default=today) == today    # True
assert parse("Thursday", default=today) == datetime(2018, 2, 15)    # False
assert parse("Jan 31", default=today) == datetime(2018, 1, 31)    # True
assert parse("December 10", default=today) == datetime(2017, 12, 10)    # False

有没有简单的方法来实现这一点？使用当前的 parse 函数，只有第一个和第三个 assert 会通过。

Answer 1

这是您修改后的代码（code.py）：

#!/usr/bin/env python3

import sys
from dateutil import parser
from datetime import datetime, timedelta


today = datetime(2018, 2, 20)

data = [
    ("Tuesday", today, today),
    ("Thursday", datetime(2018, 2, 15), today),
    ("Jan 31", datetime(2018, 1, 31), today),
    ("December 10", datetime(2017, 12, 10), today),
]


def parse(date_str, default=None):
    # this needs to be modified
    return parser.parse(date_str, default=default)


def _days_in_year(year):
    try:
        datetime(year, 2, 29)
    except ValueError:
        return 365
    return 366


def parse2(date_str, default=None):
    dt = parser.parse(date_str, default=default)
    if default is not None:
        weekday_strs = [day_str.lower() for day_tuple in parser.parserinfo.WEEKDAYS for day_str in day_tuple]
        if date_str.lower() in weekday_strs:
            if dt.weekday() > default.weekday():
                dt -= timedelta(days=7)
        else:
            if (dt.month > today.month) or ((dt.month == today.month) and (dt.day > today.day)):
                dt -= timedelta(days=_days_in_year(dt.year))
    return dt


def print_stats(parse_func):
    print("\nPrinting stats for \"{:s}\"".format(parse_func.__name__))
    for triple in data:
        d = parse_func(triple[0], default=triple[2])
        print("  [{:s}] [{:s}] [{:s}] [{:s}]".format(triple[0], str(d), str(triple[1]), "True" if d == triple[1] else "False"))


if __name__ == "__main__":
    print("Python {:s} on {:s}\n".format(sys.version, sys.platform))
    print_stats(parse)
    print_stats(parse2)

注意事项：

• 我“稍微”更改了代码的结构，以对其进行参数化，因此如果需要更改（例如要添加新示例），更改应该是最小的
  • 我添加了一个函数 (print_stats)，而不是 asserts，它打印结果（而不是引发 AssertError 并在出现问题时退出程序不匹配）
    • 接受一个参数 (parse_func)，它是一个执行解析的函数（例如 parse）
    • 将一些全局声明的数据 (data) 与（上述）函数一起使用
  • data - 是一个三元组列表，其中每个三元组包含：
    1. 要转换的文本
    2. 预计日期时间 ([Python 3.Docs]: datetime Objects) 将由转换产生
    3. 默认参数被传递给解析函数（parse_func）

• parse2 函数（parse的改进版）：

  • 接受 2 种类型的日期字符串：
    1. 工作日名称
    2. 月/日（无序）
  • 进行常规解析，如果转换后的对象在之后作为默认参数传递（通过比较两个对象的适当属性来确定） ，它减去一个句点（看看[Python 3.Docs]: timedelta Objects）：
    1. “星期四”在“星期二”之后，因此它减去一周中的天数 (7)
    2. “12 月 10 日”在“2 月 20 日”之后，因此减去了一年中的天数^*

  • weekday_strs：我最好举例说明一下：

    >>> parser.parserinfo.WEEKDAYS
    [('Mon', 'Monday'), ('Tue', 'Tuesday'), ('Wed', 'Wednesday'), ('Thu', 'Thursday'), ('Fri', 'Friday'), ('Sat', 'Saturday'), ('Sun', 'Sunday')]
    >>> [day_str.lower() for day_tuple in parser.parserinfo.WEEKDAYS for day_str in day_tuple]
    ['mon', 'monday', 'tue', 'tuesday', 'wed', 'wednesday', 'thu', 'thursday', 'fri', 'friday', 'sat', 'saturday', 'sun', 'sunday']
    

    • 展平parser.parserinfo.WEEKDAYS
    • 将字符串转换为小写（用于简化比较）
• _days_in_year^* - 您可能已经猜到，返回一年中的天数（不能简单地减去 365，因为 leap 年可能会搞砸）：

  >>> dt = datetime(2018, 3, 1)
  >>> dt
  datetime.datetime(2018, 3, 1, 0, 0)
  >>> dt - timedelta(365)
  datetime.datetime(2017, 3, 1, 0, 0)
  >>> dt = datetime(2016, 3, 1)
  >>> dt
  datetime.datetime(2016, 3, 1, 0, 0)
  >>> dt - timedelta(365)
  datetime.datetime(2015, 3, 2, 0, 0)
  

输出：

(py35x64_test) E:\Work\Dev\StackOverflow\q048884480>"e:\Work\Dev\VEnvs\py35x64_test\Scripts\python.exe" code.py
Python 3.5.4 (v3.5.4:3f56838, Aug  8 2017, 02:17:05) [MSC v.1900 64 bit (AMD64)] on win32


Printing stats for "parse"
  [Tuesday] [2018-02-20 00:00:00] [2018-02-20 00:00:00] [True]
  [Thursday] [2018-02-22 00:00:00] [2018-02-15 00:00:00] [False]
  [Jan 31] [2018-01-31 00:00:00] [2018-01-31 00:00:00] [True]
  [December 10] [2018-12-10 00:00:00] [2017-12-10 00:00:00] [False]

Printing stats for "parse2"
  [Tuesday] [2018-02-20 00:00:00] [2018-02-20 00:00:00] [True]
  [Thursday] [2018-02-15 00:00:00] [2018-02-15 00:00:00] [True]
  [Jan 31] [2018-01-31 00:00:00] [2018-01-31 00:00:00] [True]
  [December 10] [2017-12-10 00:00:00] [2017-12-10 00:00:00] [True]