Python10天冲刺-设计模型之策略模式

策略模式是一种行为设计模式,它允许你在运行时动态地改变对象的行为。这种模式的核心思想是将一组相关的算法封装在一起,并让它们相互替换。

下面是使用 Python 实现策略模式的一个示例:

示例代码

假设我们有一个简单的购物车系统,其中不同的支付方式有不同的折扣策略。我们将使用策略模式来实现这一功能。

1. 定义抽象基类(策略接口)
python 复制代码
from abc import ABC, abstractmethod

class PaymentStrategy(ABC):
    @abstractmethod
    def pay(self, amount):
        pass
2. 具体策略实现
python 复制代码
class CreditCardPayment(PaymentStrategy):
    def pay(self, amount):
        print(f"Paying ${amount} with credit card.")
        return f"${amount} paid using credit card."

class PayPalPayment(PaymentStrategy):
    def pay(self, amount):
        print(f"Paying ${amount} with PayPal.")
        return f"${amount} paid using PayPal."
3. 上下文类

上下文类负责使用具体的策略来进行支付。

python 复制代码
class ShoppingCart:
    def __init__(self, payment_strategy: PaymentStrategy):
        self._items = []
        self.payment_strategy = payment_strategy

    def add_item(self, item):
        self._items.append(item)

    def calculate_total(self):
        return sum(item['price'] for item in self._items)

    def checkout(self):
        total_amount = self.calculate_total()
        return self.payment_strategy.pay(total_amount)
4. 测试策略模式
python 复制代码
if __name__ == "__main__":
    cart = ShoppingCart(CreditCardPayment())
    cart.add_item({'name': 'Shirt', 'price': 20})
    cart.add_item({'name': 'Pants', 'price': 30})

    print(cart.checkout())

    cart = ShoppingCart(PayPalPayment())
    cart.add_item({'name': 'Shoes', 'price': 50})
    cart.add_item({'name': 'Hat', 'price': 15})

    print(cart.checkout())

解释

  1. 抽象基类(策略接口)

    • PaymentStrategy 类定义了一个抽象方法 pay,这是所有具体策略必须实现的方法。
  2. 具体策略实现

    • CreditCardPaymentPayPalPayment 分别实现了 pay 方法,提供了不同的支付逻辑。
  3. 上下文类

    • ShoppingCart 类持有一个 payment_strategy 属性,用于存放具体的支付策略。
    • checkout 方法计算总金额并调用当前策略的 pay 方法。
  4. 测试策略模式

    • 创建两个 ShoppingCart 对象,分别使用 CreditCardPaymentPayPalPayment 策略。
    • 添加商品并结算,观察不同支付策略的结果。

通过这种方式,你可以灵活地切换不同的支付策略,而无需修改上下文类的代码。这就是策略模式的主要优点之一。


完整可运行的代码库

python 复制代码
from abc import ABC, abstractmethod

# 抽象基类(策略接口)
class PaymentStrategy(ABC):
    @abstractmethod
    def pay(self, amount):
        pass

# 具体策略实现
class CreditCardPayment(PaymentStrategy):
    def pay(self, amount):
        print(f"Paying {amount} using credit card.")
        return f"{amount} paid using credit card."

class PayPalPayment(PaymentStrategy):
    def pay(self, amount):
        print(f"Paying {amount} using PayPal.")
        return f"{amount} paid using PayPal."

# 上下文类
class ShoppingCart:
    def __init__(self, payment_strategy: PaymentStrategy):
        self.items = []
        self.payment_strategy = payment_strategy

    def add_item(self, item):
        self.items.append(item)

    def remove_item(self, item):
        self.items.remove(item)

    def calculate_total(self):
        return sum(item['price'] for item in self.items)

    def checkout(self):
        total_amount = self.calculate_total()
        return self.payment_strategy.pay(total_amount)

if __name__ == "__main__":
    cart = ShoppingCart(CreditCardPayment())
    cart.add_item({'name': 'Shirt', 'price': 20})
    cart.add_item({'name': 'Pants', 'price': 30})

    print(cart.checkout())

    cart = ShoppingCart(PayPalPayment())
    cart.add_item({'name': 'Shoes', 'price': 50})
    cart.add_item({'name': 'Hat', 'price': 15})

    print(cart.checkout())```
相关推荐
凪卄12136 分钟前
图像预处理 二
人工智能·python·深度学习·计算机视觉·pycharm
巫婆理发2229 分钟前
强化学习(第三课第三周)
python·机器学习·深度神经网络
AI赋能24 分钟前
自动驾驶训练-tub详解
人工智能·深度学习·自动驾驶
seasonsyy25 分钟前
1.安装anaconda详细步骤(含安装截图)
python·深度学习·环境配置
deephub33 分钟前
AI代理性能提升实战:LangChain+LangGraph内存管理与上下文优化完整指南
人工智能·深度学习·神经网络·langchain·大语言模型·rag
EulerBlind34 分钟前
【运维】SGLang 安装指南
运维·人工智能·语言模型
心之语歌36 分钟前
Spring AI MCP 客户端
人工智能·spring·github
Edingbrugh.南空41 分钟前
Aerospike与Redis深度对比:从架构到性能的全方位解析
java·开发语言·spring
半新半旧43 分钟前
python 整合使用 Redis
redis·python·bootstrap
go54631584651 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法