Откройте контекст
Выберите папку проекта и обозначьте нужные файлы или поведение.
Сценарии
Сценарии работы с кодом · Сдвиг рабочих дней с учётом праздников
03
Конкретная отправная точка
Сдвиньте дату на число рабочих дней от минус 366 до 366, пропуская выходные и до 1000 праздников. Проверьте ноль, обратное движение и границы календаря, отклоняйте неверные даты без изменения праздников.
Опишите изменение
Укажите входные данные, ожидаемый результат и границы, которые нужно сохранить.
Изучите работу
Проверьте предлагаемые изменения и команды, прежде чем полагаться на результат.
Проверьте результат
Сравните фактическое поведение с ожидаемым. Если проверка не проходит, разберитесь в причине перед следующими изменениями.
Сдвиг рабочих дней с учётом праздников
Конкретная отправная точка
from datetime import date, timedelta
def working_day_shift(start, days, holidays):
if type(start) is not date or type(days) is not int or not -366 <= days <= 366:
raise ValueError("days.input")
if type(holidays) is not set or len(holidays) > 1000 or any(type(d) is not date for d in holidays):
raise ValueError("days.holidays")
remaining, step = abs(days), 1 if days >= 0 else -1
for _ in range(7*(remaining+len(holidays))+7):
if remaining == 0:
return start
try:
start += timedelta(days=step)
except OverflowError as error:
raise ValueError("days.range") from error
if start.weekday() < 5 and start not in holidays:
remaining -= 1
raise RuntimeError("days.steps")Пример — отправная точка для реального проекта, а не утверждение, что сгенерированный код уже протестирован. Адаптируйте пути, зависимости и ожидаемое поведение к своему приложению.