写了个python生成日历的东东

一个项目需要一个月的日历生成5x7的列表，要求列表里的没个日期为datetime类型，之前用了个龌龊方式实现，结果偶尔出现bug。

于是重写一个，用到python自带的 calendar 模块。

最终生成结果如下：

python test.py 2014 09
2014-08-31 2014-09-01 2014-09-02 2014-09-03 2014-09-04 2014-09-05 2014-09-06
2014-09-07 2014-09-08 2014-09-09 2014-09-10 2014-09-11 2014-09-12 2014-09-13
2014-09-14 2014-09-15 2014-09-16 2014-09-17 2014-09-18 2014-09-19 2014-09-20
2014-09-21 2014-09-22 2014-09-23 2014-09-24 2014-09-25 2014-09-26 2014-09-27
2014-09-28 2014-09-29 2014-09-30 2014-10-01 2014-10-02 2014-10-03 2014-10-04

python代码：

#coding:utf-8                                                                                                           
# Last modified: 2014-08-20 20:08:58
# by zhangdi http://jondy.net/
import calendar
import datetime
import sys

def getcal(y, m):
  # 从周日开始
  cal = calendar.Calendar(6)
  if not isinstance(y, int): y = int(y)
  if not isinstance(m, int): m = int(m)
  if m == 1: # 1月份
    py = y - 1; pm = 12;
    ny = y; nm = 2
  elif m == 12: # 12月份
    py = y; pm = 11
    ny = y + 1; nm = 1
  else:
    py = y; pm = m - 1
    ny = y; nm = m + 1
  pcal = cal.monthdayscalendar(py, pm) # 上一月
  ncal = cal.monthdayscalendar(ny, nm) # 下一月
  ccal = cal.monthdayscalendar(y, m)   # 当前
  w1 = ccal.pop(0) # 取第一周
  w2 = ccal.pop()  # 取最后一周
  wp = pcal.pop()  # 上个月的最后一周
  wn = ncal.pop(0) # 下个月的第一周
  #r1 = [datetime.date(y, m ,w1[i]) or wp[i] for i in range(7)]
  r1 = [w1[i] and datetime.date(y, m, w1[i]) or datetime.date(py, pm, wp[i]) for i in range(7)]
  r2 = [w2[i] and datetime.date(y, m, w2[i]) or datetime.date(ny, nm, wn[i]) for i in range(7)]
  # 转datetime
  result = []
  result.append(r1) # 第一周
  for c in ccal:    # 其他周
    result.append([datetime.date(y,m,i) for i in c])
  result.append(r2) # 最后一周
  return result

if __name__ == '__main__':
  for w in getcal(sys.argv[1], sys.argv[2]):
    for d in w:
      print d,
    print