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

1

2

3

4

@login_required

defmy_view(request):

...

returnrender(...)

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

1

2

3

4

@login_required(login_url='/accounts/login/')

defmy_view(request):

...

returnrender(...)

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

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

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

defdoublewrap(f):

'''

a decorator decorator, allowing the decorator to be used as:

@decorator(with, arguments, and=kwargs)

or

@decorator

'''

@wraps(f)

defnew_dec(*args,**kwargs):

iflen(args)==1andlen(kwargs)==0andcallable(args[0]):

# actual decorated function

returnf(args[0])

else:

# decorator arguments

returnlambdarealf:f(realf,*args,**kwargs)

returnnew_dec

使用起來很簡單，只要給裝飾器用@doublewrap裝飾一下，這個裝飾器就支持寫括號和不寫括號兩種寫法了。

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

deftest_doublewrap():

fromutilimportdoublewrap

fromfunctoolsimportwraps

@doublewrap

defmult(f,factor=2):

'''multiply a function's return value'''

@wraps(f)

defwrap(*args,**kwargs):

returnfactor*f(*args,**kwargs)

returnwrap

# try normal

@mult

deff(x,y):

returnx+y

# try args

@mult(3)

deff2(x,y):

returnx*y

# try kwargs

@mult(factor=5)

deff3(x,y):

returnx-y

assertf(2,3)==10

assertf2(2,5)==30

assertf3(8,1)==5*7

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

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

1

new_func=decorator(func)

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

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

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

這里有點繞，因為本來裝飾器里面一般就會有三四層函數(shù)了，(maker, decorator, wrapper, realfunc)，再加上一個裝飾器的裝飾器，會有點理解困難。如果理解不了，最好不要對著網(wǎng)上的博文(包括本文)企圖格物致知了，多去看看代碼，多寫一寫。

2017年12月31日更新：《Python3 Cookbook》中提供了另一種實現(xiàn)方法，代碼如下：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

fromfunctoolsimportwraps,partial

importlogging

deflogged(func=None,*,level=logging.DEBUG,name=None,message=None):

iffuncisNone:

returnpartial(logged,level=level,name=name,message=message)

logname=nameifnameelsefunc.__module__

log=logging.getLogger(logname)

logmsg=messageifmessageelsefunc.__name__

@wraps(func)

defwrapper(*args,**kwargs):

log.log(level,logmsg)

returnfunc(*args,**kwargs)

returnwrapper

# Example use

@logged

defadd(x,y):

returnx+y

@logged(level=logging.CRITICAL,name='example')

defspam():

print('Spam!')

這種原理也比較好理解，看例子1，我們知道這等價于logged(add)，第一個參數(shù)是函數(shù)，所以直接返回。看例子2，等價于logged(level=logging.CRITICAL, name='example')(spam)，logged的第一個參數(shù)func是None(如果是裝飾器語法，第一個參數(shù)會默認傳入函數(shù)，而這里直接是函數(shù)調(diào)用，并非裝飾器。所以會返回一個partial函數(shù)，這個partial函數(shù)是真正的裝飾器，然后再走后面裝飾器的路子。其實，這種方法也是利用了“判斷傳入裝飾器的參數(shù)”。

參考資料

總結(jié)

以上是生活随笔為你收集整理的python if后面要不要加括号_Python装饰器兼容加括号与不加括号的写法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。