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,'bc')

執行結果：

log開始 ... test1 module1
test1 .. a
log結束 ...
log開始 ... test2 module1
test2 .. a bc
log結束 ...

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。