基於hi-nginx的web開發(python篇)——路由裝飾器,hi-nginxpython
現在,有了起步的基本認識,現在需要一個可以媲美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_b15 return wrapper_a16 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 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即可查看結果。