Задача 134 Лёгкая

Новый способ доставки ломает оформление

Добавили доставку в пункт выдачи — класс PickupDelivery с тем же методом calc_price(total), что у курьера и почты. Оформление заказа с ней падает с ValueError. Перепишите checkout_total(total, delivery) так, чтобы она работала с любым способом доставки, у которого есть calc_price(total), и не упоминала классы доставки по имени.

Подсказка 1 из 2

Правила цены уже записаны в calc_price каждого класса — и продублированы в функции.

Подсказка 2 из 2

return total + delivery.calc_price(total).

main.py Загружаем Python… первый раз ~13 МБ, дальше из кэша

from decimal import Decimal


class CourierDelivery:
    def calc_price(self, total):
        return Decimal("0.00") if total >= 3000 else Decimal("299.00")


class PostDelivery:
    def calc_price(self, total):
        return Decimal("350.00")


class PickupDelivery:
    def calc_price(self, total):
        return Decimal("0.00")


def checkout_total(total, delivery):
    if isinstance(delivery, CourierDelivery):
        price = Decimal("0.00") if total >= 3000 else Decimal("299.00")
    elif isinstance(delivery, PostDelivery):
        price = Decimal("350.00")
    else:
        raise ValueError("неизвестный способ доставки")
    return total + price


print(checkout_total(Decimal("1499.00"), PickupDelivery()))