摘要：最近看到一個(gè)關(guān)于的題文章其中的一個(gè)是裝飾器的順序問題就想寫篇博客回顧下裝飾器首先強(qiáng)烈推薦很久之前看的一篇博文翻譯理解中的裝飾器關(guān)于什么是裝飾器看這篇文章就好了這里主要想寫關(guān)于多個(gè)裝飾器的執(zhí)行流程裝飾順序示例代碼初始化初始化輸出結(jié)果初始化初始

最近看到一個(gè)關(guān)于Flask的CTF(RealWorld CTF 2018 web題bookhub)文章
其中的一個(gè)trick是裝飾器的順序問題,就想寫篇博客回顧下裝飾器~

首先強(qiáng)烈推薦很久之前看的一篇博文
(翻譯)理解PYTHON中的裝飾器
關(guān)于什么是裝飾器看這篇文章就好了~ 
這里主要想寫關(guān)于多個(gè)裝飾器的執(zhí)行流程

示例代碼

# import pdb;pdb.set_trace()

def functionOne(function_to_decorate):
    print("functionOne初始化")
  
    def wrapperOne():
pass
    return wrapperOne

def functionTwo(function_to_decorate):
    print("functionTwo初始化")
  
    def wrapperTwo():
pass
    return wrapperTwo

@functionOne
@functionTwo
def testFunction():
    pass

# 輸出結(jié)果
functionTwo初始化
functionOne初始化

從上面我們能得知：裝飾順序,就近裝飾
然后我們利用下面的代碼進(jìn)行一步探究
如下我們得知：執(zhí)行這段代碼，相當(dāng)于：
首先，將testFunction函數(shù)打包給wrapperTwo,由于沒有調(diào)用，functionTwo整體返回了wrapperTwo，而沒有執(zhí)行
然后，functionOne將wrapperTwo作為參數(shù)，打包成wrapperOne

# import pdb;pdb.set_trace()

def functionOne(function_to_decorate):
    print("functionOne初始化")
  
    def wrapperOne():
print("第一處"+function_to_decorate.__name__)
function_to_decorate()
    return wrapperOne

def functionTwo(function_to_decorate):
    print("functionTwo初始化")
  
    def wrapperTwo():
print("第二處"+function_to_decorate.__name__)
function_to_decorate()
    return wrapperTwo

@functionOne
@functionTwo
def testFunction():
    print("index")

testFunction()

#輸出結(jié)果
functionTwo初始化
functionOne初始化
第一處wrapperTwo
第二處testFunction
index

從上面的第二段代碼我們已經(jīng)能看出部分執(zhí)行順序了
就是它會(huì)優(yōu)先執(zhí)行我們打包好的wrapperOne,因?yàn)閺钠鹗嫉膖estFunction，wrapperTwo都已經(jīng)打包在wrapperOne
可以說成執(zhí)行順序，就遠(yuǎn)執(zhí)行
我們繼續(xù)執(zhí)行下面的代碼：

# import pdb;pdb.set_trace()

def functionOne(function_to_decorate):
    print("functionOne初始化")
  
    def wrapperOne():
print("第一處"+function_to_decorate.__name__)
function_to_decorate()
print("wrapperOne")
    return wrapperOne

def functionTwo(function_to_decorate):
    print("functionTwo初始化")
  
    def wrapperTwo():
print("第二處"+function_to_decorate.__name__)
function_to_decorate()
print("wrapperTwo")
    return wrapperTwo

@functionOne
@functionTwo
def testFunction():
    print("index")

testFunction()

# 輸出結(jié)果
functionTwo初始化
functionOne初始化
第一處wrapperTwo
第二處testFunction
index
wrapperTwo
wrapperOne

這個(gè)執(zhí)行順序可能也困擾了很多人，現(xiàn)在我們從輸出結(jié)果看
對照代碼，就很容易清楚了，執(zhí)行到wrapperOne中的function_to_decorate時(shí)
其實(shí)相當(dāng)于跳轉(zhuǎn)到了函數(shù)wrapperTwo,然后執(zhí)行wrapperTwo

從上面的幾個(gè)例子我們應(yīng)該大概了解了，多個(gè)裝飾器進(jìn)行裝飾以及執(zhí)行的順序
我們來看這道CTF題目，我們首先需要知道的是Flask中路由就是一個(gè)裝飾

from flask import Flask

app = Flask(__name__)
app.debug = True

# import pdb;pdb.set_trace()

