python函數(shù)修飾符@的作用是為現(xiàn)有函數(shù)增加額外的功能，常用于插入日志、性能測(cè)試、事務(wù)處理等等。
創(chuàng)建函數(shù)修飾符的規(guī)則：
（1）修飾符是一個(gè)函數(shù)
（2）修飾符取被修飾函數(shù)為參數(shù)
（3）修飾符返回一個(gè)新函數(shù)
（4）修飾符維護(hù)被維護(hù)函數(shù)的簽名
例子1：被修飾函數(shù)不帶參數(shù)
def log(func):
def wrapper():
print('log開始 ...')
func()
print('log結(jié)束 ...')
return wrapper
@log
def test():
print('test ..')
test()
運(yùn)行結(jié)果：
log開始 ...
test ..
log結(jié)束 ...
例子2：使用functools模塊提供的修改函數(shù)屬性的方法wraps
def log(func):
def wrapper():
print('log開始 ...')
func()
print('log結(jié)束 ...')
return wrapper
@log
def test1():
print('test1 ..')
def test2():
print('test2 ..')
print(test1.__name__)
print(test2.__name__)
運(yùn)行結(jié)果：
wrapper
test2
可見test1的函數(shù)名稱變了，如果某些代碼用到就會(huì)出問題，可以使用functools模塊提供的修改函數(shù)屬性的方法wraps
from functools import wraps
def log(func):
@wraps(func)
def wrapper():
print('log開始 ...')
func()
print('log結(jié)束 ...')
return wrapper
@log
def test1():
print('test1 ..')
def test2():
print('test2 ..')
print(test1.__name__)
print(test2.__name__)
運(yùn)行結(jié)果：
test1
test2
例子3：被修飾函數(shù)帶參數(shù)
from functools import wraps
def log(func):
@wraps(func)
def wrapper(*args,**kwargs):
print('log開始 ...',func.__name__)
ret = func(*args,**kwargs)
print('log結(jié)束 ...')
return ret
return wrapper
@log
def test1(s):
print('test1 ..', s)
return s
@log
def test2(s1, s2):
print('test2 ..', s1, s2)
return s1 + s2
test1('a')
test2('a','bc')
運(yùn)行結(jié)果：
log開始 ... test1
test1 .. a
log結(jié)束 ...
log開始 ... test2
test2 .. a bc
log結(jié)束 ...
例子4：修飾符帶參數(shù)，需要比上面例子多一層包裝
from functools import wraps
def log(arg):
def _log(func):
@wraps(func)
def wrapper(*args,**kwargs):
print('log開始 ...',func.__name__, arg)
ret = func(*args,**kwargs)
print('log結(jié)束 ...')
return ret
return wrapper
return _log
@log('module1')
def test1(s):
print('test1 ..', s)
return s
@log('module1')
def test2(s1, s2):
print('test2 ..', s1, s2)
return s1 + s2
test1('a')
test2('a','bc')
運(yùn)行結(jié)果：
log開始 ... test1 module1
test1 .. a
log結(jié)束 ...
log開始 ... test2 module1
test2 .. a bc
log結(jié)束 ...
以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號(hào)聯(lián)系： 360901061
您的支持是博主寫作最大的動(dòng)力，如果您喜歡我的文章，感覺我的文章對(duì)您有幫助，請(qǐng)用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點(diǎn)擊下面給點(diǎn)支持吧，站長(zhǎng)非常感激您！手機(jī)微信長(zhǎng)按不能支付解決辦法：請(qǐng)將微信支付二維碼保存到相冊(cè)，切換到微信，然后點(diǎn)擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對(duì)您有幫助就好】元
