Pythonで月末、月初。ついでに月を加減するadd_months

月初はまあいいとしても、月末、月の足し引きは、こんな方法しか思いつかず。

カッコ悪いが使えることは使える、と思う。

2008.3.14追記

  • calendarモジュールを使って月の日数を取得することで、月末計算をシンプルに。
  • add_monthsも見直し。

それほどカッコ悪くはなくなったんじゃないかと思う。修正の経緯

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""月末、月初のdatetime.dateオブジェクトを返す。
Oracleのadd_months互換の月加減。
"""
import datetime
from calendar import monthrange

def first_day(date):
	"""月初を返す"""
	return datetime.date(date.year, date.month, 1)

def is_last_day(date):
	"""月末日付ならTrueを返す"""
	return days_of_month(date.year, date.month) == date.day

def days_of_month(year,month):
	"""年,月の日数を返す"""
	return monthrange(year, month)[1]

def last_day(date):
	"""月末を返す"""
	return datetime.date(year=date.year, month=date.month, day=days_of_month(date.year, date.month))

def add_months(date,months):
	"""月を加減する。
	dateにはdatetime.dateクラスのオブジェクト
	monthsには整数で月数を指定する。
	月末をOracleのadd_months互換の方法で処理する。
	例えば、
	2007年2月28日(月末)に1ヶ月足すと3月31日(月末)。
	2008年2月29日(月末)に1ヶ月足すと2008年3月31日(月末)。
	2008年2月28日(月末ではない)に1ヶ月足すと2008年3月28日(同じ日)。
	"""
	if months==0:
		return date

	year, month = divmod(date.month + months, 12)
	year = year + date.year

	#ちょうど割り切れたら12月で、マイナス1年。
	if month == 0:
		month = 12
		year = year - 1

	#入力日付がその月の月末なら、加算後月の日数を。
	#そうじゃなければ入力日付の日。
	day = date.day
	if date.day > days_of_month(year, month):
		day = days_of_month(year, month)
	return datetime.date(year=year, month=month, day=day)

最新版はGist:227554をどうぞ。