Passo 29

Instruções

É hora de testar o sistema completo de desconto! Adicione uma declaração if verificando se __name__ == '__main__' e mova product = Product('Wireless Mouse', 50.0) para dentro dela. Depois disso, dentro do bloco if, crie uma variável user_tier e defina-a como a string Premium. Em seguida, crie uma variável strategies definida como uma lista de estratégias contendo PercentageDiscount(10), FixedAmountDiscount(5) e PremiumUserDiscount().

O que fazer:

Testes:

Você deve ter uma declaração `if` verificando se `__name__ == '__main__'`.
Você deve mover `product = Product('Wireless Mouse', 50.0)` para dentro do bloco `if`.
Você deve definir `user_tier` para a string `Premium` dentro do bloco `if`.
Você deve criar uma lista chamada `strategies` contendo `PercentageDiscount(10)`, `FixedAmountDiscount(5)` e `PremiumUserDiscount()`.

from abc import ABC, abstractmethod

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

    def __str__(self) -> str:
        return f'{self.name} - ${self.price}'

class DiscountStrategy(ABC):
    @abstractmethod
    def is_applicable(self, product: Product, user_tier: str) -> bool:
        pass

    @abstractmethod
    def apply_discount(self, product: Product) -> float:
        pass

class PercentageDiscount(DiscountStrategy):
    def __init__(self, percent: int) -> None:
        self.percent = percent

    def is_applicable(self, product: Product, user_tier: str) -> bool:
        return self.percent <= 70

    def apply_discount(self, product: Product) -> float:
        return product.price * (1 - self.percent / 100)

class FixedAmountDiscount(DiscountStrategy):
    def __init__(self, amount: int) -> None:
        self.amount = amount

    def is_applicable(self, product: Product, user_tier: str) -> bool:
        return product.price * 0.9 > self.amount

    def apply_discount(self, product: Product) -> float:
        return product.price - self.amount

class PremiumUserDiscount(DiscountStrategy):
    def is_applicable(self, product: Product, user_tier: str) -> bool:
        return user_tier.lower() == 'premium'

    def apply_discount(self, product: Product) -> float:
        return product.price * 0.8

class DiscountEngine:
    def __init__(self, strategies: list[DiscountStrategy]) -> None:
        self.strategies = strategies

    def calculate_best_price(self, product: Product, user_tier: str) -> float:
        prices = [product.price]

        for strategy in self.strategies:
            if strategy.is_applicable(product, user_tier):
                discounted = strategy.apply_discount(product)
                prices.append(discounted)

        return min(prices)

product = Product('Wireless Mouse', 50.0)

Preview