問題
今天在之前的代碼中發現了一個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是月的差異。
安裝這個庫也很簡單,執行命令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
@完