13272.上周/本周/下周 周一的日期

题目链接

题目受版权保护，不可转载，参见上述链接

input

2019-03-21
2019-01-01

output

输入日期: 2019-03-21
上周一: 2019-03-11
本周一: 2019-03-18
下周一: 2019-03-25
输入日期: 2019-01-01
上周一: 2018-12-24
本周一: 2018-12-31
下周一: 2019-01-07

My Code

import sys
import datetime
import calendar


def last_monday(date_str):
    today = date_str.split("-")
    todayweekday = calendar.weekday(int(today[0]), int(today[1]), int(today[2]))
    d = datetime.datetime.strptime(date_str, "%Y-%m-%d")
    return (d + datetime.timedelta(days = -todayweekday - 7)).strftime("%Y-%m-%d")


def this_monday(date_str):
    today = date_str.split("-")
    todayweekday = calendar.weekday(int(today[0]), int(today[1]), int(today[2]))
    d = datetime.datetime.strptime(date_str, "%Y-%m-%d")
    return (d + datetime.timedelta(days = -todayweekday)).strftime("%Y-%m-%d")

    
def next_monday(date_str):
    today = date_str.split("-")
    todayweekday = calendar.weekday(int(today[0]), int(today[1]), int(today[2]))
    d = datetime.datetime.strptime(date_str, "%Y-%m-%d")
    return (d + datetime.timedelta(days = 7 - todayweekday )).strftime("%Y-%m-%d")


while True:
    line = sys.stdin.readline()
    line = line.strip()
    if line == '':
        break
    print('输入日期:', line)
    print('上周一:', last_monday(line))
    print('本周一:', this_monday(line))
    print('下周一:', next_monday(line))