Passo 23

Instruções

Agora você precisa de uma maneira de gerenciar múltiplas estratégias de desconto e buscar o melhor preço. Crie uma classe chamada DiscountEngine. Esta classe gerenciará todas as estratégias de desconto e calculará o melhor preço para um produto. Dê a ele um método __init__ que receba self e strategies como parâmetros, onde strategies tem uma dica de tipo de list[DiscountStrategy]. Usar list[DiscountStrategy] especifica que o parâmetro é uma lista contendo objetos DiscountStrategy. O tipo de retorno do método deve ser None. Dentro do método __init__, armazene strategies como um atributo de instância.

O que fazer:

Testes:

Você deve criar uma classe chamada `DiscountEngine`.
Sua classe `DiscountEngine` deve ter um método `__init__`.
Seu método `__init__` deve ter os parâmetros `self` e `strategies: list[DiscountStrategy]`.
Seu método `__init__` deve ter um tipo de retorno `None`.
Você deve atribuir `strategies` a `self.strategies` dentro do seu método `__init__`.

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



product = Product('Wireless Mouse', 50.0)
print(product)

discount = PercentageDiscount(10)
print(discount.apply_discount(product))

fixed_discount = FixedAmountDiscount(5)
print(fixed_discount.apply_discount(product))

Preview