一、介绍

首先我们先来看一个简单的例子,在基础平台中有一个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
我最帅

  

 

05-18 03:29