"""
4. Склад товаров

Класс Product.

Поля:

- название
- цена
- количество

Создай 5 объектов.

Задача:

- посчитать общую стоимость склада

"""


class Product:
    def __init__(self, name: str, price: float, count: int) -> None:
        self.name: str = name
        self.price: float = price
        self.count: int = count


class Warehouse:
    def __init__(self) -> None:
        self.storage: list[Product] = []

    def add(self, product: Product) -> None:
        if isinstance(product, Product):
            self.storage.append(product)
        else:
            raise ValueError("Не корректный тип продукта")

    def total_cost(self) -> float:
        total: float = 0.0
        for product in self.storage:
            total += product.price * product.count
        return total


water: Product = Product(name="вода, святой источник", price=76.0, count=30)
chocolate: Product = Product(name="киткат", price=60.0, count=10)

warehouse = Warehouse()
warehouse.add(product=water)
warehouse.add(product=chocolate)
print(warehouse.total_cost())