# 為了更好的控制輸出，自定義了loginRequire裝飾器
def loginRequire(function_to_decorate):
    print("loginRequire初始化")
  
    def wrapperTwo():
print("loginRequire裝飾成功")
print(function_to_decorate.__name__)
return function_to_decorate()
    return wrapperTwo

@loginRequire
@app.route("/up")
def up():
    return "裝飾路由放在上面！"

@app.route("/down")
@loginRequire
def down():
    return "裝飾路由放在下面！"

if __name__ == "__main__":
    app.run()

# 分別訪問兩個(gè)url輸出結(jié)果
loginRequire初始化
loginRequire初始化
 * Debugger is active!
 * Debugger PIN: 244-957-971
 * Running on http://127.0.0.1:5000/ (Press CTRL+C to quit)
127.0.0.1 - - [24/Aug/2018 19:01:30] "GET /up HTTP/1.1" 200 -
loginRequire裝飾成功
down
127.0.0.1 - - [24/Aug/2018 19:01:35] "GET /down HTTP/1.1" 200 -

從輸出結(jié)果我們能清楚的看到up的裝飾，并沒有執(zhí)行裝飾器
如果按照我們上面的分析，無論在上面還是下面都會(huì)執(zhí)行的啊？？只是順序不同罷了~
我們利用pdb來一步步調(diào)試查看哪里的問題,部分log如下：

> c:usersayidesktop	est256.py(17)()
-> @loginRequire
(Pdb) s
> c:usersayidesktop	est256.py(18)()
-> @app.route("/up")
(Pdb) s
> c:usersayi.virtualenvs	est-gq7eoxbqlibsite-packagesflaskapp.py(1252)route()->
-> return decorator
(Pdb) s
--Call--
> c:usersayi.virtualenvs	est-gq7eoxbqlibsite-packagesflaskapp.py(1248)decorator()
-> def decorator(f):
(Pdb) f

(Pdb) s
> c:usersayi.virtualenvs	est-gq7eoxbqlibsite-packagesflaskapp.py(1249)decorator()
-> endpoint = options.pop("endpoint", None)
(Pdb) s
> c:usersayi.virtualenvs	est-gq7eoxbqlibsite-packagesflaskapp.py(1250)decorator()
-> self.add_url_rule(rule, endpoint, f, **options)
(Pdb) f

#===================================================================================#
        上方up 下方down
#===================================================================================#
> c:usersayidesktop	est256.py(22)()
-> @app.route("/down")
(Pdb) s
> c:usersayidesktop	est256.py(23)()
-> @loginRequire
(Pdb) s
--Call--
> c:usersayidesktop	est256.py(6)loginRequire()
-> def loginRequire(function_to_decorate):
(Pdb) s
> c:usersayidesktop	est256.py(7)loginRequire()
-> print("loginRequire初始化")
(Pdb) s
loginRequire初始化
> c:usersayidesktop	est256.py(9)loginRequire()
-> def wrapperTwo():
(Pdb) s
> c:usersayidesktop	est256.py(13)loginRequire()
-> return wrapperTwo
(Pdb) s
--Return--
> c:usersayidesktop	est256.py(13)loginRequire()->
-> return wrapperTwo
(Pdb) s
--Call--
> c:usersayi.virtualenvs	est-gq7eoxbqlibsite-packagesflaskapp.py(1248)decorator()
-> def decorator(f):
(Pdb) f
.wrapperTwo at 0x0071C468>

從上面的執(zhí)行流程，打印出不斷出現(xiàn)的f,我們能看出，兩個(gè)順序的f值不同
在up中，f=up()
在down中，f=wrapperTwo()
這點(diǎn)符合預(yù)期，裝飾位置不同，然而在執(zhí)行Flask源碼 add_url_rule時(shí)
如上面log所示，直接添加了f的值

> c:usersayi.virtualenvs	est-gq7eoxbqlibsite-packagesflaskapp.py(1249)decorator()
-> endpoint = options.pop("endpoint", None)
(Pdb) s
> c:usersayi.virtualenvs	est-gq7eoxbqlibsite-packagesflaskapp.py(1250)decorator()
-> self.add_url_rule(rule, endpoint, f, **options)
(Pdb) f

也就是添加路由的時(shí)候會(huì)選擇丟失外層的路由，只裝飾route下方的函數(shù)
在add_url_rule中，有這段注釋：

Basically this example::

    @app.route("/")
    def index():
        pass

Is equivalent to the following::

    def index():
        pass
    app.add_url_rule("/", "index", index)

博客地址