裝飾器是 Python 的一個(gè)重要部分。簡(jiǎn)單地說(shuō)：他們是修改其他函數(shù)的功能的函數(shù)。他們有助于讓我們的代碼更簡(jiǎn)短，也更Pythonic。大多數(shù)初學(xué)者不知道在哪兒使用它們，所以我將要分享下，哪些區(qū)域里裝飾器可以讓你的代碼更簡(jiǎn)潔。首先，讓我們討論下如何寫(xiě)你自己的裝飾器

什么是裝飾器

從字面意思上來(lái)看，裝飾器是用來(lái)裝飾其他東西的工具。在python中裝飾器分為函數(shù)裝飾器和類(lèi)裝飾器。
簡(jiǎn)而言之，函數(shù)裝飾器是用來(lái)裝飾函數(shù)的裝飾器，其主要目的是增加目標(biāo)函數(shù)的功能，類(lèi)裝飾器也就是裝飾類(lèi)的裝飾器，增加類(lèi)的功能。

函數(shù)裝飾器

裝飾器本質(zhì)是嵌套函數(shù)
下面是一個(gè)簡(jiǎn)單的裝飾器

# fun1為裝飾器名稱，function指的是被裝飾的函數(shù)
def fun1(function):
	def fun2():
		print("開(kāi)始了！")
		function() # 執(zhí)行被裝飾的函數(shù)或者 或者 return function()
	return fun2

外面的fun1()需要傳入一個(gè)參數(shù)，這個(gè)參數(shù)是用來(lái)傳入要被裝飾的函數(shù)，里面的fun2()需要執(zhí)行function這個(gè)參數(shù)。
這是將函數(shù)當(dāng)成參數(shù)傳入另一個(gè)函數(shù)。

使用方法：
在需要裝飾的函數(shù)上添加@裝飾器名稱

@fun1
def fun3():
	print("aaa")

fun3()

可以等價(jià)成這樣：

def fun3():
	print("aaa")
fun = fun1(fun3)
fun()

輸出結(jié)果是一致的

執(zhí)行結(jié)果：

開(kāi)始了！
aaa

如果傳入的函數(shù)需要參數(shù)可以這樣寫(xiě)：

def fun1(function):
	def fun2(a):
		print("開(kāi)始了！")
		function(a) 
	return fun2

@fun1
def fun3(a):
	print(a)
fun3("aaa")

輸出結(jié)果：

開(kāi)始了！
aaa

如果不清楚傳入的函數(shù)需要多少個(gè)參數(shù)，可以這樣寫(xiě)：

def fun1(function):
	def fun2(*args, **kwargs):
		print("開(kāi)始了！")
		function(*args, **kwargs) 
	return fun2

@fun1
def fun3(a,b):
	print(a,b)

@fun1	
def fun4(a,b,c):
    print(a,b,c)
    
fun3("aaa","bbb")
fun4("ccc","ddd","eee")

運(yùn)行結(jié)果：

開(kāi)始了！
aaa bbb
開(kāi)始了！
ccc ddd eee

如果函數(shù)裝飾器需要添加參數(shù)，可以這樣寫(xiě)：

def fun(msg):
	def fun1(function):
		def fun2(*args, **kwargs):
			print("開(kāi)始了！{}".format(msg))
			function(*args, **kwargs) 
		return fun2
	return fun1

@fun(msg="小朋友")
def fun3(a,b):
	print(a,b)

fun3("aaa","bbb")

運(yùn)行結(jié)果：

開(kāi)始了！小朋友
aaa bbb

類(lèi)裝飾器

類(lèi)裝飾器的用法基本和函數(shù)裝飾器一樣，只不過(guò)是用類(lèi)寫(xiě)的裝飾器

class fun1(object):
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print(("開(kāi)始了！")
        self.func(*args, **kwargs)

@fun1
def fun2(a,b):
	print(a,b)

fun2("aaa","bbb")

類(lèi)裝飾用到了類(lèi)的__call__方法

運(yùn)行結(jié)果：

開(kāi)始了！
aaa bbb

還可以這樣寫(xiě)：

class fun(object):
    def __init__(self, msg):
        self.msg = msg
        

    def __call__(self, func):
        def fun1(*args, **kwargs):
            print("開(kāi)始了！{0}".format(self.msg))
            func(*args, **kwargs)
        return fun1

@fun(msg="小朋友")
def fun2(a, b):
    print(a, b)

fun2("aaa,","bbb")

運(yùn)行結(jié)果：