Python设计模式之策略模式

前言

设计模式是我们实际应用开发中必不可缺的,对设计模式的理解有助于我们写出可读性和扩展更高的应用程序。虽然设计模式与语言无关,但并不意味着每一个模式都能在任何语言中使用,所以有必要去针对语言的特性去做了解。设计模式特别是对于java语言而言,已经有过非常多的大牛写过,所以这里我就不重复了。对于Python来说就相对要少很多,特别是python语言具有很多高级的特性,而不需要了解这些照样能满足开发中的很多需求,所以很多人往往忽视了这些,这里我们来在Pythonic中来感受一下设计模式。

1.介绍

策略模式也是常见的设计模式之一,它是指对一系列的算法定义,并将每一个算法封装起来,而且使它们还可以相互替换。策略模式让算法独立于使用它的客户而独立变化。
这是比较官方的说法,看着明显的一股比较抽象的感觉,通俗来讲就是针对一个问题而定义出一个解决的模板,这个模板就是具体的策略,每个策略都是按照这个模板来的。这种情况下我们有新的策略时就可以直接按照模板来写,而不会影响之前已经定义好的策略。

2.具体实例

这里我用的《流畅的Python》中的实例,刚好双11过去不久,相信许多小伙伴也是掏空了腰包,哈哈。那这里就以电商领域的根据客户的属性或订单中的商品数量来计算折扣的方式来进行讲解,首先来看看下面这张图。


电商大促销

通过这张图,相信能对策略模式的流程有个比较清晰的了解了。然后看看具体的实现过程,首先我们用namedtuple来定义一个Customer,虽然这里是说设计模式,考虑到有些小伙伴可能对Python中的具名元组不太熟悉,所以这里也简单的说下。
namedtuple用来构建一个带字段名的元组和一个有名字的类,这样说可能还是有些抽象,这里来看看下面的代码

from collections import namedtuple
City = namedtuple('City','name country provinces')

这里测试就直接如下

changsha = City('Changsha','China','Hunan')
print(changsha)

结果如下

City(name='Changsha', country='China', province='Hunan')

还可以直接调用字段名

print(changsha.name)

更多用法可以去看看官方文档,这里重点还是讲设计模式。
好了,先来看看用类实现的策略模式

# 策略设计模式实例

from abc import ABC, abstractmethod
from collections import namedtuple

# 创建一个具名元组
Customer = namedtuple('Customer', 'name fidelity')


class LineItem:

    def __init__(self, product, quantity, price):
        self.product = product
        self.quantity = quantity
        self.price = price

    def total(self):
        return self.price * self.quantity


# 上下文
class Order:
  
  # 传入三个参数,分别是消费者,购物清单,促销方式
    def __init__(self, customer, cart, promotion=None):
        self.customer = customer
        self.cart = list(cart)
        self.promotion = promotion

    def total(self):
        if not hasattr(self, '__total'):
            self.__total = sum(item.total() for item in self.cart)
        return self.__total

    def due(self):
        if self.promotion is None:
            discount = 0
        else:
            discount = self.promotion.discount(self)
        return self.total() - discount
     
    # 输出具体信息
    def __repr__(self):
        fmt = '<Order total: {:.2f} due: {:.2f}>'
        return fmt.format(self.total(), self.due())


# 策略 抽象基类
class Promotion(ABC):

    @abstractmethod
    def discount(self, order):
        """
        :param order:
        :return: 返回折扣金额(正值)
        """


# 第一个具体策略
class FidelityPromo(Promotion):
    """ 为积分为1000或以上的顾客提供5%的折扣 """

    def discount(self, order):
        return order.total() * .05 if order.customer.fidelity >= 1000 else 0


# 第二个具体策略
class BulkItemPromo(Promotion):
    """ 单个商品为20个或以上时提供10%折扣"""

    def discount(self, order):
        discount = 0
        for item in order.cart:
            if item.quantity >= 20:
                discount = item.total() * .1
        return discount


# 第三个具体策略
class LargeOrderPromo(Promotion):
    """ 订单中的不同商品达到10个或以上时提供%7的折扣"""

    def discount(self, order):
        distinct_items = {item.product for item in order.cart}
        if len(distinct_items) >= 10:
            return order.total() * .07
        return 0

这里是用类对象来实现的策略模式,每个具体策略类(折扣方式)都继承了Promotion这个基类,因为discount()是一个抽象函数,所以继承Promotion的子类都需要重写discount()函数(也就是进行具体的打折信息的函数),这样一来,就很好的实现对象之间的解耦。这里的折扣方式有两类,一类是根据用户的积分,一类是根据用户所购买商品的数量。具体的折扣信息也都在代码块里面注释了,这里就不重复了,接下来我们来看看具体的测试用例

joe = Customer('John Doe', 0)
ann = Customer('Ann Smith', 1100)
cart = [LineItem('banana', 4, .5),
        LineItem('apple', 10, 1.5),
        LineItem('watermellon', 5, 5.0)]
