面试题编程输入一个年份，输出这一年2月份有多少天

这几天面试真的是身心疲惫，大厂的一个外派安排4面，到最后我都说不动了，这是实话，下面是某厂四面的时候测试组长问我的，如何用python来实现这个算法，回答的一般吧，找个小本本记一下，万一下次再问。
题目16：
编程输入一个年份，输出这一年2月份有多少天

year = int(input("请输入年份:"))
month = int(input("请输入月份:"))

day=1
if month >= 1 and month <=12:
    if month in(1,3,5,7,8,10,12):
        day = 31
    elif month in (4,6,9,11):
        day = 30
    else:
        if year % 400 == 0 or (year % 4 ==0 and year % 100 != 0):
            day = 29
        else:
            day=28
    print("%d年%d月有%d天" % (year,month,day))
else:
    print("月输入有误！")