Python 當前時間增長或減小一個月

問題

今天在以前的代碼中發現了一個bug，有個計算當前時間減小一個月的函數，其報出下面的異常信息：

ValueError: day is out of range for month

看一下代碼：

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日，因此就拋出了上面的異常信息。

---

解決辦法

對於日期操做，網上的寫法都不太同樣，並且不肯定存在什麼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-dateutil。

---

源碼分析

代碼在 https://github.com/dateutil/dateutil/blob/master/dateutil/relativedelta.py

判斷應該是在354行開始：

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

@完