Python打印指定日期日历

下面的代码只是演示其中的原理和思路，真正使用时可以直接调用Python标准库calendar中的方法。

from datetime import date

daysOfMonth = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

def myCalendar(year, month):

    #获取year年month月1日是周几

    start = date(year, month, 1).timetuple().tm_wday

    #打印头部信息

    print('{0}年{1}月日历'.format(year,month).center(56))

    print('\t'.join('日 一 二 三 四 五 六'.split()))

    #获取该月有多少天，如果是2月并且是闰年，适当调整一下

    day = daysOfMonth[month-1]

    if month==2:

        if year%400==0 or (year%4==0 and year%100!=0):

            day += 1

    #生成数据，根据需要在前面填充空白

    result = [' '*8 for i in range(start+1)]

    result += list(map(lambda d: str(d).ljust(8), range(1, day+1)))

    #打印数据

    for i, day in enumerate(result):

        if i!=0 and i%7==0:

            print()

        print(day, end='')

    print()

def main(year, month=-1):

    if type(year)!=int or year<1000 or year>10000:

        print('Year error')

        return

    if type(month)==int:

        #如果没有指定月份，就打印全年的日历

        if month==-1:

            for m in range(1, 13):

                myCalendar(year, m)

        #如果指定了月份，就只打印这一个月的日历

        elif month in range(1,13):

            myCalendar(year, month)

        else:

            print('Month error')

            return

    else:

        print('Month error')

        return

main(2017)