求一个判断天数的Python程序

https://img-mid.csdnimg.cn/release/static/image/mid/ask/633065449466144.jpg

import re

ymd=input('请输入yyyy-mm-dd格式日期：')
if re.match(r"^\d{4}(-\d{2}){2}$",ymd):
    year,month,day=map(int,ymd.split('-'))
    isRN=year%400==0 or (year%4==0 and year%100!=0)
    mth_days=[31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    for i in range(month-1):
        day+=mth_days[i]
    if month>2 and isRN:
        day+=1
    print('第%d天'%day)
else:
    print('input error')

import datetime


def which_day(year, month, date):
    end = datetime.date(year, month, date)
    start = datetime.date(year, 1, 1)
    return (end - start).days + 1


while True:
    date = input("请输入一个时间x-x-x格式, q退出：")
    if date in ['q', 'Q']:
        break
    try:
        Y, M, D = map(int, date.split("-"))
        print(which_day(Y, M, D))
    except Exception as e:
        print("input error")

您好，我是有问必答小助手，您的问题已经有小伙伴帮您解答，感谢您对有问必答的支持与关注！
PS：问答VIP年卡 【限时加赠：IT技术图书免费领】，了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632