python-装饰器的简单使用

一、介绍

首先我们先来看一个简单的例子,在基础平台中有一个home()和tv()函数，在业务平台中调用此函数时，给出了响应的打印内容：

基础平台：
def home():
    print('welcome to home page')
def tv(name):
    print('welcome to tv page')

业务平台A：
home()
tv()

业务平台B:
home()
tv()

　　　　

但是，在后续的需求中规定，在调用此函数时，还需要进行一个用户验证

因此程序猿小A的做法是和每个业务部门交涉，每个业务部门自己写代码，在调用基础平台的时候，先进行验证：

基础平台：
def home():
    print('welcome to home page')
def tv(name):
    print('welcome to tv page')

业务平台A：
#验证
home()
tv()

业务平台B:
#验证
home()
tv()

小A当天被开除了。。。

后来程序猿小B接手了这份工作，小B的做法是在基础平台中加入验证：

基础平台：
def home():
    #验证
    print('welcome to home page')
def tv(name):
    #验证
    print('welcome to tv page')

业务平台A：
home()
tv()

业务平台B:
home()
tv()

小B的做法相比于小A来说，好了一点，避免了业务平台的修改，老板为了奖励小B，因此在一周后把小B开除了。。。

因为小B改动了源代码

此后，程序猿小C又接受了这个工作。。。

小C说：

写代码要满足以下下两个原则：

　　封闭：已实现的功能代码不允许被修改

　　开放：已实现的功能代码可以被扩展：

因此，小C打算在此处使用了装饰器的功能（见下文）

总结：

装饰器本质上是一个函数，可以在其他函数不做任何代码变动的情况下，提供一些扩展的功能

二、简单装饰器的使用

小C对代码做出了如下修改（拿tv方法举例）：

def login(func):                   #定义了一个登录验证的方法
    print('登录验证成功')            #假设此处就是一个验证功能
    return func                    

def tv():
    print('welcome to tv page' )
tv=login(tv)                       #对tv进行了赋值，此时还没执行tv方法，只是把该方法放入到了内存中

tv()

---执行结果---
登录验证成功
welcom to tv page

该代码是如何实现验证的呢，步骤如下：

　　①执行login(tv)时，先不执行tv()方法，只是把tv()方法到了内存中，然后执行login()

　　②在login()中进行登录验证，之后返回tv()的内存地址，此时tv()还未执行

　　③把tv()的内存地址重新赋值给tv，此时tv变量的值依旧为tv()方法的内容

　　④通过调动tv()执行tv()方法

再经过一些细小的调整，可以写成如下的形式：

def login(func):
    print('登录验证成功')
    return func                    

@login
def tv():
    print('welcome to tv page' )                   

tv()

其中， @XXX  语法，就是一个装饰器，也称为“语法糖”

至此，一个简单的装饰器雏形已经完成

但是，此处依旧有一个缺点，那就是 即使业务方没有调用tv()，代码中也依旧会执行login()中的验证，为了让login()中的验证

不预先执行，因此小C继续对代码进行调整

def login(func):
    def inner():
        print('登录验证成功')
        func()
    return inner

@login
def tv():
    print('welcome to tv page' )

tv()

此处实现验证功能的步骤如下：

　　①调用login()方法，返回inner方法的内存地址，此时func的值为tv()内存地址即内容为：

      def tv():
           print('welcome to tv page' )

　　②把inner方法的内存地址赋值给了变量tv

　　③调用tv()，因为tv的值已经为inner的内存地址，因此调用的是inner()方法

　　④实现inner(）中的验证功能

　　⑤验证成功后，执行inner()中的func()，打印出‘welcome to tv page’

至此，一个简单的装饰器才算真正完成

如果想在调用方法中实现带参数，可以这样写：

def login(func):
    def inner(arg):
        print('登录验证成功')
        func(arg)
    return inner

@login
def tv(name):
    print('welcom to [%s] tv page' %name )

@login
def movie(name):
    print('welcom to [%s] movie page' %name )

tv('nee')
movie('nee')

-----结果-----
登录验证成功
welcom to [nee] tv page
登录验证成功
welcom to [nee] movie page

在上面的例子中，都是实现了在调用tv()和movie(）时带入一个参数的情况的。但如果在调用tv()时需要一个参数，在调用movie()时用两个参数呢？

因此可以使用一个动态参数的方法：

def login(func):
    def inner(*args):                #代表动态参数
        print('登录验证成功')
        func(*args)
    return inner

@login
def tv(name,password):
    print('welcom to [%s] tv page' %name )

@login
def movie(name):
    print('welcom to [%s] movie page' %name )

tv('nee',123456)
movie('nee')

　　

如果再进一步，想要在调用tv()方法后，打印一个结果

def login(func):
    def inner(*args,**kwargs):
        print('登录验证成功')
        return func(*args,**kwargs)        #注意此处的return
    return inner

@login
def tv(name,password):
    print('welcom to [%s] tv page' %name )
    return '我最帅'

a=tv('nee',password=123456)
print(a)

-----结果-----
登录验证成功
welcom to [nee] tv page
我最帅