print('John: ', Order(joe, cart, FidelityPromo()))
print('Ann: ', Order(ann, cart, FidelityPromo()))

这里定义了两消费者,John初始积分为0,Ann初始积分为1100,然后商品购买了4个香蕉,10个苹果,5个西瓜...说的都要流口水了,哈哈哈。回到正题,输出时采用第一种折扣方式,Run一下

John:  <Order total: 42.00 due: 42.00>
Ann:  <Order total: 42.00 due: 39.90>

3.优化措施

➀类变函数

上面的策略模式是使用的类对象实现的,其实我们还可以用函数对象的方法实现,看看具体的代码

# 策略设计模式实例

from collections import namedtuple

# 创建一个具名元组
Customer = namedtuple('Customer', 'name fidelity')


class LineItem:

    def __init__(self, product, quantity, price):
        self.product = product
        self.quantity = quantity
        self.price = price

    def total(self):
        return self.price * self.quantity


# 上下文
class Order:

    def __init__(self, customer, cart, promotion=None):
        self.customer = customer
        self.cart = list(cart)
        self.promotion = promotion

    def total(self):
        if not hasattr(self, '__total'):
            self.__total = sum(item.total() for item in self.cart)
        return self.__total

    def due(self):
        if self.promotion is None:
            discount = 0
        else:
            discount = self.promotion.discount(self)
        return self.total() - discount

    def __repr__(self):
        fmt = '<Order total: {:.2f} due: {:.2f}>'
        return fmt.format(self.total(), self.due())


# 第一个具体策略
def fidelity_promo(order):
    """ 为积分为1000或以上的顾客提供5%的折扣 """

    return order.total() * .05 if order.customer.fidelity >= 1000 else 0


# 第二个具体策略
def bulk_item_promo(order):
    """ 单个商品为20个或以上时提供10%折扣"""

    discount = 0
    for item in order.cart:
        if item.quantity >= 20:
            discount = item.total() * .1
    return discount


# 第三个具体策略
def large_order_promo(order):
    """ 订单中的不同商品达到10个或以上时提供%7的折扣"""

    distinct_items = {item.product for item in order.cart}
    if len(distinct_items) >= 10:
        return order.total() * .07
    return 0    

这种方式没有了抽象类,并且每个策略都是函数,实现同样的功能,代码量更加少,并且测试的时候可以直接把促销函数作为参数传入,这里就不多说了。

➁选择最佳策略

细心的朋友可能观察到,我们这样每次对商品进行打折处理时,都需要自己选择折扣方式,这样数量多了就会非常的麻烦,那么有没有办法让系统帮我们自动选择呢?当然是有的,这里我们可以定义一个数组,把折扣策略的函数当作元素传进去。

promos = [fidelity_promo,bulk_item_promo,large_order_promo]

然后定义一个函数

def best_promo(order):
    """  选择可用的最佳折扣 """
    
    return max(promo(order) for promo in promos)

这样一来就省了很多时间,系统帮我们自动选择。但是仍然有一个问题,这个数组的元素需要我们手动输入,虽然工作量小,但是对于有强迫症的猿来说,依然是不行的,能用自动化的方式就不要用手动,所以继续做优化。

promos = [globals()[name] for name in globals()
              if name.endswith('_promo')
              and name != 'best_promo']

这里使用了globals()函数,我们就是使用这个函数来进行全局查找以'_promo'结尾的函数,并且过滤掉best_promo函数,又一次完成了我们的自动化优化。

最后,这篇blog就到这里了,相信你我都更加了解Python中的策略模式了,这里我推荐对Python感兴趣的朋友去看一下《Fluent Python》这本书,里面讲述了很多的高级特性, 更加让我们体验到Python中的美学。
首发legend's blog

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,560评论 4 361
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,104评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,297评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,869评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,275评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,563评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,833评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,543评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,245评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,512评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,011评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,359评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,006评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,062评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,825评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,590评论 2 273
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,501评论 2 268

推荐阅读更多精彩内容

  • 策略模式 大多数问题都可以使用多种方法来解决。以排序问题为例,对于以一定次序把元素放入一个列表,排序算法有很多。通...
    英武阅读 1,733评论 0 50
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 170,566评论 25 707
  • 设计模式汇总 一、基础知识 1. 设计模式概述 定义:设计模式(Design Pattern)是一套被反复使用、多...
    MinoyJet阅读 3,820评论 1 15
  • 工厂模式类似于现实生活中的工厂可以产生大量相似的商品,去做同样的事情,实现同样的效果;这时候需要使用工厂模式。简单...
    舟渔行舟阅读 7,621评论 2 17
  • 最近因为待业,所以平日里的写的东西开始天马行空、没有方向,想到哪里就写到哪里。看到朋友们在写连载小说。就心血来头,...
    芃芃的书房阅读 1,243评论 0 2