基於hi-nginx的web開發(python篇)——路由裝飾器

zmark發表於2018-03-24

現在,有了起步的基本認識,現在需要一個可以媲美flask或者bottle的簡潔易用的路由功能,可以用裝飾器寫法任意對映 URLs 到程式碼。

這個,並不難。首先,來一個叫做hi的模組:hi.py:

 1 import re
 2 
 3 class hi:
 4     def __init__(self):
 5         self.uri_map={}
 6         self.uri_regex_map={}
 7 
 8     def route(self,pattern,method):
 9         def wrapper_a(func):
10             self.uri_map[pattern]={`method`:method,`callback`:func}
11             self.uri_regex_map[pattern]=re.compile(pattern)
12             def wrapper_b(req,res,param):
13                 func(req,res,param)
14             return wrapper_b
15         return wrapper_a
16 
17     def run(self,req,res):
18         for k,v in self.uri_map.items():
19             if req.method() in v[`method`]:
20                 m=self.uri_regex_map[k].match(req.uri())
21                 if m:
22                     v[`callback`](req,res,m.groupdict())
23                     break

把它和index.py放在同一個目錄中。以下就是使用路由裝飾器後的新程式碼:

 1 import sys
 2 sys.path.append(`/usr/local/nginx/python`)
 3 
 4 from hi import hi
 5 app =hi()
 6 
 7 @app.route(r`^/test/?$`,[`GET`,`POST`])
 8 @app.route(r"^/$",[`GET`])
 9 def hello_world(req,res,param):
10     res.header(`Content-Type`,`text/plain;charset=utf-8`)
11     res.content(`hello,world`)
12     res.status(200)
13 
14 @app.route(r"^/client/?$",[`GET`,`POST`])
15 def client(req,res,param):
16     res.content(`{}<br>{}<br>{}<br>{}<br>{}`.format(req.client(),req.method(),req.uri(),req.user_agent(),req.param()))
17     res.status(200)
18 
19 @app.route(r"^/hello/(?P<who>w+)?$",[`GET`])
20 def hello(req,res,param):
21     res.content(`{}={}`.format(`who`,param[`who`]))
22     res.status(200)
23 
24 
25 
26 if __name__ == `__main__`:
27     app.run(hi_req,hi_res)

 

是不是跟些flask或者bottle一樣簡單?而且還快得多喔!

訪問http://localhost:8080/,http://localhost:8080/client?a=90,http://localhost:8080/hello/cnblogs即可檢視結果。

 

當然,也可以先安裝hi.py:https://github.com/webcpp/hi.py 

這樣的話,上面程式碼的第1,2行就可以免了。

 


相關文章