python实现输入日期，确定星期

python实现输入日期，确定星期

# 输入日期，确定星期
import calendar


def check_week(str1):
    date = str1.split("-")
    year, month, day = int(date[0]), int(date[1]), int(date[2])
    ctrl_code = calendar.weekday(year, month, day)
    print("%s是" % str1, end="")
    if ctrl_code == 0:
        print("周一")
    elif ctrl_code == 1:
        print("周二")
    elif ctrl_code == 2:
        print("周三")
    elif ctrl_code == 3:
        print("周四")
    elif ctrl_code == 4:
        print("周五")
    elif ctrl_code == 5:
        print("周六")
    elif ctrl_code == 6:
        print("周日")


if __name__ == "__main__":
    str2 = input("请输入您要查询的日期：（如：1991-01-24）")
    check_week(str2)