当前位置：首页 > news >正文

python策略模式

news 来源：原创 2025/6/15 11:24:49

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

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

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