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
我最帅

  

 

 

 

 

 

 

 

posted @ 2017-06-25 16:28  KoishNee  阅读(306)  评论(0编辑  收藏  举报