Дата неделю спустя

Напишите функцию, которая будет принимать дату в строковом формате и возвращать дату на неделю позже.

Примечания:

  • функция и принимает, и возвращает строку
  • дата должна иметь формат день/месяц/год
  • если число или месяц обозначаются одной цифрой, перед ней ставим ноль (т.е. 1 января — 01/01).

Обратите внимание на функцию strftime() и модуль datetime.

Примеры

week_after("12/03/2020") ➞ "19/03/2020"

week_after("21/12/1989") ➞ "28/12/1989"

week_after("01/01/2000") ➞ "08/01/2000"

Решение

from datetime import datetime, timedelta
def week_after(d):
    date = datetime.strptime(d,'%d/%m/%Y') + timedelta(days=7)
    return date.strftime('%d/%m/%Y')