策略模式python

策略模式

介绍

策略模式是定义一个算法家族，以算法为粒度，将不同算法分别封装起来，让它们之间可以互相替换。此模式让算法的变化不会影响到使用算法的客户。和简单工厂类类似，存在有三个角色：

• 策略抽象类
• 策略具体类
• Context上下文类

策略抽象类和策略具体类和简单工厂中的产品类是相同的，抽象类是所有策略类的公共接口，而具体类是不同算法对接口独有的实现,其中重要的是Context上下文。名为上下文的原始类必须包含一个成员变量来存储对于每种策略的引用。 上下文并不执行任务， 而是将工作委派给已连接的策略对象。实际上， 上下文并不十分了解策略， 它会通过同样的通用接口与所有策略进行交互， 而该接口只需暴露一个方法来触发所选策略中封装的算法即可。

结构图

模式解析

优点

• 策略模式的Strategy类层次为Context定义了一系列的可供重用的算法或行为。继承有助于析取出这些算法中的公共功能
• 简化了单元测试，每个算法都有自己的类，可以通过自己的接口单独测试
• 消除条件语句并且扩展性良好

缺点

• 所有策略类都需要对外暴露
• 策略类会不断增多

使用场景

• 如果在一个系统里面有许多类，它们之间的区别仅在于它们的行为，那么使用策略模式可以动态地让一个对象在许多行为中选择一种行为。
• 一个系统需要动态地在几种算法中选择一种。
• 如果一个对象有很多的行为，如果不用恰当的模式，这些行为就只好使用多重的条件选择语句来实现。

注意事项

• 如果一个系统的策略多于四个，就需要考虑使用混合模式，解决策略类膨胀的问题

代码用例

# =============================
# 内容: 策略模式
# 形式：商场促销
# 促销类型：打折和返现
# =============================

from abc import abstractclassmethod

from numpy import double
from pymysql import NULL

# 策略类型
NONE_STRATEGY   = 0x00
NORMAL_STRATEGY = 0x01
REBATE_STRATEGY = 0x02
RETURN_STRATEGY = 0x03


# 策略的公共接口类
class ICashStrategy:
    def __init__(self):
        self.type = NONE_STRATEGY  # 默认没有策略
        self.__real_money = 0  # 私有变量，不被公用

    @abstractclassmethod
    def accept_cash(self, money):
        pass


# 正常收费子类
class CashNormal(ICashStrategy):
    def __init__(self):
        """super()使用
            super()方法实际和获得父类是不同的,self是当前实例类,python会根据这个实例类
            获取一个MRO列表(方法解析顺序表),它代表了类继承的顺序,这个列表可以通过C.mro()
            获得,super()就会根据这个列表便利所有超类的方法
        """
        super(CashNormal, self).__init__()  # 调用父类的初始化函数
        self.type = NORMAL_STRATEGY
        self.__real_money = 0

    def accept_cash(self, money):
        self.__real_money = money
        return self.__real_money


# 打折收费子类
class CashRebate(ICashStrategy):
    def __init__(self, rate):
        super(CashRebate, self).__init__()
        self.rate = rate
        self.type = REBATE_STRATEGY
        self.__real_money = 0

    def accept_cash(self, money):
        self.__real_money = (self.rate * money)
        return self.__real_money


# 返利收费子类
class CashReturn(ICashStrategy):
    def __init__(self, money_condition, money_return):
        super(CashReturn, self).__init__()
        self.money_condition = money_condition
        self.money_return = money_return
        self.type = RETURN_STRATEGY
        self.__real_money = 0

    def accept_cash(self, money):
        if (money < self.money_condition):
            self.__real_money = money
        elif (money < 0 or money > 1000000):
            print("价格输入错误")
        else:
            return_money = int(
                money / self.money_condition) * self.money_return
            self.__real_money = money - return_money
        return self.__real_money


# 策略模式+简单工厂类
class CashContext:
    def __init__(self):
        self.strategy = NULL
        self.strategy_map = {
            "正常收费": "CashNormal",
            "返现折扣": "CashReturn",
            "打折折扣": "CashRebate",
        }
        self.params = {}

    # 装载策略
    def load_strategy(self, type, **kwargs):
        self.params = kwargs
        cmd = "self.strategy = " + self.strategy_map[type] + "(**self.params)"
        exec(cmd)
        
	# 调用策略的公共接口
    def get_result(self, money):
        return self.strategy.accept_cash(money)


# 客户端代码（主函数）
if __name__ == '__main__':

    money = 0

    # 录入商品信息
    while True:
        print("-" * 50)
        print("商品结算结束请输入1, 未结束输入0")
        end_flag = int(input())

        # 判断商品结算是否已经完成
        if (end_flag):
            print("-" * 50)
            break

        # 输入商品价格和数目(这里可以在确认的地方优化，但不是这段demo的重点)
        item_price = double(input("请输入商品价格"))
        item_num = int(input("请输入商品数量"))
        print(f"\n请确认商品价格和数量:\n\t商品价格:{item_price}\n\t商品数目:{item_num}")
        money += item_price * item_num
        print("-" * 50)
    print("商品的价格为", money)

    # 加载折扣,这里也可以优化成输入添加
    cash_context = CashContext()
    cash_context.load_strategy("正常收费")
    real_money = cash_context.get_result(money)
    print("正常收费收取金额：", real_money)

    cash_context.load_strategy("返现折扣", money_condition=50, money_return=10)
    real_money = cash_context.get_result(money)
    print("返现折扣收取金额：", real_money)

    cash_context.load_strategy("打折折扣", rate=0.8)
    real_money = cash_context.get_result(money)
    print("打折折扣收取金额：", real_money)