Python设计模式中的行为型策略模式_python

import abc

# 抽象策略
class Strategy(metaclass=abc.ABCMeta):
    @abc.abstractmethod
    def execute(self, data):
        pass

# 具体策略 1
class Strategy1(Strategy):
    def execute(self, data):
        print(f"使用策略1执行{data}")

# 具体策略2 
class Strategy2(Strategy):
    def execute(self, data):
        print(f"使用策略2执行{data}")

# 上下文
class Context:
    def __init__(self, data):
        self.strategy = None
        self.data = data

    def set_strategy(self, strategy):
        self.strategy = strategy

    def do_strategy(self):
        if self.strategy:
            self.strategy.execute(self.data)
        else:
            print("请先配置策略！")


if __name__ == "__main__":
    data = "测试数据"
    s1 = Strategy1()
    s2 = Strategy2()
    context = Context(data)
    context.set_strategy(s1)
    context.do_strategy()
    context.set_strategy(s2)
    context.do_strategy()