Python中裝飾器怎么用

這篇文章將為大家詳細(xì)講解有關(guān)Python中裝飾器怎么用，小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

裝飾器(decorator)功能

• 引入日志

• 函數(shù)執(zhí)行時(shí)間統(tǒng)計(jì)

• 執(zhí)行函數(shù)前預(yù)備處理

• 執(zhí)行函數(shù)后清理功能

• 權(quán)限校驗(yàn)等場(chǎng)景

• 緩存

裝飾器示例

例1:無參數(shù)的函數(shù)

from time import ctime, sleep

def timefun(func):
 def wrappedfunc():
  print("%s called at %s"%(func.__name__, ctime()))
  func()
 return wrappedfunc

@timefun
def foo():
 print("I am foo")

foo()
sleep(2)
foo()

分析如下：

上面代碼理解裝飾器執(zhí)行行為可理解成

foo = timefun(foo)


1，foo先作為參數(shù)賦值給func后,foo接收指向timefun返回的wrappedfunc
2，調(diào)用foo(),即等價(jià)調(diào)用wrappedfunc()
3，內(nèi)部函數(shù)wrappedfunc被引用，所以外部函數(shù)的func變量(自由變量)并沒有釋放
4，func里保存的是原foo函數(shù)對(duì)象

例2:被裝飾的函數(shù)有參數(shù)

from time import ctime, sleep

def timefun(func):
 def wrappedfunc(a, b):
  print("%s called at %s"%(func.__name__, ctime()))
  print(a, b)
  func(a, b)
 return wrappedfunc

@timefun
def foo(a, b):
 print(a+b)

foo(3,5)
sleep(2)
foo(2,4)

例3:被裝飾的函數(shù)有不定長(zhǎng)參數(shù)

from time import ctime, sleep

def timefun(func):
 def wrappedfunc(*args, **kwargs):
  print("%s called at %s"%(func.__name__, ctime()))
  func(*args, **kwargs)
 return wrappedfunc

@timefun
def foo(a, b, c):
 print(a+b+c)

foo(3,5,7)
sleep(2)
foo(2,4,9)

例4:裝飾器中的return

from time import ctime, sleep

def timefun(func):
 def wrappedfunc():
  print("%s called at %s"%(func.__name__, ctime()))
  func()
 return wrappedfunc

@timefun
def foo():
 print("I am foo")

@timefun
def getInfo():
 return '----hahah---'

foo()
sleep(2)
foo()


print(getInfo())

執(zhí)行結(jié)果:

foo called at Sun Jun 18 00:31:53 2017
I am foo
foo called at Sun Jun 18 00:31:55 2017
I am foo
getInfo called at Sun Jun 18 00:31:55 2017
None

如果修改裝飾器為return func()，則運(yùn)行結(jié)果：

foo called at Sun Jun 18 00:34:12 2017
I am foo
foo called at Sun Jun 18 00:34:14 2017
I am foo
getInfo called at Sun Jun 18 00:34:14 2017
----hahah---

小結(jié)：一般情況下為了讓裝飾器更通用，可以有return

例5:裝飾器帶參數(shù),在原有裝飾器的基礎(chǔ)上，設(shè)置外部變量

from time import ctime, sleep

def timefun_arg(pre="hello"):
 def timefun(func):
  def wrappedfunc():
   print("%s called at %s %s"%(func.__name__, ctime(), pre))
   return func()
  return wrappedfunc
 return timefun

@timefun_arg("itcast")
def foo():
 print("I am foo")

@timefun_arg("python")
def too():
 print("I am too")

foo()
sleep(2)
foo()

too()
sleep(2)
too()
可以理解為

foo()==timefun_arg("itcast")(foo)()

例6：類裝飾器

裝飾器函數(shù)其實(shí)是這樣一個(gè)接口約束，它必須接受一個(gè)callable對(duì)象作為參數(shù)，然后返回一個(gè)callable對(duì)象。在Python中一般callable對(duì)象都是函數(shù)，但也有例外。只要某個(gè)對(duì)象重寫了 call() 方法，那么這個(gè)對(duì)象就是callable的。

class Test():
 def __call__(self):
  print('call me!')

t = Test()
t() # call me
類裝飾器demo


class Test(object):
 def __init__(self, func):
  print("---初始化---")
  print("func name is %s"%func.__name__)
  self.__func = func
 def __call__(self):
  print("---裝飾器中的功能---")
  self.__func()

說明：

1. 當(dāng)用Test來裝作裝飾器對(duì)test函數(shù)進(jìn)行裝飾的時(shí)候，首先會(huì)創(chuàng)建Test的實(shí)例對(duì)象,并且會(huì)把test這個(gè)函數(shù)名當(dāng)做參數(shù)傳遞到init方法中
即在init方法中的func變量指向了test函數(shù)體
2. test函數(shù)相當(dāng)于指向了用Test創(chuàng)建出來的實(shí)例對(duì)象
3. 當(dāng)在使用test()進(jìn)行調(diào)用時(shí)，就相當(dāng)于讓這個(gè)對(duì)象()，因此會(huì)調(diào)用這個(gè)對(duì)象的call方法
4. 為了能夠在call方法中調(diào)用原來test指向的函數(shù)體，所以在init方法中就需要一個(gè)實(shí)例屬性來保存這個(gè)函數(shù)體的引用
所以才有了self.func = func這句代碼，從而在調(diào)用__call方法中能夠調(diào)用到test之前的函數(shù)體

@Test 
def test(): 
print(“—-test—”) 
test() 
showpy()#如果把這句話注釋，重新運(yùn)行程序，依然會(huì)看到”–初始化–”

運(yùn)行結(jié)果如下：

---初始化---
func name is test
---裝飾器中的功能---
----test---

wraps函數(shù)

使用裝飾器時(shí)，有一些細(xì)節(jié)需要被注意。例如，被裝飾后的函數(shù)其實(shí)已經(jīng)是另外一個(gè)函數(shù)了（函數(shù)名等函數(shù)屬性會(huì)發(fā)生改變）。

添加后由于函數(shù)名和函數(shù)的doc發(fā)生了改變，對(duì)測(cè)試結(jié)果有一些影響，例如:

def note(func):
 "note function"
 def wrapper():
  "wrapper function"
  print('note something')
  return func()
 return wrapper

@note
def test():
 "test function"
 print('I am test')

test()
print(test.__doc__)

運(yùn)行結(jié)果

note something
I am test
wrapper function

所以，Python的functools包中提供了一個(gè)叫wraps的裝飾器來消除這樣的副作用。例如：

import functools
def note(func):
 "note function"
 @functools.wraps(func)
 def wrapper():
  "wrapper function"
  print('note something')
  return func()
 return wrapper

@note
def test():
 "test function"
 print('I am test')

test()
print(test.__doc__)

運(yùn)行結(jié)果

note something
I am test
test function

關(guān)于“Python中裝飾器怎么用”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。