Задача 149 Средняя

Оплата упала, а заказ оформлен

Reservation резервирует товар и снимает резерв, если оплата не прошла. Но когда платёжный сервис падает, checkout всё равно отвечает «заказ оформлен». Исправьте Reservation: при ошибке резерв снимается, а исключение доходит до вызывающего кода.

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

Что with делает с исключением, если __exit__ вернул True?

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

Уберите return True — __exit__ вернёт None, и ошибка полетит дальше после отката.

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

STOCK = {1: 5}


class Reservation:
    def __init__(self, product_id, quantity):
        self.product_id = product_id
        self.quantity = quantity

    def __enter__(self):
        STOCK[self.product_id] -= self.quantity
        return self

    def __exit__(self, exc_type, exc, tb):
        if exc_type is not None:
            STOCK[self.product_id] += self.quantity
        return True


def charge(card):
    if card == "0000":
        raise ConnectionError("платёжный сервис недоступен")


def checkout(product_id, quantity, card):
    with Reservation(product_id, quantity):
        charge(card)
    return "заказ оформлен"


try:
    print(checkout(1, 2, "0000"))
except ConnectionError as e:
    print("ошибка:", e)
print("на складе:", STOCK[1])