python编写的一个日历 签到中可用

#coding:utf-8
# Last modified: 2014-08-21 11:08:08
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 "=="