Python之装饰器

86754

假设有这样一种场景：
开发人员做了一个App，提醒用户下单秒杀商品。

def buy()
    print "快下单秒杀家具吧!"

def buy()
    print "快下单秒杀蔬菜吧!"

def buy()

    print "快下单秒杀服装吧!"

......
下面有1000个类似的函数。

现在产品经理提了一个需求，要求用户在秒杀前需要经过登录认证，才能下单。

我们可以这样实现：

1 2 3 4 5 6 7 8 9 10 11 12 13 14

def login()     print "登录成功！" def buy()     login()     print "快下单秒杀家具吧!" def buy()     login()     print "快下单秒杀蔬菜吧!" def buy()     login()     print "快下单秒杀服装吧!"

......

上面这种方法可以实现功能，但是修改了源代码。开发需要遵循一个原则，就是已经实现的功能代码是不允许被修改的。像这样的场景，我们可以使用装饰器。

先看看装饰器怎么写

login(buy):    buy@buy():    buy()执行结果：
Welcome login
快下单秒杀吧!

上面这段代码的执行过程：

执行buy()函数时，需要先执行login()函数，然后返回buy()函数的内存地址。然后再执行buy()函数。
这样一来，在执行buy()函数时，即执行了登录验证的功能，又执行了下单秒杀的功能。
所有python装饰器的原理是，先写一个新的函数，然后将另外一个函数的函数名当成参数写在新函数中，并在新函数中返回这个函数的内存地址。注意return buy 和return buy()的区别