自定義的修飾器.
一個不帶參數(shù)的裝飾器:
@decorator
def func():
pass
上面這段代碼就等于下面的實現(xiàn):
fun = decorator(fun)
而帶有參數(shù)的裝飾器:
@decorator('Amrzs')
def func():
pass
這段代碼就等于下面的實現(xiàn):
fun = decorator('Amrzs')(func)
但是這個操作因為每次實現(xiàn)裝飾器都要這樣實現(xiàn)一遍的話太麻煩,所以python內(nèi)置的@語法給我們實現(xiàn)了這些功能.
又由于被decorator裝飾之后的函數(shù)霍转，它們的__name__已經(jīng)從原來的now變成了裝飾器中的函數(shù)簽名, 所以，需要把原始函數(shù)的__name__等屬性復(fù)制到wrapper()函數(shù)中秃诵，否則，有些依賴函數(shù)簽名的代碼執(zhí)行就會出錯颅围。
即需要編寫wrapper.__name__ = func.__name__這樣的代碼忍燥，但是Python內(nèi)置的functools.wraps函數(shù)已經(jīng)幫我們做了這個事情. 所以砚嘴，一個完整的decorator的寫法如下：
import functools
def log(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print('call %s():' % func.__name__)
return func(*args, **kw)
return wrapper
或者說帶參數(shù)的:
import functools
def log(text):
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print('%s %s():' % (text, func.__name__))
return func(*args, **kw)
return wrapper
return decorator
而有了上面的知識,再基于這個裝飾器的@原理就很容易實現(xiàn)一個更加具有特定功能的裝飾器,比如我們現(xiàn)在有一個@permission_required(permission)這樣的裝飾器.這個裝飾器的功能是根據(jù)permission參數(shù)來執(zhí)行相應(yīng)權(quán)限的操作.
現(xiàn)在我們來實現(xiàn)一個只允許管理員操作的裝飾器:
def admin_required(func):
return permission_required(Permission.ADMINISTER)(func)
原理很簡單,其實就是內(nèi)定了permission_required()函數(shù)參數(shù)的值,默認(rèn)為管理員權(quán)限,又因為這種結(jié)構(gòu)符合python的@語法糖,所以直接就可以用@admin_required這樣的方式來使用這個裝飾器!
哈哈!
