Python 当前时间增长或减小一个月

问题

今天在以前的代码中发现了一个bug,有个计算当前时间减小一个月的函数,其报出下面的异常信息:python

ValueError: day is out of range for month

看一下代码:git

import datatime
def _last_month(now_time):
    last_month = now_time.month - 1
    last_year = now_time.year
    if last_month == 0:
        last_month = 12
        last_year -= 1
    month_time = datetime.datetime(month=last_month, year=last_year, day=now_time.day)
    return month_time

缘由

问题出如今day=now_time.day上。后来想了一下,发现问题缘由是3月30日减小一个月是2月30日,而2月没有30日,因此就抛出了上面的异常信息。github


解决办法

对于日期操做,网上的写法都不太同样,并且不肯定存在什么bug。日期函数是靠时间来验证的,没准一年之后就出现了(我这个bug是在指定的3月29日之后才能出现,神奇不:D)。 因此我找了一个现有的日期扩展库,但愿别人已经踩过大部分坑了。代码以下函数

import datetime
from dateutil.relativedelta import relativedelta

if __name__ == "__main__":
    print(datetime.date.today() - relativedelta(months=+1))

能够看出,主要是使用relativedelta类。初始化参数months是月的差别。若是想增长一个月,那么就变成datetime.date.today() + relativedelta(months=+1),就是减号变加号。源码分析

安装这个库也很简单,执行命令pip install python-dateutilcode


源码分析

代码在 https://github.com/dateutil/dateutil/blob/master/dateutil/relativedelta.pyip

判断应该是在354行开始:get

if self.months:
            assert 1 <= abs(self.months) <= 12
            month += self.months
            if month > 12:
                year += 1
                month -= 12
            elif month < 1:
                year -= 1
                month += 12
        day = min(calendar.monthrange(year, month)[1],
                  self.day or other.day)

参考: https://dateutil.readthedocs.io/en/stable/ https://github.com/dateutil/dateutil源码

@完it

相关文章
相关标签/搜索