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())```
相关推荐
程序媛徐师姐7 分钟前
Python基于Django的主观题自动阅卷系统【附源码、文档说明】
python·django·python主观题自动阅卷系统·主观题自动阅卷系统·python主观题评分系统·主观题评分系统·主观题评分
Pluchon13 分钟前
硅基计划2.0 学习总结 伍 类的继承 初稿
java·开发语言·学习·算法·markdown
珊瑚里的鱼14 分钟前
第十讲 | 继承
开发语言·c++·笔记·visualstudio·学习方法·visual studio
闭月之泪舞20 分钟前
OpenCv高阶(十九)——dlib关键点定位
人工智能·opencv·计算机视觉
AI technophile23 分钟前
OpenCV计算机视觉实战(9)——阈值化技术详解
人工智能·opencv·计算机视觉
HealthScience25 分钟前
pycharm debug的时候无法debug到指定的位置就停住不动了
人工智能·pytorch·深度学习
jndingxin25 分钟前
OpenCV CUDA模块直方图计算------在 GPU 上计算输入图像的直方图(histogram)函数histEven()
人工智能·opencv·计算机视觉
qyresearch_31 分钟前
商用飞机电池市场:现状、挑战与未来展望
大数据·人工智能
正在走向自律31 分钟前
AI新手入门解锁元生代MaaS平台:API工作流调用全攻略
人工智能·自然语言处理·知识库·rag·maas·openai 兼容接口
EasyDSS42 分钟前
视频监控汇聚平台EasyCVR工业与安全监控:防爆摄像机的安全应用与注意事项
网络·人工智能·音视频