基于hi-nginx的web开发（python篇）——路由装饰器

现在，有了起步的基本认识，现在需要一个可以媲美flask或者bottle的简洁易用的路由功能，可以用装饰器写法任意映射 URLs 到代码。

这个，并不难。首先，来一个叫做hi的模块:hi.py:

 import re

 class hi:
     def __init__(self):
         self.uri_map={}
         self.uri_regex_map={}

     def route(self,pattern,method):
         def wrapper_a(func):
             self.uri_map[pattern]={'method':method,'callback':func}
             self.uri_regex_map[pattern]=re.compile(pattern)
             def wrapper_b(req,res,param):
                 func(req,res,param)
             return wrapper_b
         return wrapper_a

     def run(self,req,res):
         for k,v in self.uri_map.items():
             if req.method() in v['method']:
                 m=self.uri_regex_map[k].match(req.uri())
                 if m:
                     v['callback'](req,res,m.groupdict())
                     break

把它和index.py放在同一个目录中。以下就是使用路由装饰器后的新代码：

 import sys
 sys.path.append('/usr/local/nginx/python')

 from hi import hi
 app =hi()

 @app.route(r'^/test/?$',['GET','POST'])
 @app.route(r"^/$",['GET'])
 def hello_world(req,res,param):
     res.header('Content-Type','text/plain;charset=utf-8')
     res.content('hello,world')
     res.status(200)

 @app.route(r"^/client/?$",['GET','POST'])
 def client(req,res,param):
     res.content('{}<br>{}<br>{}<br>{}<br>{}'.format(req.client(),req.method(),req.uri(),req.user_agent(),req.param()))
     res.status(200)

 @app.route(r"^/hello/(?P<who>\w+)?$",['GET'])
 def hello(req,res,param):
     res.content('{}={}'.format('who',param['who']))
     res.status(200)

 if __name__ == '__main__':
     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行就可以免了。