{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Композиция против наследования\n",
    "\n",
    "«Является» — наследование, «содержит» — композиция. Объект хранит другие объекты и делегирует им работу; почему class Menu(list) — плохая идея и как заменить части объекта на ходу.\n",
    "\n",
    "Курс ООП в Python · урок `oop-composition` · [урок на сайте](https://ds-edu-trainer.vercel.app/courses/oop/composition).\n",
    "\n",
    "Выполняйте ячейки по порядку (Shift+Enter), начиная с первой — она готовит урок. После каждого упражнения идёт\n",
    "ячейка проверки: она печатает ✓ / ✗ и итог."
   ],
   "id": "intro"
  },
  {
   "cell_type": "code",
   "metadata": {
    "cellView": "form",
    "jupyter": {
     "source_hidden": true
    }
   },
   "source": [
    "# @title Подготовка урока — выполните эту ячейку первой\n",
    "# Проверяет версии пакетов, записывает файлы данных в папку data/ и определяет _check — проверку упражнений.\n",
    "import base64, os, zlib\n",
    "\n",
    "import sys\n",
    "\n",
    "if sys.version_info < (3, 10):\n",
    "    print(f\"Урок написан для Python 3.10 и новее, у вас {sys.version.split()[0]} — часть кода может не работать\")\n",
    "\n",
    "# ─── Проверка упражнений: тот же раннер, что у задач на сайте ───\n",
    "# ─── Запуск тестов ───────────────────────────────────────────────────────────\n",
    "\n",
    "_TIMEOUT = 2.0  # секунд на тест; отдельный тест может задать свой: test_x.timeout = 10\n",
    "\n",
    "\n",
    "def _call_with_timeout(target, seconds: float) -> bool:\n",
    "    \"\"\"Запускает target() в отдельном потоке. Возвращает False, если за seconds он не завершился.\"\"\"\n",
    "    import sys\n",
    "    import threading\n",
    "\n",
    "    if sys.platform == \"emscripten\":  # Python в браузере: потоков нет, таймаут обеспечивает страница\n",
    "        target()\n",
    "        return True\n",
    "    # Большой стек нужен, чтобы глубокая рекурсия давала RecursionError, а не аварийное завершение Python.\n",
    "    for megabytes in (256, 64, 0):\n",
    "        try:\n",
    "            threading.stack_size(megabytes * 1024 * 1024)\n",
    "            worker = threading.Thread(target=target, daemon=True)\n",
    "            worker.start()\n",
    "        except (ValueError, RuntimeError, MemoryError):\n",
    "            continue\n",
    "        worker.join(seconds)\n",
    "        return not worker.is_alive()\n",
    "    target()  # потоки недоступны — запускаем без таймаута\n",
    "    return True\n",
    "\n",
    "\n",
    "def _test_info(fn) -> dict:\n",
    "    \"\"\"Описание теста, его лимит времени и сообщение на случай зависания.\"\"\"\n",
    "    limit = getattr(fn, \"timeout\", _TIMEOUT)\n",
    "    hint = getattr(fn, \"timeout_hint\", \"\")\n",
    "    message = (\n",
    "        f\"тест не завершился за {limit:g} с — возможно, бесконечный цикл или рекурсия: \"\n",
    "        \"проверьте, что границы/аргументы изменяются на каждом шаге\" + (f\". {hint}\" if hint else \"\")\n",
    "    )\n",
    "    return {\"title\": (fn.__doc__ or fn.__name__).strip(), \"timeout\": limit, \"timeout_message\": message}\n",
    "\n",
    "\n",
    "def _error_result(title: str, error: BaseException, user_file: str | None) -> dict:\n",
    "    \"\"\"Результат упавшего теста: статус, сообщение, тип исключения и строка в коде решения.\"\"\"\n",
    "    line = None\n",
    "    if user_file:  # самый глубокий кадр в коде решения; ошибка внутри самих тестов — без номера строки\n",
    "        tb = error.__traceback__\n",
    "        while tb is not None:\n",
    "            if tb.tb_frame.f_code.co_filename == user_file:\n",
    "                line = tb.tb_lineno\n",
    "            tb = tb.tb_next\n",
    "    status, message = \"error\", f\"{type(error).__name__}: {error}\"\n",
    "    if isinstance(error, NotImplementedError):\n",
    "        status, message, line = \"not_written\", \"функция ещё не написана\", None\n",
    "    elif isinstance(error, AssertionError):\n",
    "        status, message = \"failed\", str(error) or \"проверка не прошла\"\n",
    "    elif isinstance(error, RecursionError):\n",
    "        message = \"RecursionError: слишком глубокая рекурсия — проверьте базовый случай и то, что задача уменьшается\"\n",
    "    elif isinstance(error, MemoryError) or (isinstance(error, OverflowError) and \"index-sized\" in str(error)):\n",
    "        message = f\"{type(error).__name__}: не хватило памяти — возможно, создаётся слишком большая структура данных\"\n",
    "    return {\"title\": title, \"status\": status, \"message\": message, \"error_type\": type(error).__name__, \"line\": line}\n",
    "\n",
    "\n",
    "def _run_tests(verbose: bool = True, on_start=None, on_result=None, user_file: str | None = None) -> list[dict]:\n",
    "    \"\"\"Запускает тесты задачи (_TESTS — функции из tests.py) по порядку, каждый — с таймаутом.\n",
    "\n",
    "    Возвращает список результатов: {\"title\", \"status\", \"message\", \"error_type\", \"line\"}, где status —\n",
    "    passed · failed · error · not_written · timeout · not_run. Если тест завис, при verbose=True печатает\n",
    "    итог и завершает процесс: зависший поток остановить нельзя. on_start(i) и on_result(i, результат)\n",
    "    вызываются по мере выполнения; user_file — имя файла с кодом решения для номеров строк в ошибках.\n",
    "    \"\"\"\n",
    "    import os\n",
    "    import sys\n",
    "\n",
    "    if verbose:\n",
    "        try:  # ✓ и кириллица в консоли Windows со старой кодировкой\n",
    "            sys.stdout.reconfigure(encoding=\"utf-8\")\n",
    "        except Exception:  # noqa: BLE001 — reconfigure может не быть (перенаправленный вывод, браузер)\n",
    "            pass\n",
    "\n",
    "    tests = list(_TESTS)\n",
    "    results: list[dict] = []\n",
    "    not_written = hung = False\n",
    "    for index, fn in enumerate(tests):\n",
    "        info = _test_info(fn)\n",
    "        outcome: dict = {}\n",
    "\n",
    "        def target(fn=fn, outcome=outcome):\n",
    "            try:\n",
    "                fn()\n",
    "            except BaseException as e:\n",
    "                outcome[\"error\"] = e\n",
    "\n",
    "        if on_start:\n",
    "            on_start(index)\n",
    "        if not _call_with_timeout(target, info[\"timeout\"]):\n",
    "            results.append({\"title\": info[\"title\"], \"status\": \"timeout\", \"message\": info[\"timeout_message\"], \"error_type\": None, \"line\": None})\n",
    "            for other in tests[index + 1:]:\n",
    "                results.append({\"title\": _test_info(other)[\"title\"], \"status\": \"not_run\", \"message\": \"не запускался: предыдущий тест завис\", \"error_type\": None, \"line\": None})\n",
    "            hung = True\n",
    "            break\n",
    "        error = outcome.get(\"error\")\n",
    "        if error is None:\n",
    "            result = {\"title\": info[\"title\"], \"status\": \"passed\", \"message\": \"\", \"error_type\": None, \"line\": None}\n",
    "        else:\n",
    "            result = _error_result(info[\"title\"], error, user_file)\n",
    "            not_written = not_written or result[\"status\"] == \"not_written\"\n",
    "        results.append(result)\n",
    "        if on_result:\n",
    "            on_result(index, result)\n",
    "\n",
    "    if not_written:  # пока функция не написана, «даром» пройденные тесты не засчитываются\n",
    "        for r in results:\n",
    "            if r[\"status\"] == \"passed\":\n",
    "                r.update(status=\"not_written\", message=\"не засчитан, пока функция не написана\")\n",
    "    for r in results:\n",
    "        if len(r[\"message\"]) > 300:\n",
    "            r[\"message\"] = r[\"message\"][:300] + \"…\"\n",
    "\n",
    "    if verbose:\n",
    "        if not_written:\n",
    "            print(\"Функция ещё не написана — замените raise NotImplementedError своим кодом.\\n\")\n",
    "        for r in results:\n",
    "            mark = {\"passed\": \"✓\", \"not_run\": \"–\"}.get(r[\"status\"], \"✗\")\n",
    "            print(f\"{mark} {r['title']}\")\n",
    "            if r[\"message\"]:\n",
    "                print(f\"    {r['message']}\")\n",
    "        passed = sum(r[\"status\"] == \"passed\" for r in results)\n",
    "        print(\"─\" * 40)\n",
    "        print(f\"Прошло {passed} из {len(results)}\" + (\" — всё верно!\" if passed == len(results) else \"\"))\n",
    "        if hung:\n",
    "            print(\"Выполнение остановлено: один из тестов завис.\")\n",
    "            sys.stdout.flush()\n",
    "            os._exit(1)\n",
    "    return results\n",
    "\n",
    "\n",
    "def _check(*tests, names=()):\n",
    "    \"\"\"Запускает проверки упражнения и печатает ✓ / ✗ и итог — как на сайте.\"\"\"\n",
    "    for name in names:\n",
    "        if \".\" in name:  # «Класс.метод» из заготовки: класс должен быть объявлен\n",
    "            if not isinstance(globals().get(name.split(\".\")[0]), type):\n",
    "                print(f\"✗ класс {name.split('.')[0]} не объявлен — выполните ячейку упражнения\")\n",
    "                return\n",
    "            continue\n",
    "        if name not in globals():\n",
    "            print(f\"✗ переменная {name} не создана — выполните ячейку упражнения\")\n",
    "            return\n",
    "        if globals()[name] is Ellipsis:\n",
    "            print(f\"✗ {name} пока равна ... — замените многоточие своим кодом\")\n",
    "            return\n",
    "    globals()[\"_TESTS\"] = list(tests)\n",
    "    _run_tests()"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "setup"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Наследование — не единственный способ построить класс из готовых частей. Корзина из модуля 2 не была\n",
    "подклассом списка или товара — она **содержала** список товаров. Это второй способ, **композиция**\n",
    "(composition): объект хранит другие объекты в атрибутах и пользуется ими. Как выбрать между ними?\n",
    "\n",
    "## «Является» или «содержит»\n",
    "\n",
    "Простой тест — сказать фразу вслух:\n",
    "\n",
    "- латте **является** напитком, сезонный товар **является** товаром → наследование: `class Latte(Drink)`;\n",
    "- заказ **содержит** корзину, кофемашина **содержит** бойлер → композиция: `self.cart = cart`.\n",
    "\n",
    "Если фраза «X является Y» звучит странно (заказ — это корзина? касса — это список?), наследование не подходит."
   ],
   "id": "text-9"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "**Вопрос.** Как связать классы `Order` (заказ) и `Customer` (покупатель)?\n",
    "\n",
    "- `class Order(Customer)` — заказ наследует от покупателя\n",
    "- у заказа атрибут `customer` с экземпляром `Customer`\n",
    "- `class Customer(Order)` — покупатель наследует от заказа\n",
    "\n",
    "<details><summary>Ответ</summary>\n",
    "\n",
    "у заказа атрибут `customer` с экземпляром `Customer`\n",
    "\n",
    "Заказ не является покупателем, а покупатель — заказом. Заказ *содержит* ссылку на покупателя — это композиция.\n",
    "\n",
    "</details>"
   ],
   "id": "quiz-23"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Делегирование\n",
    "\n",
    "Объект, который содержит другой объект, часто передаёт ему работу. Кофемашина не греет воду сама — она\n",
    "просит об этом свой бойлер. Это называется **делегирование** (delegation). Кофемашина при этом сама решает,\n",
    "какие возможности бойлера показать снаружи:"
   ],
   "id": "text-32"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "# Кофемашина делегирует бойлеру\n",
    "class Boiler:\n",
    "    def __init__(self):\n",
    "        self.temperature = 20\n",
    "\n",
    "    def heat(self, target):\n",
    "        self.temperature = target\n",
    "        return f\"вода нагрета до {target} °C\"\n",
    "\n",
    "\n",
    "class CoffeeMachine:\n",
    "    def __init__(self):\n",
    "        self.boiler = Boiler()  # кофемашина содержит бойлер\n",
    "\n",
    "    def brew(self, drink):\n",
    "        step = self.boiler.heat(93)  # делегирует нагрев бойлеру\n",
    "        return f\"{step}; {drink} готов\"\n",
    "\n",
    "\n",
    "machine = CoffeeMachine()\n",
    "print(machine.brew(\"американо\"))\n",
    "print(machine.boiler.temperature)"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "delegate"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Для упражнений понадобятся знакомые товар и корзина:"
   ],
   "id": "text-40"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "# Product и Cart\n",
    "class Product:\n",
    "    def __init__(self, name, price):\n",
    "        self.name = name\n",
    "        self.price = price\n",
    "\n",
    "    def __repr__(self):\n",
    "        return f\"Product({self.name!r}, {self.price})\"\n",
    "\n",
    "\n",
    "class Cart:\n",
    "    def __init__(self):\n",
    "        self.items = []\n",
    "\n",
    "    def add(self, product):\n",
    "        self.items.append(product)\n",
    "\n",
    "    @property\n",
    "    def subtotal(self):\n",
    "        return sum(p.price for p in self.items)\n",
    "\n",
    "\n",
    "cart = Cart()\n",
    "cart.add(Product(\"Латте\", 220))\n",
    "print(cart.items, cart.subtotal)"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "parts"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Упражнение 1. Заказ с корзиной\n",
    "\n",
    "Объявите класс `Order`: `Order(\"Анна\", cart)` хранит имя покупателя в `customer_name` (для простоты — строкой) и корзину в `cart`. Метод\n",
    "`add(product)` кладёт товар в корзину заказа, а свойство `total` возвращает сумму корзины (`subtotal`).\n",
    "Не копируйте код корзины — делегируйте ей.\n",
    "\n",
    "<details><summary>Подсказка 1</summary>\n",
    "\n",
    "Каждый метод заказа — одна строка: он вызывает то же у корзины `self.cart`.\n",
    "\n",
    "</details>\n",
    "<details><summary>Подсказка 2</summary>\n",
    "\n",
    "`total` — свойство, которое возвращает `self.cart.subtotal`.\n",
    "\n",
    "</details>"
   ],
   "id": "order-task"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "class Order:\n",
    "    def __init__(self, customer_name, cart):\n",
    "        ...\n",
    "\n",
    "    def add(self, product):\n",
    "        ...\n",
    "\n",
    "    @property\n",
    "    def total(self):\n",
    "        ..."
   ],
   "execution_count": null,
   "outputs": [],
   "id": "order"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "# Проверка упражнения 1 — выполните после своего решения\n",
    "def _order():\n",
    "    try:\n",
    "        return Order(\"Анна\", Cart())\n",
    "    except TypeError as e:\n",
    "        assert False, f\"Order(\\\"Анна\\\", Cart()) не создаётся: {e}\"\n",
    "\n",
    "\n",
    "def test_parts():\n",
    "    \"customer_name и cart\"\n",
    "    assert not issubclass(Order, Cart), \"заказ не является корзиной — не наследуйте Order от Cart, храните корзину в атрибуте\"\n",
    "    basket = Cart()\n",
    "    order = Order(\"Анна\", basket)\n",
    "    assert getattr(order, \"customer_name\", None) == \"Анна\", \"имя покупателя хранится в атрибуте customer_name\"\n",
    "    assert getattr(order, \"cart\", None) is basket, \"в атрибуте cart должна быть переданная корзина — тот же объект\"\n",
    "\n",
    "\n",
    "def test_delegates():\n",
    "    \"add и total работают через корзину\"\n",
    "    assert isinstance(Order.__dict__.get(\"total\"), property), \"total должен быть свойством — с @property\"\n",
    "    order = _order()\n",
    "    order.add(Product(\"Латте\", 220))\n",
    "    order.add(Product(\"Чай\", 120))\n",
    "    assert len(order.cart.items) == 2, \"add должен класть товар в корзину заказа: self.cart.add(product)\"\n",
    "    assert order.total == 340, f\"total = {order.total!r}, а латте и чай стоят 340\"\n",
    "\n",
    "_check(test_parts, test_delegates, names=[\"Order.__init__\",\"Order.add\",\"Order.total\"])"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "order-check"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details><summary>Решение</summary>\n",
    "\n",
    "```python\n",
    "class Order:\n",
    "    def __init__(self, customer_name, cart):\n",
    "        self.customer_name = customer_name\n",
    "        self.cart = cart\n",
    "\n",
    "    def add(self, product):\n",
    "        self.cart.add(product)\n",
    "\n",
    "    @property\n",
    "    def total(self):\n",
    "        return self.cart.subtotal\n",
    "```\n",
    "\n",
    "</details>"
   ],
   "id": "order-solution"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Ловушка: `class Menu(list)`\n",
    "\n",
    "Меню — это список товаров, так что хочется написать `class Menu(list)`. Но подкласс получает **все** методы\n",
    "списка — и вместе с ними способы сломать меню:"
   ],
   "id": "text-53"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "# Меню-список принимает что угодно\n",
    "class ListMenu(list):  # ошибка: меню — не список\n",
    "    pass\n",
    "\n",
    "\n",
    "menu = ListMenu()\n",
    "menu.append(Product(\"Латте\", 220))\n",
    "menu.append(\"кирпич\")  # и это тоже можно\n",
    "print(len(menu), type(menu + []).__name__)\n",
    "menu.clear()\n",
    "print(menu)"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "menu-list"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "> **Типичная ошибка: наследование от list ради хранения.**\n",
    ">\n",
    "> `class Menu(list)` делает меню списком, а не объектом со своими правилами: в него можно `append` строку,\n",
    "> `clear()` всё меню, а `menu + other` вернёт обычный `list`, а не `Menu`. Если класс только *хранит* элементы,\n",
    "> храните их в атрибуте (`self._items = []`) и откройте наружу ровно те методы, которые нужны меню."
   ],
   "id": "component-61"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Упражнение 2. Меню через композицию\n",
    "\n",
    "Объявите класс `Menu` без наследования от `list`. Товары хранятся во внутреннем списке `_items`. Метод\n",
    "`add(product)` добавляет товар, а если это не экземпляр `Product` — выбрасывает `TypeError`. Метод\n",
    "`names()` возвращает список названий, метод `find(name)` — товар с таким названием или `None`.\n",
    "\n",
    "<details><summary>Подсказка 1</summary>\n",
    "\n",
    "Проверка в `add`: `if not isinstance(product, Product): raise TypeError(\"в меню можно добавить только товар\")`.\n",
    "\n",
    "</details>"
   ],
   "id": "menu-task"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "class Menu:\n",
    "    def __init__(self):\n",
    "        ...\n",
    "\n",
    "    def add(self, product):\n",
    "        ...\n",
    "\n",
    "    def names(self):\n",
    "        ...\n",
    "\n",
    "    def find(self, name):\n",
    "        ..."
   ],
   "execution_count": null,
   "outputs": [],
   "id": "menu"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "# Проверка упражнения 2 — выполните после своего решения\n",
    "def test_not_list():\n",
    "    \"Menu — не подкласс list\"\n",
    "    assert not issubclass(Menu, list), \"Menu наследует от list — храните товары во внутреннем списке _items\"\n",
    "    assert not hasattr(Menu(), \"append\"), \"у меню есть append — оно не должно открывать методы списка\"\n",
    "\n",
    "\n",
    "def test_add_names():\n",
    "    \"add и names\"\n",
    "    menu = Menu()\n",
    "    assert hasattr(menu, \"_items\"), \"товары хранятся во внутреннем списке _items — создайте его в __init__\"\n",
    "    menu.add(Product(\"Латте\", 220))\n",
    "    menu.add(Product(\"Чай\", 120))\n",
    "    assert menu.names() == [\"Латте\", \"Чай\"], f\"names() = {menu.names()!r}\"\n",
    "\n",
    "\n",
    "def test_rejects():\n",
    "    \"не товар — TypeError\"\n",
    "    menu = Menu()\n",
    "    try:\n",
    "        menu.add(\"кирпич\")\n",
    "    except TypeError:\n",
    "        pass\n",
    "    except Exception as e:\n",
    "        assert False, f\"menu.add(\\\"кирпич\\\") выбросил {type(e).__name__}, а нужен именно TypeError — значение неверного типа\"\n",
    "    else:\n",
    "        assert False, \"menu.add(\\\"кирпич\\\") не выбросил TypeError — проверьте isinstance(product, Product)\"\n",
    "    assert menu.names() == [], \"после отказа меню должно остаться пустым\"\n",
    "\n",
    "\n",
    "def test_find():\n",
    "    \"find — товар или None\"\n",
    "    menu = Menu()\n",
    "    tea = Product(\"Чай\", 120)\n",
    "    menu.add(tea)\n",
    "    assert menu.find(\"Чай\") is tea, \"find(\\\"Чай\\\") должен вернуть сам товар\"\n",
    "    assert menu.find(\"Какао\") is None, \"для товара, которого нет, find возвращает None\"\n",
    "\n",
    "_check(test_not_list, test_add_names, test_rejects, test_find, names=[\"Menu.__init__\",\"Menu.add\",\"Menu.names\",\"Menu.find\"])"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "menu-check"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details><summary>Решение</summary>\n",
    "\n",
    "```python\n",
    "class Menu:\n",
    "    def __init__(self):\n",
    "        self._items = []\n",
    "\n",
    "    def add(self, product):\n",
    "        if not isinstance(product, Product):\n",
    "            raise TypeError(\"в меню можно добавить только товар\")\n",
    "        self._items.append(product)\n",
    "\n",
    "    def names(self):\n",
    "        return [p.name for p in self._items]\n",
    "\n",
    "    def find(self, name):\n",
    "        for p in self._items:\n",
    "            if p.name == name:\n",
    "                return p\n",
    "        return None\n",
    "```\n",
    "\n",
    "</details>"
   ],
   "id": "menu-solution"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Части можно менять на ходу\n",
    "\n",
    "У композиции есть ещё одно преимущество: часть объекта можно заменить другой, пока программа работает.\n",
    "Класс через наследование выбирается раз и навсегда, а атрибут можно переприсвоить. Покупатель передумал\n",
    "и вместо курьера выбрал самовывоз — меняется только атрибут заказа. Способы доставки — классы из модуля 5:"
   ],
   "id": "text-74"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "# Самовывоз и курьер\n",
    "class Pickup:\n",
    "    def cost(self, km):\n",
    "        return 0\n",
    "\n",
    "\n",
    "class Courier:\n",
    "    def cost(self, km):\n",
    "        return 150 + 30 * km\n",
    "\n",
    "\n",
    "print(Pickup().cost(5), Courier().cost(5))"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "deliveries"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Упражнение 3. Заказ с доставкой\n",
    "\n",
    "Объявите класс `DeliveryOrder`: `DeliveryOrder(cart, delivery)` хранит корзину в `cart` и способ доставки\n",
    "в `delivery` — любой объект с методом `cost(km)` (например, `Courier()` или `Pickup()` из демонстрации).\n",
    "Метод `total(km)` возвращает сумму корзины плюс стоимость доставки. Затем создайте `order` — заказ\n",
    "с корзиной `cart` из заготовки и курьером, сохраните в `by_courier` его `total(5)`, поменяйте у этого же заказа\n",
    "доставку на самовывоз и сохраните `total(5)` в `by_pickup`.\n",
    "\n",
    "<details><summary>Подсказка 1</summary>\n",
    "\n",
    "`total` складывает две вещи, которые знают вложенные объекты: сумму корзины и стоимость доставки.\n",
    "\n",
    "</details>\n",
    "<details><summary>Подсказка 2</summary>\n",
    "\n",
    "Поменять доставку — присвоить атрибуту: `order.delivery = Pickup()`.\n",
    "\n",
    "</details>"
   ],
   "id": "delivery-order-task"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "class DeliveryOrder:\n",
    "    def __init__(self, cart, delivery):\n",
    "        ...\n",
    "\n",
    "    def total(self, km):\n",
    "        ...\n",
    "\n",
    "\n",
    "cart = Cart()\n",
    "cart.add(Product(\"Латте\", 220))\n",
    "order = ...\n",
    "by_courier = ...\n",
    "by_pickup = ..."
   ],
   "execution_count": null,
   "outputs": [],
   "id": "delivery-order"
  },
  {
   "cell_type": "code",
   "metadata": {},
   "source": [
    "# Проверка упражнения 3 — выполните после своего решения\n",
    "class _Drone:\n",
    "    \"доставка, которой заказ заранее не знает\"\n",
    "\n",
    "    def cost(self, km):\n",
    "        return 99\n",
    "\n",
    "\n",
    "def test_class():\n",
    "    \"total — корзина плюс доставка\"\n",
    "    basket = Cart()\n",
    "    basket.add(Product(\"Раф\", 260))\n",
    "    try:\n",
    "        got = DeliveryOrder(basket, _Drone()).total(3)\n",
    "    except (TypeError, AttributeError) as e:\n",
    "        assert False, f\"DeliveryOrder(корзина, доставка).total(3) падает: {e}\"\n",
    "    assert got == 359, f\"раф 260 ₽ + доставка 99 ₽ = {got!r}, а нужно 359 — берите стоимость у self.delivery.cost(km)\"\n",
    "\n",
    "\n",
    "def test_values():\n",
    "    \"by_courier и by_pickup\"\n",
    "    assert isinstance(order, DeliveryOrder), f\"order — это {type(order).__name__}, а нужен DeliveryOrder\"\n",
    "    assert by_courier == 520, f\"by_courier = {by_courier!r}, а корзина 220 ₽ + курьер на 5 км 300 ₽ = 520\"\n",
    "    assert by_pickup == 220, f\"by_pickup = {by_pickup!r}, а с самовывозом остаётся сумма корзины — 220\"\n",
    "    assert isinstance(order.delivery, Pickup), \"в конце у заказа должна быть доставка Pickup — присвойте order.delivery\"\n",
    "\n",
    "_check(test_class, test_values, names=[\"DeliveryOrder.__init__\",\"DeliveryOrder.total\",\"order\",\"by_courier\",\"by_pickup\"])"
   ],
   "execution_count": null,
   "outputs": [],
   "id": "delivery-order-check"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details><summary>Решение</summary>\n",
    "\n",
    "```python\n",
    "class DeliveryOrder:\n",
    "    def __init__(self, cart, delivery):\n",
    "        self.cart = cart\n",
    "        self.delivery = delivery\n",
    "\n",
    "    def total(self, km):\n",
    "        return self.cart.subtotal + self.delivery.cost(km)\n",
    "\n",
    "\n",
    "cart = Cart()\n",
    "cart.add(Product(\"Латте\", 220))\n",
    "order = DeliveryOrder(cart, Courier())\n",
    "by_courier = order.total(5)\n",
    "order.delivery = Pickup()\n",
    "by_pickup = order.total(5)\n",
    "```\n",
    "\n",
    "</details>"
   ],
   "id": "delivery-order-solution"
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Правило, которое часто повторяют: **предпочитайте композицию наследованию**. Наследование — для настоящего\n",
    "«является», когда подкласс годится везде, где ждут базовый класс. Во всех остальных случаях объект лучше\n",
    "собрать из частей."
   ],
   "id": "text-93"
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "name": "python"
  },
  "colab": {
   "provenance": []
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}
