如何从简单字符串构造timedelta对象

Question

我正在写一个函数，需要把字符串转换成一个timedelta（时间差）。用户必须输入像"32m"（32分钟）或者"2h32m"（2小时32分钟），甚至是"4:13"（4分钟13秒）或者"5hr34m56s"（5小时34分钟56秒）这样的格式……有没有什么库或者工具已经实现了这种功能？

Answer 1

昨天我有点闲暇时间，所以我把@virhilo的回答做成了一个Python模块，并且增加了一些新的时间表达格式，包括@priestc请求的所有格式。

如果有人想要的话，源代码可以在GitHub上找到（MIT许可证）。它也在PyPI上：

pip install pytimeparse

这个模块会把时间转换成秒数：

>>> from pytimeparse.timeparse import timeparse
>>> timeparse('32m')
1920
>>> timeparse('2h32m')
9120
>>> timeparse('4:13')
253
>>> timeparse('5hr34m56s')
20096
>>> timeparse('1.2 minutes')
72

Answer 2

在我看来，最优雅的解决方案就是不需要使用像 dateutil 这样的外部库，也不需要手动解析输入，而是使用 datetime 提供的强大 strptime 字符串解析方法。

from datetime import datetime, timedelta
# we specify the input and the format...
t = datetime.strptime("05:20:25","%H:%M:%S")
# ...and use datetime's hour, min and sec properties to build a timedelta
delta = timedelta(hours=t.hour, minutes=t.minute, seconds=t.second)

之后，你可以像平常一样使用你的 timedelta 对象，可以把它转换成秒数，以确保我们做的事情是正确的等等。

print(delta)
assert(5*60*60+20*60+25 == delta.total_seconds())

Answer 3

对于第一个格式（5hr34m56s），你应该使用正则表达式来解析。

下面是一个基于正则表达式的解决方案：

import re
from datetime import timedelta


regex = re.compile(r'((?P<hours>\d+?)hr)?((?P<minutes>\d+?)m)?((?P<seconds>\d+?)s)?')


def parse_time(time_str):
    parts = regex.match(time_str)
    if not parts:
        return
    parts = parts.groupdict()
    time_params = {}
    for name, param in parts.items():
        if param:
            time_params[name] = int(param)
    return timedelta(**time_params)


>>> from parse_time import parse_time
>>> parse_time('12hr')
datetime.timedelta(0, 43200)
>>> parse_time('12hr5m10s')
datetime.timedelta(0, 43510)
>>> parse_time('12hr10s')
datetime.timedelta(0, 43210)
>>> parse_time('10s')
datetime.timedelta(0, 10)
>>>