python策略模式

码界孔乙己2025-03-29 13:57

策略模式是一种行为设计模式，它通过定义一系列算法，将每个算法封装起来，并使它们可以相互替换。策略模式让算法的变化独立于使用算法的客户。

下面我们将创建一个示例，其中有多种支付方式（例如信用卡支付和支付宝支付）。我们将使用策略模式来实现这几种支付方式。

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()