Вопрос Помогите мне!! напиши программу, вычисляющую новую стоимость арендной платы

Напиши программу, вычисляющую новую стоимость арендной платы


Цена за сутки в отеле может меняться в зависимости от дня недели. В отеле Summer:

- сутки в дни недели ПН, ВТ, СР, ЧТ стоят 2100,

- сутки в дни недели ПТ, СБ, ВС стоят 2850.



Клиент отметил на бланке желаемые даты бронирования. Напиши программу, считающую общую стоимость бронирования.



Программа должна печатать только число. Например:

1000000.

Нейросети для даунов, всё ясно...

В реальной жизни нужны дополнительные условия: Стоимость вычисляется по дню приезда. День приезда и день отъезда считается одним днем.

import datetime

def calculate_total_cost(arrival_date, departure_date):
arrival_day_of_week = arrival_date.weekday()
departure_day_of_week = departure_date.weekday()
if arrival_day_of_week in [0, 1, 2, 3]:
daily_rate = 2100 # ПН, ВТ, СР, ЧТ
else:
daily_rate = 2850 # ПТ, СБ, ВС
total_days = (departure_date - arrival_date).days
total_cost = total_days * daily_rate
return total_cost

arrival_date_str = input("Введите дату приезда в формате ГГГГ-ММ-ДД: ")
departure_date_str = input("Введите дату отъезда в формате ГГГГ-ММ-ДД: ")
arrival_date = datetime.datetime.strptime(arrival_date_str, "%Y-%m-%d").date()
departure_date = datetime.datetime.strptime(departure_date_str, "%Y-%m-%d").date()
total_cost = calculate_total_cost(arrival_date, departure_date)
print("Общая стоимость бронирования:", total_cost)

Калькулятор

.