快捷導(dǎo)航

Python中裝飾器兼容加括號(hào)和不加括號(hào)的寫法詳解

更新時(shí)間：2017年07月05日 15:20:21 作者：卡瓦邦噶！

這篇文章主要給大家介紹了關(guān)于Python中裝飾器兼容加括號(hào)和不加括號(hào)寫法的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起看看吧。

使用Django的時(shí)候，我發(fā)現(xiàn)一個(gè)很神奇的裝飾器： @login_required，這是控制一個(gè)view的權(quán)限的，比如一個(gè)視圖必須登錄才可以訪問(wèn)，可以這樣用：

@login_required
def my_view(request):
 ...
 return render(...)

同時(shí)，如果要達(dá)到這樣一種效果：如果用戶沒有登錄，那么就把用戶重定向到登錄界面，可以這樣用：

@login_required(login_url='/accounts/login/')
def my_view(request):
 ...
 return render(...)

所以這個(gè)裝飾器可以帶括號(hào)寫，又可以不帶括號(hào)寫。很神奇有沒有。正常的接收參數(shù)的裝飾器，就算沒參數(shù)也應(yīng)該寫成@login_required的

好奇去查了一下，在stackoverflow找到一種實(shí)現(xiàn)，挺有意思的。先曬出答案：

def doublewrap(f):
 '''
 a decorator decorator, allowing the decorator to be used as:
 @decorator(with, arguments, and=kwargs)
 or
 @decorator
 '''
 @wraps(f)
 def new_dec(*args, **kwargs):
  if len(args) == 1 and len(kwargs) == 0 and callable(args[0]):
   # actual decorated function
   return f(args[0])
  else:
   # decorator arguments
   return lambda realf: f(realf, *args, **kwargs)
 
 return new_dec

使用起來(lái)很簡(jiǎn)單，只要給裝飾器用@doublewrap裝飾一下，這個(gè)裝飾器就支持寫括號(hào)和不寫括號(hào)兩種寫法了。

def test_doublewrap():
 from util import doublewrap
 from functools import wraps 
 
 @doublewrap
 def mult(f, factor=2):
  '''multiply a function's return value'''
  @wraps(f)
  def wrap(*args, **kwargs):
   return factor*f(*args,**kwargs)
  return wrap
 
 # try normal
 @mult
 def f(x, y):
  return x + y
 
 # try args
 @mult(3)
 def f2(x, y):
  return x*y
 
 # try kwargs
 @mult(factor=5)
 def f3(x, y):
  return x - y
 
 assert f(2,3) == 10
 assert f2(2,5) == 30
 assert f3(8,1) == 5*7

原理也不難，只有短短不到10行代碼。

裝飾器我們都知道，是用來(lái)處理一個(gè)函數(shù)，返回一個(gè)新的函數(shù)的（如果你不理解裝飾器，可以看一下這個(gè)經(jīng)典的解釋）。

new_func = decorator(func)

我們使用的，就是被裝飾器裝飾的新函數(shù)了。裝飾器只是一個(gè)語(yǔ)法糖，其實(shí)它也是一個(gè)函數(shù)，給它傳入一個(gè)函數(shù)作為參數(shù)，就返回一個(gè)新的函數(shù)。那么既然裝飾器也是一個(gè)函數(shù)，我們就可以用裝飾器裝飾這個(gè)函數(shù)。也就是，“裝飾器的裝飾器”。

裝飾器第一個(gè)參數(shù)肯定是原函數(shù)，如果裝飾器可以接收參數(shù)的話，要么第一個(gè)參數(shù)是原函數(shù)，后面跟別的參數(shù)；要么就只有原函數(shù)一個(gè)參數(shù)。所以，我們這個(gè)“裝飾器的裝飾器”做的事情就是，判斷裝飾器接收的參數(shù)，如果只有一個(gè)并且第一個(gè)參數(shù)是可調(diào)用的（callable），那么這就是一個(gè)無(wú)參數(shù)的裝飾器（不需要加括號(hào)）。如果還有別的參數(shù)，就返回一個(gè)生成裝飾器的函數(shù)(decorator_maker)。

裝飾器是一個(gè)函數(shù)。裝飾器被裝飾過(guò)之后，這個(gè)裝飾器運(yùn)行之前就會(huì)先運(yùn)行裝飾器的裝飾器的代碼，也就是我們的doublewrapp。然后返回值可能是一個(gè)裝飾器，也可能是一個(gè)裝飾器的maker（有參數(shù)的裝飾器），然后裝飾器再執(zhí)行，裝飾原函數(shù)。

這里有點(diǎn)繞，因?yàn)楸緛?lái)裝飾器里面一般就會(huì)有三四層函數(shù)了，（maker, decorator, wrapper, realfunc），再加上一個(gè)裝飾器的裝飾器，會(huì)有點(diǎn)理解困難。如果理解不了，最好不要對(duì)著網(wǎng)上的博文（包括本文）企圖格物致知了，多去看看代碼，多寫一寫。

總結(jié)

以上就是這篇文章的全部?jī)?nèi)容了，希望本文的內(nèi)容對(duì)大家的學(xué)習(xí)或者工作能帶來(lái)一定的幫助，如果有疑問(wèn)大家可以留言交流，謝謝大家對(duì)腳本之家的支持。

參考資料

How to create a Python decorator that can be used either with or without parameters?

您可能感興趣的文章: