基于hi-nginx的web開發（python篇）——路由裝飾器

現在，有了 起步

的基本認識，現在需要一個可以媲美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行就可以免了。