86754 发表于 2016-7-21 09:21:01

Python之装饰器

假设有这样一种场景:
开发人员做了一个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()的区别





页: [1]
查看完整版本: Python之装饰器