# strategy_best3.py # Strategy pattern -- function-based implementation # selecting best promotion from imported module """ >>> from decimal import Decimal >>> from strategy import Customer, LineItem, Order >>> from promotions import * >>> joe = Customer('John Doe', 0) >>> ann = Customer('Ann Smith', 1100) >>> cart = [LineItem('banana', 4, Decimal('.5')), ... LineItem('apple', 10, Decimal('1.5')), ... LineItem('watermelon', 5, Decimal(5))] >>> Order(joe, cart, fidelity_promo) >>> Order(ann, cart, fidelity_promo) >>> banana_cart = [LineItem('banana', 30, Decimal('.5')), ... LineItem('apple', 10, Decimal('1.5'))] >>> Order(joe, banana_cart, bulk_item_promo) >>> long_cart = [LineItem(str(item_code), 1, Decimal(1)) ... for item_code in range(10)] >>> Order(joe, long_cart, large_order_promo) >>> Order(joe, cart, large_order_promo) # tag::STRATEGY_BEST_TESTS[] >>> Order(joe, long_cart, best_promo) >>> Order(joe, banana_cart, best_promo) >>> Order(ann, cart, best_promo) # end::STRATEGY_BEST_TESTS[] """ # tag::STRATEGY_BEST3[] from decimal import Decimal import inspect from strategy import Order import promotions promos = [func for _, func in inspect.getmembers(promotions, inspect.isfunction)] def best_promo(order: Order) -> Decimal: """Compute the best discount available""" return max(promo(order) for promo in promos) # end::STRATEGY_BEST3[]