coffee.py

import abc

# TODO : Add Dark roast, Soy, Whip


class Beverage(abc.ABC):
    """Base class for all beverages"""

    def __init__(self):
        self.description = "Unknown Beverage"

    def get_description(self):
        return self.description

    @abc.abstractmethod
    def cost(self):
        raise NotImplementedError


class CondimentDecorator(Beverage, abc.ABC):
    """Base class for all condiments"""

    @abc.abstractmethod
    def get_description(self):
        raise NotImplementedError


class Espresso(Beverage):
    def __init__(self):
        self.description = "Espresso"

    def cost(self):
        return 1.99


class HouseBlend(Beverage):
    def __init__(self):
        self.description = "House Blend Coffee"

    def cost(self):
        return 0.89


class Mocha(CondimentDecorator):
    def __init__(self, beverage: Beverage):
        self.beverage = beverage

    def get_description(self):
        return self.beverage.get_description() + ", Mocha"

    def cost(self):
        return self.beverage.cost() + 0.20


def star_buzz_coffee():
    """Test code"""
    beverage = Espresso()
    print(beverage.get_description(), beverage.cost())

    beverage3 = HouseBlend()
    beverage3 = Mocha(beverage3)
    beverage3 = Mocha(beverage3)
    print(beverage3.get_description(), beverage3.cost())


if __name__ == "__main__":
    star_buzz_coffee()