python策略模式

发布于:2025-03-29 ⋅ 阅读:(21) ⋅ 点赞:(0)

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

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

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)

4. 主程序中调用


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="user@example.com")
    payment_context = PaymentContext(payment_strategy=alipay_payment)
    payment_context.execute_payment(200.0)

if __name__ == "__main__":
    main()