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