如何正確使用裝飾器

本篇內(nèi)容主要講解“如何正確使用裝飾器”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“如何正確使用裝飾器”吧!

一、裝飾器的簡單定義

外層函數(shù)返回里層函數(shù)的引用，里層函數(shù)引用外層函數(shù)的變量。

二、裝飾器的作用

通俗來講裝飾器的作用就是在不改變已有函數(shù)代碼前提下，為該函數(shù)增加新的功能。

def run():    print('我會(huì)跑') fun()

現(xiàn)在我想在原有函數(shù)的基礎(chǔ)上新增一個(gè)功能：我會(huì)唱歌。這個(gè)時(shí)候利用裝飾器則輕松可以幫我們實(shí)現(xiàn)這個(gè)功能。

三、實(shí)例理解

(1)不傳參的裝飾器

def outer(fun):    def inner():       fun()  //fun是外層函數(shù)的變量，在inner里面用    return inner //inner就是里層函數(shù)的引用

(2)傳遞參數(shù)的裝飾器：

def func(fun):    def add(*args,**kwarge):       return fun(*args,**kwargs)    return add

現(xiàn)在對(duì)于裝飾器的基本格式有一定的了解，就可以直接寫函數(shù)了。下面實(shí)現(xiàn)文章開頭的 我會(huì)唱歌 的功能

def outer(fun):     def inner(*args, **kwarge):         print("我會(huì)唱歌")         return fun(*args, **kwarge)     return inner

四、如何使用裝飾器

方法一：使用@符號(hào)+裝飾器的名字   把它放在想要裝飾函數(shù)的上一行即可 @outer def run():    print('我會(huì)跑')     run()   方法二： def run():     print('我會(huì)跑')  run=outer(run)   #就等價(jià)于@outer run()  最終打印結(jié)果是： 我會(huì)唱歌 我會(huì)跑

如果我想知道fun 傳遞的參數(shù)是什么，在裝飾器內(nèi)部可以使用如下方式：

def outer(fun):     a = 1     def inner(*args, **kwarge): # args是一個(gè)數(shù)組，kwargs一個(gè)字典         print(fun.__name__) #打印fun接收的函數(shù)的名字         print("我會(huì)唱歌")         return fun(*args, **kwarge)     return inner

但是如果我們 print(run.__name__,6666666)  輸出的結(jié)果是inner，并不是我們想要的run，這里的函數(shù)被warpTheFunction替代了。它重寫了我們函數(shù)的名字和注釋文檔(docstring)。解決方法如下：

from functools import wraps  def outer(fun):     @wraps(fun)     def inner(*args, **kwargs):         print(fun.__name__,11111111111)         print("我會(huì)唱歌")         return fun(*args, **kwargs)     return inner  @outer def run():    print('我會(huì)跑')      print(run.__name__,6666666)  //輸出結(jié)果為 run 666666

五、自己實(shí)現(xiàn)裝飾器

def subuser_keymanage(view_func): 
    '''功能是實(shí)現(xiàn)用戶管理權(quán)限的判定''' 
    def _wrapper_view(request, *args, **kwargs): 
        user = request.user #一個(gè)Customer對(duì)象，包含了用戶名/密碼等信息 
        customer = user.customer.customer_id #用戶的id 
        select_status = get_curuser_permission(user=user, customer=customer)#調(diào)用函數(shù)返回的值有兩種0和1 
        if not select_status:#如果返回0表示沒有權(quán)限，返回錯(cuò)誤碼 
            return render_response(request, ErrorCode.FAILED) 
        return view_func(request, *args, **kwargs) 
    return _wrapper_view 

@subuser_keymanage  def generate_subuser_ak_sk(request):     params = json.loads(request.body) #獲取卡前端傳遞的參數(shù)     user_id_only = params.get("user_id") #獲取用戶表示id值     中間代碼就忽略了......     return render_response(request, ErrorCode.FAILED)

六、裝飾器小結(jié)

通過裝飾器很大程度上可以減少代碼的復(fù)用，在代碼規(guī)范中這一點(diǎn)是很重要的。

以上就是裝飾器的基本知識(shí)，即便沒有任何基礎(chǔ)，按照作者的思路，套用固定的格式，不需要完全理解，只要按照流程一步一步就能寫出高端大氣上檔次的裝飾器了，恭喜你!

前方高能請注意：裝飾器傳參，三層嵌套函數(shù)一般用的比較少，其實(shí)也不難，一層一層看，跟上文講的一樣，僅作為知識(shí)的拓寬。

import logging 
def use_logging(level): 
    def decorator(func): 
        def wrapper(*args, **kwargs): 
            if level == "warn": 
                logging.warn("%s is running" % func.__name__) 
            elif level == "info": 
                logging.info("%s is running" % func.__name__) 
            return func(*args) 
        return wrapper 
 
    return decorator 
 
@use_logging(level="warn") 
def foo(name='foo'): 
    print("i am %s" % name) 
 
foo() 

i am foo WARNING:root:foo is running

到此，相信大家對(duì)“如何正確使用裝飾器”有了更深的了解，不妨來實(shí)際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！