Python中万年历的代码实现

Python 中万年历的实现
题目要求
万年历：
日历是以1900年1月1日00：00：00 为基础的 那天的星期1是已知的
时间戳 — 将已知时间获取其对应的秒数 — 是以1970年1月1号为基础

import math
#定义万年历函数
def calendar(year,month):
	#total_days记录输入年份的上一年到时间戳1900年的总天数；year_days记录输入月的一号是输入年的第几天；sum_days 是从1900年一月开始到输入月共有多少天。
    sum_days = 0
    total_days = 0
    year_days = 0
    days = 0
    遍历1900年到输入年的前一年，计算天数。其中闰年加366天，非闰年加365天。即total_days
    for i in range(year - 1,1899,-1):
        if i % 4 == 0 and i % 100 != 0 or i % 400 == 0:
            total_days += 366
        else:
            total_days += 365
     #遍历月份，计算输入年一月一号到输入月前面一个月的最后一天共有几天，即year_days，其中			    #二月还的判断次年是否闰年
    for j in range(month - 1,0,-1):
        if j in [1,3,5,7,8,10,12]:
            year_days += 31
        elif j in [4,6,9,11]:
            year_days += 30
        else:
            if year % 4 == 0 and yea