快捷導(dǎo)航

Python 裝飾器代碼解析

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

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

前言：

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

普通裝飾器：

直接上代碼吧。

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


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

在這里插入圖片描述

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

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


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

在這里插入圖片描述

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

def demo(func_test):
    def wrapper(*name,**kwargs):
        print('定義的第一個(gè)裝飾器')
        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))

在這里插入圖片描述

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

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í)行的。
類裝飾器：
在使用裝飾器時(shí)我們總不能在一個(gè)文件里寫的都是裝飾器和各個(gè)方法吧，總是要分開的。這里就要是講一個(gè)類裝飾器。
這里呢，我大概了寫了一個(gè)讀取文件裝飾器的例子。

class Mydecorator():
    def __init__(self,func):
    # 定義為私有屬性
        self.func = func
    # 實(shí)現(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('讀取文件出錯(cuò)拉：{}'.format(a))
@Mydecorator
def name(name):
    print('請輸入對應(yīng)文件路徑：%s'%name)

在這里插入圖片描述