swh-source/src/kr/co/swh/lecture/algorithm/Lv5-4.py

def totalDayFromCalendar(year, month, day):
    dayOfMonth = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30]
    i=1
    totaldays = 365 * (int(year) - 1)
    while i<year:
        if i % 4 == 0 and i % 100 != 0 or i % 400 == 0:
            totaldays+=1
        i += 1
    premonth = month - 1
    for b in range(len(dayOfMonth)):
        if premonth >= (b+1):
            totaldays += dayOfMonth[b]
    if month > 2 and year % 4 == 0 and year % 100 != 0 or year % 400 == 0:
            totaldays +=1
    totaldays +=1
    totaldays = totaldays + day
    return totaldays

def endDayFromTotalDay(year, month):
    if month == 2:
        lastday = 28
        if year % 4 == 0 and year % 100 != 0 or year % 400 == 0:
            lastday = 29
        else:
            lastday = 28
    else:
        if month == 4 or month == 6 or month == 9 or month == 11:
            lastday = 30
        else:
            lastday = 31
    return lastday

def calendar_month(year, month):
    yoil = ["일", "월", "화", "수", "목", "금", "토"]
    a = totalDayFromCalendar(year, month, 1)
    k = endDayFromTotalDay(year,month)
    print("%10d 년 %d 월" % (year, month))
    for b in range(6):
        print("%s" % yoil[b], end="      ")
    print("%s" % yoil[-1])
    for b in range(1, k+1):
        if b == 1:
            for c in range(7-(a % 7)):
                print("%2s" % " ", end="  ")
        if a%7 == 1:
            print()
        print("%2d" % b, end="  ")
        a+=1

y = input("년월을 입력:")

calendar_month(int(y[:4]), int(y[4:6]))