7、装饰器模式

一、装饰器模式

装饰器模式定义如下：动态地给一个对象添加一些额外的职责。在增加功能方面，装饰器模式比生成子类更为灵活。修饰器（Decorator）模式能够以透明的方式（不会影响其他对象）动态地将功能添加到一个对象中。

从实现的角度来说，Python修饰器是一个可调用对象（函数、方法、类），接受一个函数对象fin作为输入，并返回另一个函数对象fout。这意味着可以将任何具有这些属性的可调用对象当作一个修饰器。

修饰器模式和Python修饰器之间并不是一对一的等价关系。Python修饰器能做的实际上比修饰器模式多得多，其中之一就是实现修饰器模式。下面看一个快餐店买饮料的示例：

class Beverage(object):
    name = ""
    price = 0.0
    type = "BEVERAGE"
    
    def get_price(self):
        return self.price
    
    def set_price(self, price):
        self.price = price
        
    def get_name(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元。怎么解决这样的问题？可以选择装饰器模式来解决这一类的问题。首先，定义装饰器类：

class DrinkDecorator(object):
    def get_name(self):
        pass

    def get_price(self):
        pass


class IceDecorator(DrinkDecorator):
    def __init__(self, beverage):
        self.beverage = beverage

    def get_name(self):
        return self.beverage.get_name() + " +ice"

    def get_price(self):
        return self.beverage.get_price() + 0.3


class SugarDecorator(DrinkDecorator):
    def __init__(self, beverage):
        self.beverage = beverage

    def get_name(self):
        return self.beverage.get_name() + " +sugar"

    def get_price(self):
        return self.beverage.get_price() + 0.5

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

if __name__ == "__main__":
    coke_cola = Coke()
    print("Name:%s" % coke_cola.get_name())
    print("Price:%s" % coke_cola.get_price())

    ice_coke = IceDecorator(coke_cola)
    print("Name:%s" % ice_coke.get_name())
    print("Price:%s" % ice_coke.get_price())

打印结果：

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

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

二、装饰器模式的优点和应用场景

优点：
1、装饰器模式是继承方式的一个替代方案，可以轻量级的扩展被装饰对象的功能；
2、Python的装饰器模式是实现AOP的一种方式，便于相同操作位于不同调用位置的统一管理。
应用场景：
1、需要扩展、增强或者减弱一个类的功能，如本例。

三、装饰器模式的缺点

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