python中的decorator的作用詳解

更新時間：2018年07月26日 08:28:27 作者：丹華抱一鷇音子

這篇文章主要介紹了python中的decorator的作用詳解，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

1、概念

裝飾器（decorator）就是：定義了一個函數(shù)，想在運行時動態(tài)增加功能，又不想改動函數(shù)本身的代碼?？梢云鸬綇陀么a的功能，避免每個函數(shù)重復性編寫代碼，簡言之就是拓展原來函數(shù)功能的一種函數(shù)。在python中，裝飾器（decorator）分為函數(shù)裝飾器和類裝飾器兩種。python中內置的@語言就是為了簡化裝飾器調用。

列出幾個裝飾器函數(shù)：

打印日志：@log

檢測性能：@performance

數(shù)據(jù)庫事務：@transaction

URL路由：@post('/register')

2、使用方法

（1）無參數(shù)decorator

編寫一個@performance，它可以打印出函數(shù)調用的時間。

import time

def performance(f):
 def log_time(x):
  t1 = time.time()
  res = f(x)
  t2 = time.time()
  print 'call %s() in %fs' %(f.__name__,(t2 - t1))
  return res
 return log_time

@performance
def factorial(n):
 return reduce(lambda x,y : x*y,range(1,n+1))

print factorial(10)

運行結果：

call factorial() in 0.006009s 2 3628800

運行原理：

此時，factorial就作為performance的函數(shù)對象，傳遞給f。當調用factorial(10)的時候也就是調用log_time(10)函數(shù)，而在log_time函數(shù)內部，又調用了f，這就造成了裝飾器的效果。說明f是被裝飾函數(shù)，而x是被裝飾函數(shù)的參數(shù)。

（2）帶參數(shù)decorator

請給 @performace 增加一個參數(shù)，允許傳入's'或'ms'。

import time

def performance(unit):
 def perf_decorator(f):
  def wrapper(*args, **kw):
   t1 = time.time()
   r = f(*args, **kw)
   t2 = time.time()
   t = (t2 - t1)*1000 if unit =='ms' else (t2 - t1)
   print 'call %s() in %f %s'%(f.__name__, t, unit)
   return r
  return wrapper
 return perf_decorator

@performance('ms') 
def factorial(n):
 return reduce(lambda x,y: x*y, range(1, n+1))

print factorial(10)

運行結果：