快捷導(dǎo)航

Python 裝飾器代碼解析

更新時間：2021年11月02日 16:03:58 作者：TORTOISR

裝飾器(Decorators)是 Python 的一個重要部分。簡單地說：他們是修改其他函數(shù)的功能的函數(shù)。他們有助于讓我們的代碼更簡短，也更Pythonic（Python范兒）。大多數(shù)初學(xué)者不知道在哪兒使用它們，所以我將要分享幾個示例

前言：

以往看到我博客的小伙伴可能都知道，我的前言一般都是吐槽和講廢話環(huán)節(jié)，哈哈哈哈。今天難得休息，最近可真是太忙了，博主已經(jīng)連續(xù)一年都在996了，所以最近沒怎么學(xué)習(xí)新東西，只能回顧以往的舊知識了，上周一起工作的小伙伴扛不住996離職了，我們?nèi)说男F(tuán)隊也正式解散了，哎。原本想著找時間好好整理一個關(guān)于關(guān)于接口自動化測試或ceph相關(guān)的東西。但由于篇幅過長這里目前可能不會著手寫這方面東西。其實啊寫是很簡單的，主要例子難找。哈哈哈哈。
好了回歸正題吧?？催^我以往博客的小伙伴肯定見我用過@parameterized.expand()用作參數(shù)化的裝飾器。這屬于系統(tǒng)定義裝飾器，當(dāng)然我們自己也可以自定義裝飾器已適合我們?nèi)粘５拈_發(fā)需求。
**什么是裝飾器：**裝飾器的本質(zhì)就是一個函數(shù)，作用是在不改變源代碼的情況下，給函數(shù)增加額外的功能；裝飾器的使用通過@語法糖進(jìn)行調(diào)用。

普通裝飾器：

直接上代碼吧。

def demo(func_test):
    def wrapper():
        print('定義的第一個裝飾器')
        func_test()
        print('裝飾器結(jié)束')
    return wrapper    #這里注意不要加括號


@demo
def func_test():
    print('小白小白！?。?！')

在這里插入圖片描述

直接使用自定義裝飾器強(qiáng)化的原方法。
帶有參數(shù)的裝飾器：

def demo(func_test):
    def wrapper(name):
        print('定義的第一個裝飾器')
        func_test(name)
        print('哦，你就是%s啊'%name)
        print('裝飾器結(jié)束')
    return wrapper


@demo
def func_test(name):
    print('大家好我叫%s'%name)

在這里插入圖片描述

是不是有點感覺了啊。
接下來就是不帶自定義參數(shù)的裝飾器

def demo(func_test):
    def wrapper(*name,**kwargs):
        print('定義的第一個裝飾器')
        func_test(*name,**kwargs)
        print('哦，你就是%s啊'%name[0])
        print('你今年%s歲啊'%name[1])
        print('裝飾器結(jié)束')
    return wrapper


@demo
def func_test(name,age):
    print('大家好我叫%s'%name)
    print('今年{}'.format(age))

在這里插入圖片描述

是不是很簡答啊。哈哈哈哈
多個裝飾一起使用。

def deco01(func):
    def wrapper(*args,**kwargs):
        print('第一層的裝飾器')
        func(*args,**kwargs)
        print('第一層裝飾器結(jié)束')
    return wrapper


def deco02(func):
    def wrapper(*args,**kwargs):
        print('這是第二層裝飾器')
        print('第二層裝飾器結(jié)束')
        func(*args, **kwargs)
    return wrapper


@deco02
@deco01
def func(a,b,c,name):
    print('hello ,here is a func')
    print("result is %d" %(a+b+c))
    print('name:{}'.format(name))

在這里插入圖片描述

看看執(zhí)行順序可以看出裝飾器是從上往下執(zhí)行的。
類裝飾器：
在使用裝飾器時我們總不能在一個文件里寫的都是裝飾器和各個方法吧，總是要分開的。這里就要是講一個類裝飾器。
這里呢，我大概了寫了一個讀取文件裝飾器的例子。

class Mydecorator():
    def __init__(self,func):
    # 定義為私有屬性
        self.func = func
    # 實現(xiàn)__call__方法，讓對象變成可以調(diào)用的對象，可調(diào)用的對象可以想函數(shù)那樣使用
    def __call__(self, *args, **kwargs):   
        print('測試類裝飾器')
        self.func(*args)
        catalogue = args[0]
        suffix = catalogue.split('.')[2]
        try:
            if suffix == 'json':
                with open(*args,'r',encoding='utf-8') as file_object:
                    contents = json.load(file_object)
                    print(contents)
            else:
                with open(*args,'r',encoding='utf-8') as file_object:
                    contents = file_object.read()
                    print(contents)
        except Exception as a:
            print('讀取文件出錯拉：{}'.format(a))
@Mydecorator
def name(name):
    print('請輸入對應(yīng)文件路徑：%s'%name)

在這里插入圖片描述