将时间单位作为变量传递给时间增量函数中的关键字参数。

Question

有可能实现这样的目标吗？

def subtract_from_date(date, time_unit, num):
    return date - timedelta(time_unit=num)

基本上我觉得如果elif条件适用于各种输入时间单位，如分钟，小时，周，天等，我会感到懒得。

Answer 1

您可以使用字典并将其应用为映射参数：

timedelta(**{time_unit: num})

演示：

>>> from datetime import datetime, timedelta, date
>>> def subtract_from_date(date, time_unit, num):
...     return date - timedelta(**{time_unit: num})
... 
>>> subtract_from_date(date.today(), 'days', 5)
datetime.date(2015, 3, 21)