策略模式是一种行为设计模式,它通过定义一系列算法,将每个算法封装起来,并使它们可以相互替换。策略模式让算法的变化独立于使用算法的客户。
下面我们将创建一个示例,其中有多种支付方式(例如信用卡支付和支付宝支付)。我们将使用策略模式来实现这几种支付方式。
1.定义策略接口
from abc import ABC, abstractmethod
class PaymentStrategy(ABC):
@abstractmethod
def pay(self, amount: float):
pass
2.实现具体策略
class CreditCardPayment(PaymentStrategy):
def __init__(self, card_number: str):
self.card_number = card_number
def pay(self, amount: float):
print(f"支付 {amount} 元,通过信用卡 {self.card_number} 完成支付。")
class AlipayPayment(PaymentStrategy):
def __init__(self, account_id: str):
self.account_id = account_id
def pay(self, amount: float):
print(f"支付 {amount} 元,通过支付宝账户 {self.account_id} 完成支付。")
3.创建上下文,使用具体的策略进行支付。
class PaymentContext:
def __init__(self, payment_strategy: PaymentStrategy):
self.payment_strategy = payment_strategy
def execute_payment(self, amount: float):
self.payment_strategy.pay(amount)
-
主程序中调用
def main():
# 使用信用卡支付
credit_card_payment = CreditCardPayment(card_number="1234-5678-9012-3456")
payment_context = PaymentContext(payment_strategy=credit_card_payment)
payment_context.execute_payment(100.0)print() # 使用支付宝支付 alipay_payment = AlipayPayment(account_id="[email protected]") payment_context = PaymentContext(payment_strategy=alipay_payment) payment_context.execute_payment(200.0)
if name == "main":
main()