首先來(lái)分析下需求,web程序后臺(tái)需要認(rèn)證,后臺(tái)頁(yè)面包含多個(gè)頁(yè)面,最普通的方法就是為每個(gè)url添加認(rèn)證,但是這樣就需要每個(gè)每個(gè)綁定url的后臺(tái)函數(shù)都需要添加類(lèi)似或者相同的代碼,但是這樣做代碼就過(guò)度冗余,而且不利于擴(kuò)展.
接下來(lái)我們先不談及裝飾器,我們都知道Python是個(gè)很強(qiáng)大的語(yǔ)言,她可以將函數(shù)當(dāng)做參數(shù)傳遞給函數(shù),最簡(jiǎn)單的:
1
2
3
4
5
6
7
8
9
10
11
12
13
|
def p(): print 'Hello,world' def funcfactor(func): print 'calling function named' , func.__name__ func() print 'end' funcfactor(p) # 輸出為: # calling function named p # Hello,world # end |
一目了然的程序,定義一個(gè)函數(shù)p(),將函數(shù)p當(dāng)做參數(shù)傳遞給喊出funcfactor,在執(zhí)行p函數(shù)前后加上一些動(dòng)作.
我們還可以這么做:
1
2
3
4
5
6
7
8
9
10
11
|
def p(): print 'Hello,world' def funcfactor(func): print 'calling function named' , func.__name__ return func func = funcfactor(p) func() # 輸出為: # calling function named p Hello,world |
正如你看到的,我們可以將函數(shù)返回然后賦予一個(gè)變量,留待稍后調(diào)用.但是這種情況下我們要想在函數(shù)執(zhí)行后做點(diǎn)什么就不可能,但是我們的Python是強(qiáng)大的,Python可以在函數(shù)中再嵌套一個(gè)函數(shù),我們可以像下面這么做:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
def p(): print 'Hello, world' def funcfactor(func): def wrapper(): print 'do something at start' func() print 'do something at end' return wrapper func = funcfactor(p) func() #輸出為: # do something at start # Hello, world # do something at end |
下面我們來(lái)看看裝飾器,上面的代碼雖然實(shí)現(xiàn)的一個(gè)很困難的任務(wù),但是還不夠優(yōu)雅,而且代碼不符合Python的哲學(xué)思想,所以裝飾器就應(yīng)聲而出,裝飾器沒(méi)有和上面的原理相同,同樣用于包裝函數(shù),只是代碼實(shí)現(xiàn)上更加優(yōu)雅和便于閱讀.裝飾器以@開(kāi)頭后面跟上裝飾器的名稱(chēng),緊接著下一行就是要包裝的函數(shù)體,上面的例子用裝飾器可用如下方式實(shí)現(xiàn):
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
def decorator(func): def wrapper(): print 'do something at start' func() print 'do something at end' return wrapper @decorator def p(): print 'Hello, world' p() #輸出為: # do something at start # Hello, world # do something at end |
實(shí)際上裝飾器并沒(méi)有性能方面或其他方面的提升,僅僅是一種語(yǔ)法糖,就是上面一個(gè)例子的改寫(xiě),這樣更加優(yōu)雅和便與閱讀. 如果我們的p()函數(shù)不想僅僅只輸Hello,world,我們想向某些我們指定的人打招呼:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
def decorator(func): def wrapper( * args, * * kargs): print 'do something at start' func( * * kargs) print 'do something at end' return wrapper @decorator def p(name): print 'Hello' , name p(name = "Jim" ) #輸出為: # do something at start # Hello Jim # do something at end |
裝飾器在裝飾不需要參數(shù)的裝飾器嵌套函數(shù)不是必須得,如果被裝飾的函數(shù)需要參數(shù),必須嵌套一個(gè)函數(shù)來(lái)處理參數(shù). 寫(xiě)到這里想必大家也知道裝飾器的用法和作用.現(xiàn)在回到正題,如何優(yōu)雅的給后臺(tái)url加上驗(yàn)證功能?毫無(wú)疑問(wèn)我們使用裝飾器來(lái)處理:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
def blog_auth(func): ''' 定義一個(gè)裝飾器用于裝飾需要驗(yàn)證的頁(yè)面 裝飾器必須放在route裝飾器下面 ''' # 定義包裝函數(shù) def wrapper( * args, * * kargs): try : # 讀取cookie user = request.COOKIES[ 'user' ] shell = request.COOKIES[ 'shell' ] except : # 出現(xiàn)異常則重定向到登錄頁(yè)面 redirect( '/login' ) # 驗(yàn)證用戶(hù)數(shù)據(jù) if checkShell(user, shell): # 校驗(yàn)成功則返回函數(shù) return func( * * kargs) else : # 否則則重定向到登錄頁(yè)面 redirect( '/login' ) return wrapper |
可以再需要驗(yàn)證的地方添加blog_auth裝飾器:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
@route ( '/admin:#/?#' ) @blog_auth def admin(): ''' 用于顯示后臺(tái)管理首頁(yè) ''' TEMPLATE[ 'title' ] = '儀表盤(pán) | ' + TEMPLATE[ 'BLOG_NAME' ] TEMPLATE[ 'user' ] = request.COOKIES[ 'user' ] articles = [] for article in db.posts.find().sort( "date" ,DESCENDING).limit( 10 ): articles.append(article) # 將文章列表交給前臺(tái)模版 TEMPLATE[ 'articles' ] = articles return template( 'admin.html' ,TEMPLATE) |
至此bottle驗(yàn)證的問(wèn)題就很優(yōu)雅的用裝飾器解決了.