装饰器模式(Decorator Pattern)

1 点餐系统


In [1]:
class Beverage(object):
    name = ""
    price = 0.0
    type = "BEVERAGE"
    def getPrice(self):
        return self.price
    def setPrice(self, price):
        self.price = price
    def getName(self):
        return self.name


class coke(Beverage):
    def __init__(self):
        self.name = "coke"
        self.price = 4.0


class milk(Beverage):
    def __init__(self):
        self.name = "milk"
        self.price = 5.0

除了基本配置,快餐店卖可乐时,可以选择加冰,如果加冰的话,要在原价上加0.3元;卖牛奶时,可以选择加糖,如果加糖的话,要原价上加0.5元。怎么解决这样的问题?可以选择装饰器模式来解决这一类的问题。首先,定义装饰器类:


In [5]:
class drinkDecorator(object):
    def getName(self):
        pass
    def getPrice(self):
        pass

class iceDecorator(drinkDecorator):
    def __init__(self,beverage):
        self.beverage=beverage
    def getName(self):
        return self.beverage.getName()+"+ice"
    def getPrice(self):
        return self.beverage.getPrice()+0.3

class sugarDecorator(drinkDecorator):
    def __init__(self,beverage):
        self.beverage=beverage
    def getName(self):
        return self.beverage.getName()+"+sugar"
    def getPrice(self):
        return self.beverage.getPrice()+0.5

构建好装饰器后,在具体的业务场景中,就可以与饮料类进行关联。以可乐+冰为例,示例业务场景如下:


In [6]:
coke_cola = coke()
print('Name:', coke_cola.getName(),' Price:',coke_cola.getPrice())
ice_coke=iceDecorator(coke_cola)
print('Name:', ice_coke.getName(), ' Price:',ice_coke.getPrice())


Name: coke  Price: 4.0
Name: coke+ice  Price: 4.3

装饰器模式和上一节说到的代理模式非常相似,可以认为,装饰器模式就是代理模式的一个特殊应用,两者的共同点是都具有相同的接口,不同点是侧重对主题类的过程的控制,而装饰模式则侧重对类功能的加强或减弱。

2 Python语法装饰器


In [8]:
def log(func):
    def wrapper(*args, **kw):
        print ('call %s():' % func.__name__)
        return func(*args, **kw)
    return wrapper

@log
def now():
    print('2016-12-04')
now()


call now():
2016-12-04

3 advantages

  • 装饰器模式是继承方式的一个替代方案,可以轻量级的扩展被装饰对象的功能;
  • Python的装饰器模式是实现AOP的一种方式,便于相同操作位于不同调用位置的统一管理。

4 Usages

  • 需要扩展、增强或者减弱一个类的功能,如本例。

5 Disadvantages

  • 多层装饰器的调试和维护有比较大的困难。