Python 中怎么函數(shù)功能進(jìn)行擴(kuò)展

Python 中怎么函數(shù)功能進(jìn)行擴(kuò)展，很多新手對此不是很清楚，為了幫助大家解決這個(gè)難題，下面小編將為大家詳細(xì)講解，有這方面需求的人可以來學(xué)習(xí)下，希望你能有所收獲。

裝飾器可以對原函數(shù)進(jìn)行功能擴(kuò)展，但還不需要修改原函數(shù)的內(nèi)容(開閉原則)，也不需要修改原函數(shù)的調(diào)用。

demo.py（裝飾器，@）：

# 閉包
def w1(func):
  def inner():
    # 對原函數(shù)進(jìn)行功能擴(kuò)展
    print("功能擴(kuò)展")
    func()
    # return func() # 如果原函數(shù)需要返回值,可以return
  return inner # 閉包
@w1
# 相當(dāng)于 f1 = w1(f1)
def f1():
  print('f1') # 原函數(shù)不需要修改
f1() # 原函數(shù)的調(diào)用也不需要修改

demo.py（裝飾器通用格式,對不定長參數(shù)并且有返回值的函數(shù)進(jìn)行裝飾）：

def set_func(func):
  def call_func(*args, **kwargs):
    print("裝飾器擴(kuò)展的功能")
    return func(*args, **kwargs) # 這里的*和*表示拆包。 不管有沒有返回值,return都沒問題。
  return call_func
@set_func # 相當(dāng)于 test1 = set_func(test1)
# 對含有不定長參數(shù)并且有返回值的函數(shù)進(jìn)行裝飾。
def test1(num, *args, **kwargs):
  print("-----test1----%d" % num)
  return "ok"
ret = test1(100)
print(ret)

demo.py（多個(gè)裝飾器的裝飾順序）：

def add_1(func):
  def call_func(*args, **kwargs):
    print("裝飾器1 擴(kuò)展的功能")
    return func(*args, **kwargs)
  return call_func
def add_2(func):
  def call_func(*args, **kwargs):
    print("裝飾器2 擴(kuò)展的功能")
    return func(*args, **kwargs)
  return call_func
@add_2
@add_1
# 先裝飾add_1，再裝飾add_2
def test1():
  print("------test1------")
test1() # 在調(diào)用函數(shù)之前就已經(jīng)裝飾好了。
# 裝飾器2 擴(kuò)展的功能
# 裝飾器1 擴(kuò)展的功能
# ------test1------

demo.py（用類充當(dāng)裝飾器）：

# 用類充當(dāng)裝飾器
class Test(object):
  def __init__(self, func):
    self.func = func
  def __call__(self, *args, **kwargs):
    print("這里是裝飾器添加的功能.....")
    return self.func(*args, **kwargs)
@Test # 相當(dāng)于get_str = Test(get_str) # 實(shí)例化對象,調(diào)用__init__方法。
def get_str():
  return "haha"
print(get_str())  # 實(shí)例對象(),會(huì)自動(dòng)調(diào)用對象的__call__方法。

@functools.wraps修飾裝飾器的內(nèi)層函數(shù)。(修飾內(nèi)層函數(shù)后,被裝飾器裝飾的函數(shù)的__name__、__doc__不會(huì)被裝飾器改變)

demo.py（@functools.wraps修飾裝飾器的內(nèi)層函數(shù)）：

# coding:utf-8
import functools # 導(dǎo)入
# 自定義的裝飾器
def login_required(func):
  @functools.wraps(func)
  # 裝飾器的內(nèi)層函數(shù),一般要加@functools.wraps裝飾器
  def wrapper(*arg, **kwargs):
    """wrapper的說明文檔"""
    # 。。。
    return func(*arg, **kwargs)
  return wrapper
# 使用自定義的裝飾器
@login_required
def demofunc():
  """demofunc的說明文檔"""
  pass
print(demofunc.__name__)  # 不加@functools.wraps裝飾器時(shí)："wrapper"。 加裝飾器時(shí)："demofunc"
print(demofunc.__doc__)  # 不加@functools.wraps裝飾器時(shí)："wrapper的說明文檔"。 加裝飾器時(shí)："demofunc的說明文檔"

看完上述內(nèi)容是否對您有幫助呢？如果還想對相關(guān)知識(shí)有進(jìn)一步的了解或閱讀更多相關(guān)文章，請關(guān)注億速云行業(yè)資訊頻道，感謝您對億速云的支持。